DefaultNotifier.java

package com.intrbiz.bergamot.notification;

import java.io.File;
import java.io.FileInputStream;

import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.apache.log4j.PropertyConfigurator;

import com.intrbiz.Util;
import com.intrbiz.accounting.AccountingManager;
import com.intrbiz.bergamot.accounting.BergamotAccountingQueueConsumer;
import com.intrbiz.bergamot.accounting.consumer.BergamotLoggingConsumer;
import com.intrbiz.bergamot.config.NotifierCfg;
import com.intrbiz.bergamot.health.HealthAgent;
import com.intrbiz.bergamot.health.HealthTracker;
import com.intrbiz.configuration.Configuration;
import com.intrbiz.queue.QueueManager;
import com.intrbiz.queue.rabbit.RabbitPool;

public class DefaultNotifier extends AbstractNotifier
{
    protected final Class<? extends NotifierCfg> configurationClass;

    protected final String defaultConfigFile;
    
    protected final String notifierName;

    public DefaultNotifier(Class<? extends NotifierCfg> configurationClass, String defaultConfigFile, String notifierName)
    {
        super();
        this.configurationClass = configurationClass;
        this.defaultConfigFile = defaultConfigFile;
        this.notifierName = notifierName;
    }

    @Override
    protected String getNotifierName()
    {
        return this.notifierName;
    }
    
    @Override
    public String getDaemonName()
    {
        return "bergamot-notifier-" + this.notifierName.toLowerCase();
    }

    protected void configureLogging() throws Exception
    {
        String logging = System.getProperty("bergamot.logging", "console");
        if ("console".equals(logging))
        {
            // configure logging to terminal
            BasicConfigurator.configure();
            Logger.getRootLogger().setLevel(Level.toLevel(System.getProperty("bergamot.logging.level", "trace").toUpperCase()));
        }
        else
        {
            // configure from file
            PropertyConfigurator.configure(new File(logging).getAbsolutePath());
        }
    }
    
    /**
     * Search for the configuration file
     */
    protected File getConfigurationFile()
    {
        return new File(Util.coalesceEmpty(System.getProperty("bergamot.config"), System.getenv("bergamot_config"), System.getenv("BERGAMOT_CONFIG"), this.defaultConfigFile));
    }

    protected NotifierCfg loadConfiguration() throws Exception
    {
        NotifierCfg config = null;
        // try the config file?
        File configFile = this.getConfigurationFile();
        if (configFile.exists())
        {
            Logger.getLogger(Notifier.class).info("Reading configuration file " + configFile.getAbsolutePath());
            config = Configuration.read(this.configurationClass, new FileInputStream(configFile));
        }
        else
        {
            config = this.configurationClass.newInstance();
        }
        config.applyDefaults();
        return config;
    }

    @Override
    public void start() throws Exception
    {
        // setup logging
        this.configureLogging();
        Logger logger = Logger.getLogger(Notifier.class);
        // configure accounting
        // load the config
        NotifierCfg config = this.loadConfiguration();
        logger.debug("Bergamot notifier, using configuration:\r\n" + config.toString());
        // setup the queue broker
        QueueManager.getInstance().registerDefaultBroker(new RabbitPool(config.getBroker().getUrl(), config.getBroker().getUsername(), config.getBroker().getPassword()));
        // setup accounting
        AccountingManager.getInstance().registerConsumer("logger", new BergamotLoggingConsumer());
        AccountingManager.getInstance().registerConsumer("queue", new BergamotAccountingQueueConsumer());
        AccountingManager.getInstance().bindRootConsumer("logger");
        AccountingManager.getInstance().bindRootConsumer("queue");
        // configure the worker
        this.configure(config);
        // go go go
        logger.info("Bergamot notifier starting.");
        // start the health agent
        HealthAgent.getInstance().init("notifier", this.getDaemonName());
        HealthTracker.getInstance().init();
        // start the notifier
        super.start();
    }
}