plugins/hypervisors/simulator/src/main/java/com/cloud/resource/AgentResourceBase.java

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
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 java.util.concurrent.atomic.AtomicReference;

import javax.inject.Inject;
import javax.naming.ConfigurationException;

import org.apache.cloudstack.managed.context.ManagedContextRunnable;
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.agent.manager.SimulatorManagerImpl;
import com.cloud.host.Host;
import com.cloud.host.Host.Type;
import com.cloud.simulator.MockHost;
import com.cloud.utils.component.ComponentContext;

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;

    @Inject
    protected 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;
    }

    @Override
    public void setName(String name) {
        _name = name;
    }

    @Override
    public boolean configure(String name, Map<String, Object> params) throws ConfigurationException {
        hostGuid = (String)params.get("guid");

        _simMgr = ComponentContext.inject(SimulatorManagerImpl.class);
        _simMgr.configure(name, params);

        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 final Answer executeRequest(final Command cmd) {
        final AtomicReference<Answer> result = new AtomicReference<Answer>();
        new ManagedContextRunnable() {
            @Override
            protected void runInContext() {
                result.set(executeRequestInContext(cmd));
            }
        }.run();

        return result.get();
    }

    public Answer executeRequestInContext(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;
    }

    @Override
    public void setConfigParams(Map<String, Object> params) {
        // TODO Auto-generated method stub

    }

    @Override
    public Map<String, Object> getConfigParams() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public int getRunLevel() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    public void setRunLevel(int level) {
        // TODO Auto-generated method stub

    }
}