ContextAwareObjectInputStream.java

package NetworkAdapter.Implementation;

import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectStreamClass;

/**
 * Created with IntelliJ IDEA.
 * User: N3trunner
 * Date: 19.06.12
 * Time: 11:12
 * To change this template use File | Settings | File Templates.
 */

/**
 * An ObjectInputStream implementation that takes into account the context classloader when loading classes.
 */
public class ContextAwareObjectInputStream extends ObjectInputStream {

    private ClassLoader currentTccl;

    public ContextAwareObjectInputStream(InputStream in) throws IOException {
        super(in);
        currentTccl = Thread.currentThread().getContextClassLoader();
    }

    @Override
    public Class resolveClass(ObjectStreamClass desc) throws IOException, ClassNotFoundException {
        try {
            return super.resolveClass(desc);
        } catch (ClassNotFoundException e) {
            return Class.forName(desc.getName(), true, currentTccl);
        }
    }

}