summaryrefslogtreecommitdiffstats
path: root/src/main/java/derms/replica/replica1/ResourceTransfer.java
blob: fbc695cd4fd7fc346a4608781ba6dbf830debf08 (plain) (blame)
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package derms.replica.replica1;

import java.io.IOException;
import java.io.Serializable;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketAddress;
import java.util.ArrayList;
import java.util.List;

public class ResourceTransfer {
	public static final int bufsize = 1024;

	public static void send(Resource[] resources, SocketAddress remoteAddr) throws IOException {
		DatagramSocket sock = new DatagramSocket();

		for (Resource resource : resources) {
			DatagramPacket pkt = ObjectPacket.create(resource, remoteAddr);
			sock.send(pkt);
		}

		DatagramPacket pkt = ObjectPacket.create(new EndOfTransmission(), remoteAddr);
		sock.send(pkt);
		sock.close();
	}

	public static Resource[] receive(DatagramSocket sock) throws IOException {
		List<Resource> resources = new ArrayList<Resource>();
		byte[] buf = new byte[bufsize];
		DatagramPacket response = new DatagramPacket(buf, buf.length);

		for (;;) {
			sock.receive(response);

			Object obj = ObjectPacket.deserialize(response, Object.class);
			if (obj.getClass() == EndOfTransmission.class) {
				break;
			}
			try {
				resources.add((Resource) obj);
			} catch (Exception e) {
				throw new IOException("expected Resource; got "+obj.getClass().toString());
			}
		}
		Resource[] arr = new Resource[0];
		return resources.toArray(arr);
	}

	private static class EndOfTransmission implements Serializable {}
}