src/main/java/derms/replica/replica2/ObjectPacket.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

package derms.replica.replica2;

import java.io.*;
import java.net.DatagramPacket;
import java.net.SocketAddress;

class ObjectPacket {
	static DatagramPacket create(Serializable obj, SocketAddress remoteAddr) throws IOException {
		ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
		ObjectOutputStream objStream = new ObjectOutputStream(byteStream);
		objStream.writeObject(obj);
		objStream.flush();
		byte[] buf = byteStream.toByteArray();
		objStream.close();
		return new DatagramPacket(buf, buf.length, remoteAddr);
	}

	static <E> E deserialize(DatagramPacket pkt, Class<E> clazz) throws IOException {
		ObjectInputStream objectStream;
		try {
			objectStream = new ObjectInputStream(
				new ByteArrayInputStream(pkt.getData()));
		} catch (Exception e) {
			throw new IOException("failed to create input stream: "+e.getMessage());
		}

 		try {
			return clazz.cast(objectStream.readObject());
		} catch (Exception e) {
			throw new IOException(e.getMessage());
		}
	}
}