summaryrefslogtreecommitdiffstats
path: root/gearman/controllerWorker/ControllerWorker/Client.java
blob: c3fb95bbf6ae9915d7bb6777f1762e5c4ffb7cc0 (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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
package ControllerWorker;

import java.util.LinkedHashMap;
import java.util.Vector;

import org.gearman.client.GearmanClient;
import org.gearman.client.GearmanJob;
import org.gearman.client.GearmanJobImpl;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;

public class Client {
	private int id;
	private String ip;
	private String mac;
	private ClientState state;
	private String type;
	private String eventName;
	private Vector<ClientPingTime> times;
	private Vector<ClientJob> jobs;
	private String error;
	private Boolean finished;
	private int statusUpdateCount;
	private int scsavPID;

	public Client(JSONObject clientJsonObj, String type, String eventName) {
		this.statusUpdateCount = 0;
		this.type = type;
		this.eventName = eventName;
		id = Integer.parseInt(clientJsonObj.get("id").toString());
		ip = clientJsonObj.get("ip").toString();
		mac = clientJsonObj.get("mac").toString();
		finished = false;
		error = "";
		state = ClientState.CLIENT_UNKNOWN;
		times = new Vector<ClientPingTime>();
		jobs = new Vector<ClientJob>();
	}

	public int getScsavPID() {
		return scsavPID;
	}

	public void setScsavPID(int scsavPID) {
		this.scsavPID = scsavPID;
	}

	public GearmanJob getJob(int jobType) {
		for (ClientJob job : jobs) {
			if (job.getJobType() == jobType) {
				return job.getGearmanJob();
			}
		}
		return null;
	}

	public void addJob(int jobType, GearmanJob gearmanJob) {
		jobs.add(new ClientJob(jobType, gearmanJob));
	}

	public void removeJob(GearmanJob gearmanJob) {
		for (ClientJob job : jobs) {
			if (job.getGearmanJob().equals(gearmanJob)) {
				jobs.remove(job);
				break;
			}
		}
	}

	public long getPingTime(int timeType) {
		for (ClientPingTime time : times) {
			if (time.getType() == timeType) {
				return time.getTime();
			}
		}
		return 0;
	}

	public void addPingTime(int timeType, long time) {
		times.add(new ClientPingTime(timeType, time));
	}

	public int getId() {
		return id;
	}

	public String getStateText() {
		return state.toString();
	}

	public String getError() {
		return error;
	}

	public String getIp() {
		return ip;
	}

	public String getMac() {
		return mac;
	}

	public ClientState getState() {
		return state;
	}

	public void setState(ClientState state_, GearmanClient gearmanClient) {
		state = state_;
		publishStateChange(gearmanClient);
	}

	public void setState(ClientState state_) {
		state = state_;
	}

	public void setError(String error) {
		this.error = error;
	}

	public Boolean isFinished() {
		return finished;
	}

	public void finish() {
		this.finished = true;
	}

	@Override
	public String toString() {
		return "Client [error=" + error + ", finished=" + finished + ", id="
				+ id + ", ip=" + ip + ", jobs=" + jobs + ", mac=" + mac
				+ ", state=" + state + ", times=" + times + "]";
	}

	private void publishStateChange(GearmanClient gearmanClient) {
		statusUpdateCount++;
		LinkedHashMap<String, Object> jsonData = new LinkedHashMap<String, Object>();
		if (type.equals("boot")) {
			jsonData.put("type", "updateClientStateBoot");
		} else if (type.equals("shutdown")) {
			jsonData.put("type", "updateClientStateShutdown");
		} else {
			return;
		}
		jsonData.put("eventName", eventName);
		jsonData.put("clientID", id);
		jsonData.put("error", error);
		jsonData.put("newState", state.name());
		String dataString = JSONValue.toJSONString(jsonData);
		GearmanJob job = GearmanJobImpl.createBackgroundJob("status", dataString
				.getBytes(), "status" + eventName + id + statusUpdateCount);
		gearmanClient.submit(job);
	}
}