com.fasterxml.clustermate.dw.DWConfigBase Maven / Gradle / Ivy

Go to download
package com.fasterxml.clustermate.dw;

import io.dropwizard.Configuration;
import io.dropwizard.jetty.HttpConnectorFactory;
import io.dropwizard.jetty.RequestLogFactory;
import io.dropwizard.logging.AppenderFactory;
import io.dropwizard.server.AbstractServerFactory;
import io.dropwizard.server.DefaultServerFactory;
import io.dropwizard.server.ServerFactory;
import io.dropwizard.server.SimpleServerFactory;
import io.dropwizard.util.Duration;

import org.skife.config.TimeSpan;

import ch.qos.logback.classic.Level;

import com.fasterxml.clustermate.service.cfg.ServiceConfig;
import com.google.common.collect.ImmutableList;

public abstract class DWConfigBase<
  SCONFIG extends ServiceConfig,
  THIS extends DWConfigBase
>
    extends Configuration
    implements Cloneable
{
    protected String _appContext = "/";

    public DWConfigBase() {
        _ensureDefaults();
    }

    /*
    /**********************************************************************
    /* Overrides, to ensure invariants
    /**********************************************************************
     */

    // Overridden to ensure we have proper defaults. May become problematic
    // if users actually do want to change defaults; but let's worry about
    // that if this becomes issue.
    @Override
    public void setServerFactory(ServerFactory factory) {
        super.setServerFactory(factory);
        _ensureDefaults();
    }
    
    private void _ensureDefaults() {
        SimpleServerFactory simpleSF = simpleServerFactory();
        if (simpleSF != null) {
            simpleSF.setApplicationContextPath(_appContext);
        }
        // and that gzip is not enabled by accident
        overrideGZIPEnabled(false);
    }
    
    /*
    /**********************************************************************
    /* Additional/abstract accessors
    /**********************************************************************
     */

    public abstract SCONFIG getServiceConfig();

    public int getApplicationPort() {
        AbstractServerFactory sf = serverFactory();
        if (sf instanceof SimpleServerFactory) {
            SimpleServerFactory ssf = (SimpleServerFactory) sf;
            return ((HttpConnectorFactory)ssf.getConnector()).getPort();
        }
        if (sf instanceof DefaultServerFactory) {
            DefaultServerFactory dsf = (DefaultServerFactory) sf;
            return ((HttpConnectorFactory)dsf.getApplicationConnectors().get(0)).getPort();
        }
        throw new IllegalStateException("Unrecognized ServerFactory: "+sf.getClass().getName());
    }

    public int getAdminPort() {
        AbstractServerFactory sf = serverFactory();
        if (sf instanceof SimpleServerFactory) {
            SimpleServerFactory ssf = (SimpleServerFactory) sf;
            return ((HttpConnectorFactory)ssf.getConnector()).getPort();
        }
        if (sf instanceof DefaultServerFactory) {
            DefaultServerFactory dsf = (DefaultServerFactory) sf;
            return ((HttpConnectorFactory)dsf.getAdminConnectors().get(0)).getPort();
        }
        throw new IllegalStateException("Unrecognized ServerFactory: "+sf.getClass().getName());
    }    

    /*
    /**********************************************************************
    /* Copying
    /**********************************************************************
     */

    @SuppressWarnings("unchecked")
    @Override
    public THIS clone() {
        try {
            return (THIS) super.clone();
        } catch (CloneNotSupportedException e) {
            throw new IllegalStateException("Failed to clone "+getClass().getName()+", WTH?");
        }
    }

    /*
    /**********************************************************************
    /* Additional mutators; needed to work around DW strict access
    /**********************************************************************
     */

    @SuppressWarnings("unchecked")
    public THIS overrideHttpPort(int p) {
        AbstractServerFactory sf = serverFactory();
        if (sf instanceof SimpleServerFactory) {
            SimpleServerFactory ssf = (SimpleServerFactory) sf;
            ((HttpConnectorFactory)ssf.getConnector()).setPort(p);
        } else if (sf instanceof DefaultServerFactory) {
            DefaultServerFactory dsf = (DefaultServerFactory) sf;
            ((HttpConnectorFactory)dsf.getApplicationConnectors().get(0)).setPort(p);
        } else {
            throw new IllegalStateException("Unrecognized ServerFactory: "+sf.getClass().getName());
        }
        return (THIS) this;
    }

    @SuppressWarnings("unchecked")
    public THIS overrideAdminPort(int p) {
        AbstractServerFactory sf = serverFactory();
        if (sf instanceof SimpleServerFactory) {
            // no admin port; just ignore
        } else if (sf instanceof DefaultServerFactory) {
            DefaultServerFactory dsf = (DefaultServerFactory) sf;
            ((HttpConnectorFactory)dsf.getAdminConnectors().get(0)).setPort(p);
        } else {
            throw new IllegalStateException("Unrecognized ServerFactory: "+sf.getClass().getName());
        }
        return (THIS) this;
    }

    @SuppressWarnings("unchecked")
    public THIS overrideGZIPEnabled(boolean state) {
        serverFactory().getGzipFilterFactory().setEnabled(state);
        return (THIS) this;
    }
    
    @SuppressWarnings("unchecked")
    public THIS overrideLogLevel(Level minLevel) {
        // !!! TODO !!!
        //config.getLoggingConfiguration().setLevel(Level.WARN);
        return (THIS) this;
    }
    
    @SuppressWarnings("unchecked")
    public THIS overrideShutdownGracePeriod(Duration d) {
        serverFactory().setShutdownGracePeriod(d);
        return (THIS) this;
    }

    @SuppressWarnings("unchecked")
    public THIS overrideSyncGracePeriod(TimeSpan t) {
        getServiceConfig().cfgSyncGracePeriod = t;
        return (THIS) this;
    }

    @SuppressWarnings("unchecked")
    public THIS overrideApplicationContextPath(String path) {
        _appContext = path;
        _overrideApplicationContextPath(path);
        return (THIS) this;
    }

    /**
     * Method called to suppress all request logging; usually during testing.
     */
    @SuppressWarnings("unchecked")
    public THIS disableRequestLog() {
        RequestLogFactory reqLog = serverFactory().getRequestLogFactory();
        reqLog.setAppenders(ImmutableList.of());        
        return (THIS) this;
    }

    protected AbstractServerFactory serverFactory() {
        return (AbstractServerFactory) getServerFactory();
    }

    protected SimpleServerFactory simpleServerFactory() {
        AbstractServerFactory sf = serverFactory();
        return (sf instanceof SimpleServerFactory) ? (SimpleServerFactory) sf : null;
    }

    private void _overrideApplicationContextPath(String path) {
        SimpleServerFactory simpleSF = simpleServerFactory();
        if (simpleSF != null) {
            simpleSF.setApplicationContextPath(path);
        }
    }
}