cloudstack/agent-simulator/src/com/cloud/resource/AgentResourceBase.java

/**
 *  Copyright (C) 2011 Cloud.com, Inc.  All rights reserved.
 */

package com.cloud.resource;

import java.io.File;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.net.URL;
import java.util.ArrayList;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import javax.naming.ConfigurationException;

import org.apache.log4j.Logger;

import com.cloud.agent.IAgentControl;
import com.cloud.agent.api.Answer;
import com.cloud.agent.api.Command;
import com.cloud.agent.api.PingCommand;
import com.cloud.agent.api.StartupCommand;
import com.cloud.agent.manager.MockAgentManager;
import com.cloud.agent.manager.MockStorageManager;
import com.cloud.agent.manager.MockVmManager;
import com.cloud.agent.manager.SimulatorManager;
import com.cloud.agent.manager.SimulatorManager.AgentType;
import com.cloud.host.Host;
import com.cloud.host.Host.Type;
import com.cloud.simulator.MockHost;
import com.cloud.utils.component.ComponentLocator;

public class AgentResourceBase implements ServerResource {
	private static final Logger s_logger = Logger
			.getLogger(AgentResourceBase.class);

	protected String _name;
	private List<String> _warnings = new LinkedList<String>();
	private List<String> _errors = new LinkedList<String>();
	
	private transient IAgentControl _agentControl;

	protected long _instanceId;

	private Type _type;

	private transient ComponentLocator _locator = null;
	protected transient SimulatorManager _simMgr;
	protected MockHost agentHost = null;
	protected boolean stopped = false;
	protected String hostGuid = null;
	

	public AgentResourceBase(long instanceId, AgentType agentType, SimulatorManager simMgr, String hostGuid) {
	    _instanceId = instanceId;	
		
		if(s_logger.isDebugEnabled()) {
			s_logger.info("New Routing host instantiated with guid:" + hostGuid);
		}
		
		if (agentType == AgentType.Routing) {
			_type = Host.Type.Routing;
		} else {
			_type = Host.Type.Storage;
		}
		
		this.hostGuid = hostGuid;
	}
	
	protected MockVmManager getVmMgr() {
	    return _simMgr.getVmMgr();
	}
	
	protected MockStorageManager getStorageMgr() {
	    return _simMgr.getStorageMgr();
	}
	
	protected MockAgentManager getAgentMgr() {
	    return _simMgr.getAgentMgr();
	}
	
	protected long getInstanceId() {
	    return _instanceId;
	}

	public AgentResourceBase() {
		if(s_logger.isDebugEnabled()) {
			s_logger.debug("Deserializing simulated agent on reconnect");
		}
	
	}

	@Override
	public String getName() {
		return _name;
	}

	public void setName(String name) {
		_name = name;
	}
	
	@Override
	public boolean configure(String name, Map<String, Object> params)
			throws ConfigurationException {
	    hostGuid = (String)params.get("guid");
	    _locator = ComponentLocator.getLocator("management-server");
        _simMgr = _locator.getManager(SimulatorManager.class);
        
	    agentHost = getAgentMgr().getHost(hostGuid);
	    return true;
	}
	

	private void reconnect(MockHost host) {
		if(s_logger.isDebugEnabled()) {
			s_logger.debug("Reconfiguring existing simulated host w/ name: " + host.getName() + " and guid: " + host.getGuid());
		}
		this.agentHost = host;
	}


	@Override
	public void disconnected() {
	    this.stopped = true;
	}

	protected void recordWarning(String msg, Throwable th) {
		String str = getLogStr(msg, th);
		synchronized (_warnings) {
			_warnings.add(str);
		}
	}

	protected void recordWarning(String msg) {
		recordWarning(msg, null);
	}

	protected List<String> getWarnings() {
		synchronized (this) {
			List<String> results = _warnings;
			_warnings = new ArrayList<String>();
			return results;
		}
	}

	protected List<String> getErrors() {
		synchronized (this) {
			List<String> result = _errors;
			_errors = new ArrayList<String>();
			return result;
		}
	}

	protected void recordError(String msg, Throwable th) {
		String str = getLogStr(msg, th);
		synchronized (_errors) {
			_errors.add(str);
		}
	}

	protected void recordError(String msg) {
		recordError(msg, null);
	}

	protected Answer createErrorAnswer(Command cmd, String msg, Throwable th) {
		StringWriter writer = new StringWriter();
		if (msg != null) {
			writer.append(msg);
		}
		writer.append("===>Stack<===");
		th.printStackTrace(new PrintWriter(writer));
		return new Answer(cmd, false, writer.toString());
	}

	protected String createErrorDetail(String msg, Throwable th) {
		StringWriter writer = new StringWriter();
		if (msg != null) {
			writer.append(msg);
		}
		writer.append("===>Stack<===");
		th.printStackTrace(new PrintWriter(writer));
		return writer.toString();
	}

	protected String getLogStr(String msg, Throwable th) {
		StringWriter writer = new StringWriter();
		writer.append(new Date().toString()).append(": ").append(msg);
		if (th != null) {
			writer.append("\n  Exception: ");
			th.printStackTrace(new PrintWriter(writer));
		}
		return writer.toString();
	}

	@Override
	public boolean start() {
		return true;
	}

	@Override
	public boolean stop() {
	    this.stopped = true;
		return true;
	}

	@Override
	public IAgentControl getAgentControl() {
		return _agentControl;
	}

	@Override
	public void setAgentControl(IAgentControl agentControl) {
		_agentControl = agentControl;
	}

	protected String findScript(String script) {
		s_logger.debug("Looking for " + script + " in the classpath");
		URL url = ClassLoader.getSystemResource(script);
		File file = null;
		if (url == null) {
			file = new File("./" + script);
			s_logger.debug("Looking for " + script + " in "
					+ file.getAbsolutePath());
			if (!file.exists()) {
				return null;
			}
		} else {
			file = new File(url.getFile());
		}
		return file.getAbsolutePath();
	}
	

	@Override
	public Answer executeRequest(Command cmd) {
		return null;
	}

	@Override
	public PingCommand getCurrentStatus(long id) {
		return null;
	}

	@Override
	public Type getType() {
		return _type;
	}

	public void setType(Host.Type _type) {
		this._type = _type;
	}
	
	@Override
	public StartupCommand[] initialize() {
		return null;
	}
	
	public SimulatorManager getSimulatorManager() {
		return _simMgr;
	}	
	
	public void setSimulatorManager(SimulatorManager simMgr) {
		_simMgr = simMgr;	
	}
	
	public boolean isStopped() {
	    return this.stopped;
	}
}