com.ge.snowizard.service.SnowizardService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snowizard-service Show documentation
There is a newer version: 1.3.0
package com.ge.snowizard.service;

import com.ge.snowizard.core.IdWorker;
import com.ge.snowizard.service.core.CorsHeadersFilter;
import com.ge.snowizard.service.core.JacksonProtobufProvider;
import com.ge.snowizard.service.core.TimedResourceMethodDispatchAdapter;
import com.ge.snowizard.service.resources.IdResource;
import com.ge.snowizard.service.resources.PingResource;
import com.ge.snowizard.service.resources.SnowizardResource;
import com.yammer.dropwizard.Service;
import com.yammer.dropwizard.assets.AssetsBundle;
import com.yammer.dropwizard.config.Bootstrap;
import com.yammer.dropwizard.config.Environment;
import com.yammer.metrics.Metrics;
import com.yammer.metrics.core.Gauge;

public class SnowizardService extends Service {
    public static void main(final String[] args) throws Exception {
        new SnowizardService().run(args);
    }

    @Override
    public void initialize(final Bootstrap bootstrap) {
        bootstrap.setName("snowizard");
        bootstrap.addBundle(new AssetsBundle("/apidocs", "/apidocs",
                "index.html"));
    }

    @Override
    public void run(final SnowizardConfiguration config,
            final Environment environment) throws Exception {

        environment.addProvider(new JacksonProtobufProvider());
        environment.addProvider(new TimedResourceMethodDispatchAdapter());
        if (config.isCORSEnabled()) {
            environment.addFilter(new CorsHeadersFilter(), "/*");
        }

        final IdWorker worker = new IdWorker(config.getWorkerId(),
                config.getDatacenterId(), config.validateUserAgent());

        Metrics.newGauge(SnowizardService.class, "worker_id",
                new Gauge() {
                    @Override
                    public Integer value() {
                        return config.getWorkerId();
                    }
                });

        Metrics.newGauge(SnowizardService.class, "datacenter_id",
                new Gauge() {
                    @Override
                    public Integer value() {
                        return config.getDatacenterId();
                    }
                });

        environment.addResource(new IdResource(worker));
        environment.addResource(new PingResource());
        environment.addResource(new SnowizardResource());
    }
}