summaryrefslogtreecommitdiffstats
path: root/gearman/controllerWorker/ControllerWorker/StatusWorker.java
blob: 7343ba59b8f0c938fc5cbef396fa3f5252c847fe (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
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
package ControllerWorker;

import java.util.HashMap;
import java.util.Vector;

import org.gearman.client.GearmanJobResult;
import org.gearman.client.GearmanJobResultImpl;
import org.gearman.util.ByteUtils;
import org.gearman.worker.AbstractGearmanFunction;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;

public class StatusWorker extends AbstractGearmanFunction {
	static HashMap<String, Boot> BOOTTHREADS = new HashMap<String, Boot>();
	static HashMap<String, Shutdown> SHUTDOWNTHREADS = new HashMap<String, Shutdown>();

	@Override
	public String getName() {
		return "status";
	}

	@Override
	public GearmanJobResult executeFunction() {
		String data = ByteUtils.fromUTF8Bytes((byte[]) this.data);
		Vector<String> events = new Vector<String>();
		JSONObject jsonObject = (JSONObject) JSONValue.parse(data);
		String type = jsonObject.get("type").toString();
		JSONArray jsonArray = (JSONArray) jsonObject.get("events");

		for (Object eventObj : jsonArray) {
			String event = eventObj.toString();
			events.add(event);
		}

		String[] res = new String[jsonArray.size()];
		int count = 0;

		if (type.equals("boot")) {
			for (String event : events) {
				Boot boot = BOOTTHREADS.get(event);
				Vector<HashMap<String, String>> clients = boot.getClients();

				if (boot.isFinished()) {
					res[count] = "Booting of " + event + " finished!";
					boot = null;
				} else if (boot.isFinishedWithErrors()) {
					res[count] = "Booting of " + event
							+ " finished with errors!";
					String[] clientErrors = new String[clients.size()];
					int statusCount = 0;
					for (HashMap<String, String> client : clients) {
						String ipAddress = client.get("ip");
						String errorText = boot.getError(client);
						clientErrors[statusCount] = "Error of the client with the ip "
								+ ipAddress + ": " + errorText;
						statusCount++;
					}
					String jsonClientErrors = JSONValue
							.toJSONString(clientErrors);
					res[count] += jsonClientErrors;
				} else {
					res[count] = "Booting of " + event + " not yet finished!";
					String[] clientStatus = new String[clients.size()];
					int statusCount = 0;
					for (HashMap<String, String> client : clients) {
						String ipAddress = client.get("ip");
						String statusText = boot.getStatusText(client);
						clientStatus[statusCount] = "Status of the client with the ip "
								+ ipAddress + ": " + statusText;
						statusCount++;
					}
					String jsonClientStatus = JSONValue
							.toJSONString(clientStatus);
					res[count] += jsonClientStatus;
				}

				count++;
			}
		} else if (type.equals("shutdown")) {
			for (String event : events) {
				Shutdown shutdown = SHUTDOWNTHREADS.get(event);

				if (shutdown.isFinished()) {
					res[count] = "Shutdown of " + event + " finished!";
				} else if (shutdown.isFinishedWithErrors()) {
					res[count] = "Shutdown of " + event
							+ " finished with errors!";
				} else {
					res[count] = "Shutdown of " + event + " not yet finished!";
				}

				count++;
			}
		}

		String jsonResult = JSONValue.toJSONString(res);

		byte[] warnings = new byte[0];
		byte[] exceptions = new byte[0];
		int numerator = 0;
		int denominator = 0;
		GearmanJobResult gjr = new GearmanJobResultImpl(this.jobHandle, true,
				jsonResult.getBytes(), warnings, exceptions, numerator,
				denominator);
		return gjr;
	}
}