inspection listener

jdk-1.6-parent/serializer-kryo2/src/main/java/org/wicketstuff/pageserializer/kryo2/inspecting/ISerializationListener.java

@@ -2,8 +2,11 @@
 
 public interface ISerializationListener {
 
+	void start(Object object);
+
 	void begin(int position, Object object);
 
 	void end(int position, Object object);
 
+	void end(Object object);
 }

jdk-1.6-parent/serializer-kryo2/src/main/java/org/wicketstuff/pageserializer/kryo2/inspecting/InspectingKryo.java

@@ -13,10 +13,10 @@
 public class InspectingKryo extends KryoReflectionFactorySupport {
 
 	private final static Logger LOG = LoggerFactory.getLogger(InspectingKryo.class);
-	private final ISerializationListener serializingListener;
+	private final InspectionKryoSerializer parent;
 
-	public InspectingKryo(ISerializationListener serializingListener) {
-		this.serializingListener = serializingListener;
+	public InspectingKryo(InspectionKryoSerializer parent) {
+		this.parent = parent;
 	}
 
 	@Override
@@ -81,11 +81,11 @@ public void writeObjectOrNull(Output output, Object object,
 	}
 
 	private void before(Output output, Object object) {
-		serializingListener.begin(output.position(),object);
+		parent.serializingListener().begin(output.position(),object);
 	}
 
 	private void after(Output output, Object object) {
-		serializingListener.end(output.position(),object);
+		parent.serializingListener().end(output.position(),object);
 	}
 
 }

jdk-1.6-parent/serializer-kryo2/src/main/java/org/wicketstuff/pageserializer/kryo2/inspecting/InspectionKryoSerializer.java

@@ -0,0 +1,35 @@
+package org.wicketstuff.pageserializer.kryo2.inspecting;
+
+import org.apache.wicket.util.lang.Bytes;
+import org.wicketstuff.pageserializer.kryo2.KryoSerializer;
+
+import com.esotericsoftware.kryo.Kryo;
+
+public class InspectionKryoSerializer extends KryoSerializer {
+
+	private final ISerializationListener serializingListener;
+
+	public InspectionKryoSerializer(Bytes size, ISerializationListener serializingListener) {
+		super(size);
+		this.serializingListener = serializingListener;
+	}
+
+	@Override
+	protected Kryo createKryo() {
+		return new InspectingKryo(this);
+	}
+
+	@Override
+	public byte[] serialize(Object object) {
+		try {
+			serializingListener.start(object);
+			return super.serialize(object);
+		} finally {
+			serializingListener.end(object);
+		}
+	}
+
+	protected final ISerializationListener serializingListener() {
+		return serializingListener;
+	}
+}

jdk-1.6-parent/serializer-kryo2/src/main/java/org/wicketstuff/pageserializer/kryo2/inspecting/LoggingSerializationListener.java

@@ -8,6 +8,12 @@ public class LoggingSerializationListener implements ISerializationListener {
 	private final static Logger LOG = LoggerFactory
 			.getLogger(LoggingSerializationListener.class);
 
+	@Override
+	public void start(Object object) {
+		LOG.error("Start for object: '{}'", 
+				object.getClass());
+	}
+
 	@Override
 	public void begin(int position, Object object) {
 		LOG.error("Start at '{}' byte for object:  '{}'", position,
@@ -19,5 +25,11 @@ public void end(int position, Object object) {
 		LOG.error("End at   '{}' bytes for object: '{}'", position,
 				object != null ? object.getClass() : "NULL");
 	}
+
+	@Override
+	public void end(Object object) {
+		LOG.error("End for object:   '{}'", 
+				object.getClass());
+	}
 
 }

jdk-1.6-parent/serializer-kryo2/src/test/java/org/wicketstuff/pageserializer/kryo2/inspecting/InspectingKryoTest.java

@@ -6,29 +6,23 @@
 
 import junit.framework.Assert;
 
+import org.apache.wicket.util.lang.Bytes;
 import org.junit.Test;
 
-import com.esotericsoftware.kryo.io.Input;
-import com.esotericsoftware.kryo.io.Output;
-
 public class InspectingKryoTest {
 
 	@Test
 	public void writeStringList() {
 		List<String> stringList=new ArrayList<String>();
 		stringList.addAll(Arrays.asList("One","2","III"));
 
-		InspectingKryo kryo=new InspectingKryo(new LoggingSerializationListener());
+		InspectionKryoSerializer kryo=new InspectionKryoSerializer(Bytes.bytes(100),new LoggingSerializationListener());
 
-		Output buffer = new Output(100);
-		kryo.writeClassAndObject(buffer, stringList);
-		byte[] data=buffer.toBytes();
-		buffer.clear();
-		System.runFinalization();
+		byte[] data=kryo.serialize(stringList);
 
 		Assert.assertNotNull(data);
 
-		List<String> readBack=(List<String>) kryo.readClassAndObject(new Input(data));
+		List<String> readBack=(List<String>) kryo.deserialize(data);
 
 		Assert.assertNotNull(readBack);
 		Assert.assertEquals(stringList, readBack);