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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
package derms.replica3;
//import logger.Logger;
import javax.xml.ws.Endpoint;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;
import java.nio.charset.StandardCharsets;
import derms.replica3.Constants;
import derms.replica3.Constants;
public class MTLServer {
public static void main(String[] args) {
try{
final Logger activityLogger = new Logger("MTLServer.log");
int udpPort = PortConstants.getUdpPort("MTL");
City ct = new City("MTL");
final Replica3 ServerImpl = new Replica3(ct, null);
// Endpoint endpoint = Endpoint.publish(Constants.MONTREAL_ADDRESS, ServerImpl);
/*Start UDP server for MTL*/
new Thread(() -> {
startUdpServer(activityLogger, ServerImpl, udpPort);
}).start();
System.out.println("#=== MTL Server is started ===#");
} catch (Exception e) {
System.err.println("ERROR: " + e);
e.printStackTrace(System.out);
}
}
private static void startUdpServer(final Logger activityLogger, final Replica3 server, int udpPort) {
DatagramSocket socket = null;
try {
socket = new DatagramSocket(udpPort);
// activityLogger.log(MessageTypeConstants.INFO, String.format(UdpServerMessages.UDP_SERVER_STARTED, server.getServerName()));
while (true) {
try {
final byte[] data = new byte[1000];
final DatagramPacket packet = new DatagramPacket(data, data.length);
socket.receive(packet);
System.out.println("request received" + udpPort);
new Thread(() -> {
processRequest(activityLogger, server, packet);
}).start();
} catch (IOException e) {
// activityLogger.log("MessageTypeConstants.ERROR", e.getMessage());
}
}
} catch (SocketException e1) {
// activityLogger.log(MessageTypeConstants.ERROR, e1.getMessage());
} finally {
if (socket != null) {
socket.close();
}
}
}
private static void processRequest(final Logger activityLogger, final Replica3 server, final DatagramPacket packet) {
byte[] response;
DatagramSocket socket = null;
final String request = new String(packet.getData(), StandardCharsets.UTF_8).trim();
final String[] packetData = request.split("-");
final String sourceServer = packetData[0].trim();
final String action = packetData[1].trim();
final String resourceName = packetData[2].trim();
String resourceID = "";
int duration = 0;
System.out.println("action:" + action);
if(packetData.length == 4) {
resourceID = packetData[2].trim();
duration = Integer.parseInt(packetData[3]);
}
try {
socket = new DatagramSocket();
if("GET_AVAILABLE_RESOURCES".equalsIgnoreCase(action)) {
response = sanitizeXml(server.getAvailableResources(resourceName).toString()).getBytes(StandardCharsets.UTF_8);
socket.send(new DatagramPacket(response, response.length, packet.getAddress(), packet.getPort()));
}else if("GET_RESOURCE".equalsIgnoreCase(action)) {
response = server.getRequestedResource(resourceID, duration).toString().getBytes();
socket.send(new DatagramPacket(response, response.length, packet.getAddress(), packet.getPort()));
}
// activityLogger.log("", action, "Response sent");
} catch (IOException e) {
// activityLogger.log("", "EXCEPTION", e.getMessage());
} finally {
if (socket != null) {
socket.close();
}
}
}
private static String sanitizeXml(String input) {
if (input == null) {
return ""; // or handle as needed
}
return input.replaceAll("[\\x00]", "");
}
}
|