org.apache.hadoop.util.SerializableUtils Maven / Gradle / Ivy
package org.apache.hadoop.util;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
public class SerializableUtils {
public static byte[] toBytes(T obj)
throws IOException {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
ObjectOutputStream out = null;
byte[] buff = null;
try {
out = new ObjectOutputStream(bos);
out.writeObject(obj);
buff = bos.toByteArray();
} finally {
if (out != null) {
out.close();
}
bos.close();
}
return buff;
}
public static Object getFromBytes(byte[] data,
Class expectedType)
throws IOException,
ClassNotFoundException {
if (data == null) {
throw new IOException("data is null");
}
ByteArrayInputStream in = new ByteArrayInputStream(data);
ObjectInputStream is = null;
Object obj = null;
try {
is = new ObjectInputStream(in);
obj = is.readObject();
if (obj.getClass() != expectedType) {
throw new IOException(
"The data provided is not a serialized object of type : "
+ expectedType);
}
} finally {
if (is != null) {
is.close();
}
in.close();
}
return obj;
}
}