UptimeExecutor.java

package com.intrbiz.bergamot.worker.engine.agent;

import java.text.DecimalFormat;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

import org.apache.log4j.Logger;

import com.intrbiz.bergamot.agent.server.BergamotAgentServerHandler;
import com.intrbiz.bergamot.model.message.agent.check.CheckUptime;
import com.intrbiz.bergamot.model.message.agent.stat.UptimeStat;
import com.intrbiz.bergamot.model.message.check.ExecuteCheck;
import com.intrbiz.bergamot.model.message.result.ActiveResultMO;
import com.intrbiz.bergamot.worker.engine.AbstractExecutor;
import com.intrbiz.gerald.polyakov.gauge.DoubleGaugeReading;

/**
 * Check the uptime of a Bergamot Agent
 */
public class UptimeExecutor extends AbstractExecutor<AgentEngine>
{
    public static final String NAME = "uptime";
    
    private static final DecimalFormat DFMT = new DecimalFormat("0.00");
    
    private Logger logger = Logger.getLogger(UptimeExecutor.class);

    public UptimeExecutor()
    {
        super();
    }

    /**
     * Only execute Checks where the engine == "agent"
     */
    @Override
    public boolean accept(ExecuteCheck task)
    {
        return AgentEngine.NAME.equals(task.getEngine()) && NAME.equals(task.getExecutor());
    }

    @Override
    public void execute(ExecuteCheck executeCheck)
    {
        if (logger.isTraceEnabled()) logger.trace("Checking Bergamot Agent Uptime");
        try
        {
            // check the host presence
            UUID agentId = executeCheck.getAgentId();
            if (agentId == null) throw new RuntimeException("No agent id was given");
            // lookup the agent
            BergamotAgentServerHandler agent = this.getEngine().getAgentServer().getRegisteredAgent(agentId);
            if (agent != null)
            {
                // get the Uptime stats
                long sent = System.nanoTime();
                agent.sendMessageToAgent(new CheckUptime(), (response) -> {
                    double runtime = ((double)(System.nanoTime() - sent)) / 1000_000D;
                    UptimeStat stat = (UptimeStat) response;
                    if (logger.isTraceEnabled()) logger.trace("Got Uptime in " + runtime + "ms: " + stat);
                    // thresholds
                    long critical = executeCheck.getLongParameter("critical", TimeUnit.MINUTES.toSeconds(5));
                    long warning  = executeCheck.getLongParameter("warning",  TimeUnit.MINUTES.toSeconds(10));
                    // the result
                    ActiveResultMO result = new ActiveResultMO().fromCheck(executeCheck);
                    // apply the check
                    if (stat.getUptime() <= warning)
                    {
                        result.warning("Uptime: " + formatUptime(stat.getUptime()));
                    }
                    else if (stat.getUptime() <= critical)
                    {
                        result.critical("Uptime: " + formatUptime(stat.getUptime()));
                    }
                    else
                    {
                        result.ok("Up " + formatUptime(stat.getUptime()));
                    }
                    // submit
                    result.runtime(runtime);
                    this.publishActiveResult(executeCheck, result);
                    // readings
                    this.publishReading(executeCheck, new DoubleGaugeReading("uptime", "s", stat.getUptime()));
                });
            }
            else
            {
                // raise an error
                this.publishActiveResult(executeCheck, new ActiveResultMO().fromCheck(executeCheck).disconnected("Bergamot Agent disconnected"));
            }
        }
        catch (Exception e)
        {
            this.publishActiveResult(executeCheck, new ActiveResultMO().fromCheck(executeCheck).error(e));
        }
    }
    
    private String formatUptime(double uptimeDouble)
    {
        long uptime  = (long) uptimeDouble;
        long seconds = uptime % 60L;
        long minutes = (uptime / 60L) % 60L;
        long hours   = (uptime / 3600L) % 24L;
        long days    = uptime / 86400L;
        //
        return days + " day" + (days > 1 ? "s" : "") + ", " + hours + ":" + minutes + ":" + seconds + " (" + DFMT.format(uptimeDouble) + ")"; 
    }
}