1 files changed, 128 insertions, 0 deletions
diff --git a/src/main/java/derms/replica/replica2/ResourceAvailability.java b/src/main/java/derms/replica/replica2/ResourceAvailability.java
new file mode 100644
index 0000000..8eacbf2
--- /dev/null
+++ b/src/main/java/derms/replica/replica2/ResourceAvailability.java
@@ -0,0 +1,128 @@
+package derms.replica.replica2;
+
+import java.io.IOException;
+import java.net.DatagramPacket;
+import java.net.DatagramSocket;
+import java.net.InetAddress;
+import java.net.InetSocketAddress;
+import java.util.Collection;
+import java.util.logging.Logger;
+
+public class ResourceAvailability {
+	public static final int port = 5556;
+
+	public static class Client implements Runnable {
+		private InetAddress serverAddr;
+		private ResourceName request;
+		private Collection<Resource> resources;
+		private Logger log;
+
+		public Client(InetAddress serverAddr, ResourceName request, Collection<Resource> response) throws IOException {
+			this.serverAddr = serverAddr;
+			this.request = request;
+			this.resources = response;
+			this.log = DermsLogger.getLogger(this.getClass());
+		}
+
+		@Override
+		public void run() {
+			DatagramSocket sock;
+			try {
+				sock = new DatagramSocket();
+			} catch (Exception e) {
+				log.severe("Error binding socket: "+e.getMessage());
+				return;
+			}
+			log.fine("Created socket");
+
+			DatagramPacket reqPkt;
+			try {
+				reqPkt = ObjectPacket.create(request, new InetSocketAddress(serverAddr, port));
+			} catch (IOException e) {
+				log.severe("Error creating request: "+e.getMessage());
+				sock.close();
+				return;
+			}
+
+			try {
+				sock.send(reqPkt);
+			} catch (IOException e) {
+				log.severe("Error sending request: "+e.getMessage());
+				sock.close();
+				return;
+			}
+			log.fine("Sent request");
+
+			Resource[] response;
+			try {
+				response = ResourceTransfer.receive(sock);
+			} catch (IOException e) {
+				log.severe(e.getMessage());
+				return;
+			} finally {
+				sock.close();
+			}
+
+			for (Resource resource : response) {
+				resources.add(resource);
+			}
+		}
+	}
+
+	public static class Server implements Runnable {
+		public static final int bufsize = 1024;
+
+		private InetAddress localAddr;
+		private Resources resources;
+		private Logger log;
+
+		public Server(InetAddress localAddr, Resources resources) throws IOException {
+			this.localAddr = localAddr;
+			this.resources = resources;
+			this.log = DermsLogger.getLogger(this.getClass());
+		}
+
+		@Override
+		public void run() {
+			DatagramSocket sock = null;
+			try {
+				sock = new DatagramSocket(port, localAddr);
+			} catch (Exception e) {
+				log.severe("Failed to bind socket to "+localAddr.toString()+": "+e.getMessage());
+				return;
+			}
+
+			log.info("Listening on "+localAddr.toString()+":"+port);
+
+			DatagramPacket request = new DatagramPacket(new byte[bufsize], bufsize);
+			try {
+				for (;;) {
+					try {
+						sock.receive(request);
+					} catch (Exception e) {
+						log.warning("Error receiving from socket: "+e.getMessage());
+						continue;
+					}
+
+					ResourceName requestedName = null;
+					try {
+						requestedName = ObjectPacket.deserialize(request, ResourceName.class);
+					} catch (IOException e) {
+						log.warning("Failed to deserialize request: "+e.getMessage());
+						continue;
+					}
+					log.info("Got request: "+requestedName);
+
+					Resource[] response = resources.getByName(requestedName);
+					try {
+						ResourceTransfer.send(response, request.getSocketAddress());
+					} catch (IOException e) {
+						log.warning("Error transfering resources: "+e.getMessage());
+					}
+				}
+			} finally {
+				sock.close();
+			}
+		}
+	}
+}