Source code

001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *   http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing,
013 * software distributed under the License is distributed on an
014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015 * KIND, either express or implied.  See the License for the
016 * specific language governing permissions and limitations
017 * under the License.
018 */
019package org.apache.reef.io.serialization;
020
021import javax.inject.Inject;
022import java.io.*;
023import java.util.logging.Logger;
024
025/**
026 * A {@link Codec} for {@link Serializable} objects.
027 * <p>
028 * It uses java serialization, use with caution.
029 *
030 * @param <T> The type of objects Serialized
031 */
032public class SerializableCodec<T extends Serializable> implements Codec<T> {
033
034  private static final Logger LOG = Logger.getLogger(SerializableCodec.class.getName());
035
036  /**
037   * Default constructor for TANG use.
038   */
039  @Inject
040  public SerializableCodec() {
041  }
042
043  @Override
044  public byte[] encode(final T obj) {
045    try (final ByteArrayOutputStream bout = new ByteArrayOutputStream()) {
046      try (final ObjectOutputStream out = new ObjectOutputStream(bout)) {
047        out.writeObject(obj);
048      }
049      return bout.toByteArray();
050    } catch (final IOException ex) {
051      throw new RuntimeException("Unable to encode: " + obj, ex);
052    }
053  }
054
055  @Override
056  public T decode(final byte[] buf) {
057    try {
058      try (final ObjectInputStream oin = new ObjectInputStream(new ByteArrayInputStream(buf))) {
059        final T result = (T) oin.readObject();
060        return result;
061      }
062    } catch (final IOException | ClassNotFoundException ex) {
063      throw new RuntimeException("Unable to decode.", ex);
064    }
065
066  }
067}