summaryrefslogblamecommitdiffstats
path: root/gearman/controllerWorker/ControllerWorker/StatusWorker.java
blob: 59f37d777cca5ba1515935e943c010590250116a (plain) (tree)
1
2
3
4
5
6
7
8
9

                         
                         
                               

                        

                                               
                                  
                                                  
                                 
                                  
                                 

                                                           

                                                                                           
 






                                                   

                                                                          


                                                                           



                                                           
                                                                                        
                              


                                                                   
                                                                     
                                                                                                    
                                                                

                                                                                         

                                                                                 
                                                                                                                                 

                                                                                                
                                                                                           
                                                                                                 


                                                                                                                        


                                                                                            



                                                                                                          
                                                
                                                                                                                                 

                                                                                                
                                                                                           
                                                                                                       


                                                                                                                         


                                                                                            



                                                                                               
                                        


                                                                                                                              
                                 



                                                     






































                                                                                                                                 
                                         



                                                                                                                                  
                                 
                                        
                         
                 
                                                                




                                                                                     

                                                                                       


                           
package ControllerWorker;

import java.util.HashMap;
import java.util.LinkedHashMap;
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);
		}
		LinkedHashMap<String, String> res = new LinkedHashMap<String, String>();
		int count = 0;
		if (type.equals("boot")) {
			for (String event : events) {
				Boot boot = BOOTTHREADS.get(event);
				if (BOOTTHREADS.containsKey(event)) {
					Vector<HashMap<String, String>> clients = boot.getClients();
					if (boot.isFinished()) {
						res.put(event, "Booting of " + event
								+ " has been finished!");
						boot = null;
					} else if (boot.isFinishedWithErrors()) {
						LinkedHashMap<String, String> clientErrors = new LinkedHashMap<String, String>();
						for (HashMap<String, String> client : clients) {
							String ipAddress = client.get("ip");
							String clientID = client.get("id");
							String errorText = boot.getError(client);
							clientErrors.put("Client " + clientID,
									"Error of the client with the ip "
											+ ipAddress + ": " + errorText);
						}
						String jsonClientErrors = JSONValue
								.toJSONString(clientErrors);
						res.put(event, "Booting of " + event
								+ " has not been finished, due to errors!"
								+ jsonClientErrors);
						boot = null;
					} else {
						LinkedHashMap<String, String> clientStatus = new LinkedHashMap<String, String>();
						for (HashMap<String, String> client : clients) {
							String ipAddress = client.get("ip");
							String clientID = client.get("id");
							String statusText = boot.getStatusText(client);
							clientStatus.put("Client " + clientID,
									"Status of the client with the ip "
											+ ipAddress + ": " + statusText);
						}
						String jsonClientStatus = JSONValue
								.toJSONString(clientStatus);
						res.put(event, "Booting of " + event
								+ " has not yet been finished!"
								+ jsonClientStatus);
					}
				} else {
					res
							.put(event,
									"The Boot process for the event has not been found.");
				}
				count++;
			}
		} else if (type.equals("shutdown")) {
			for (String event : events) {
				if (SHUTDOWNTHREADS.containsKey(event)) {
					Shutdown shutdown = SHUTDOWNTHREADS.get(event);
					Vector<HashMap<String, String>> clients = shutdown
							.getClients();
					if (shutdown.isFinished()) {
						res.put(event, "Shutdown of " + event
								+ " has been finished!");
						shutdown = null;
					} else if (shutdown.isFinishedWithErrors()) {
						LinkedHashMap<String, String> clientErrors = new LinkedHashMap<String, String>();
						for (HashMap<String, String> client : clients) {
							String ipAddress = client.get("ip");
							String clientID = client.get("id");
							String errorText = shutdown.getError(client);
							clientErrors.put("Client " + clientID,
									"Error of the client with the ip "
											+ ipAddress + ": " + errorText);
						}
						String jsonClientErrors = JSONValue
								.toJSONString(clientErrors);
						res.put(event, "Shutdown of " + event
								+ " has not been finished, due to errors!"
								+ jsonClientErrors);
						shutdown = null;
					} else {
						LinkedHashMap<String, String> clientStatus = new LinkedHashMap<String, String>();
						for (HashMap<String, String> client : clients) {
							String ipAddress = client.get("ip");
							String clientID = client.get("id");
							String statusText = shutdown.getStatusText(client);
							clientStatus.put("Client " + clientID,
									"Status of the client with the ip "
											+ ipAddress + ": " + statusText);
						}
						String jsonClientStatus = JSONValue
								.toJSONString(clientStatus);
						res.put(event, "Shutdown of " + event
								+ " has not yet been finished!"
								+ jsonClientStatus);
					}
				} else {
					res
							.put(event,
									"The Shutdown process for the event has not been found.");
				}
				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;
	}
}