src/main/java/derms/net/Packet.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

package derms.net;

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

public class Packet {
    public static DatagramPacket encode(Serializable obj, SocketAddress dst) 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, dst);
    }

    public static DatagramPacket encode(Serializable obj, InetAddress dstAddr, int dstPort) throws IOException {
        return encode(obj, new InetSocketAddress(dstAddr, dstPort));
    }

    public static <T extends Serializable> T decode(DatagramPacket pkt, Class<T> clazz) throws IOException, ClassNotFoundException, ClassCastException {
        ObjectInputStream objStream = new ObjectInputStream(
                new ByteArrayInputStream(pkt.getData()));
        T obj = clazz.cast(objStream.readObject());
        objStream.close();
        return obj;
    }
}