Java Environment.addTask方法代码示例

import com.yammer.dropwizard.config.Environment; //导入方法依赖的package包/类
@Override
public void run(LocateApiConfiguration configuration, Environment environment) throws Exception {

    /**
     * Mongo set up
     */
    MongoClient mongoClient = configureMongoClient(environment, configuration.getMongoConfiguration());

    DB locateDb = setUpDb(configuration.getMongoConfiguration().getLocateDatabase(), configuration.getMongoConfiguration(), mongoClient);
    DB credentialsDb = setUpDb(configuration.getMongoConfiguration().getCredentialsDatabase(), configuration.getMongoConfiguration(), mongoClient);

    /**
     * Dao layer
     */
    final AuthorizationTokenDao authorizationTokenDao = configureAuthorizationTokenDao(credentialsDb);
    final UsageDao usageDao = configureRateMeterDao(credentialsDb);
    final AddressDao addressDao = configureAddressDao(locateDb);
    final PostcodeToAuthorityDao postcodeToAuthorityDao = configurePostcodeToAuthorityDao(locateDb);

    /**
     * Enable cors support
     */
    environment.addFilter(CrossOriginFilter.class, "*")
            .setInitParam("allowedOrigins", configuration.getAllowedOrigins())
            .setInitParam("allowedHeaders", "X-Requested-With,Content-Type,Accept,Origin,Authorization")
            .setInitParam("allowedMethods", "OPTIONS,GET");

    /**
     * Resources
     */
    environment.addResource(new AddressResource(addressDao, configuration));
    environment.addResource(new PostcodeToAuthorityResource(postcodeToAuthorityDao));

    /**
     * Healthchecks
     */
    environment.addHealthCheck(new MongoHealthCheck(mongoClient));

    /**
     * Exception mapper
     */
    environment.addProvider(new LocateExceptionMapper());

    /**
     * Authentication
     */
    environment.addProvider(new BearerTokenAuthProvider(configuration, usageDao, new BearerTokenAuthenticator(authorizationTokenDao)));

    /**
     * Better exception mappings
     */
    removeDefaultExceptionMappers(environment);

    /**
     * Date serialisation
     */
    environment.getObjectMapperFactory().setDateFormat(internalDateFormatter);

    /**
     * Tasks
     */
    environment.addTask(new MongoIndexTask(authorizationTokenDao, usageDao));

}
 

import com.yammer.dropwizard.config.Environment; //导入方法依赖的package包/类
@Override
public void run(Environment environment) {

    environment.addTask(new ExampleNagiosCheckTask());
}
 

import com.yammer.dropwizard.config.Environment; //导入方法依赖的package包/类
/**
 * {@inheritDoc}
 *
 * @throws IOException when instance in configuration can not be opened and closed.
 */
@Override
public final void run(final KijiRESTConfiguration configuration, final Environment environment)
    throws IOException {
  final KijiURI clusterURI = KijiURI.newBuilder(configuration.getClusterURI()).build();
  final ManagedKijiClient managedKijiClient = new ManagedKijiClient(configuration);
  environment.manage(managedKijiClient);

  // Setup the health checker for the KijiClient
  environment.addHealthCheck(new KijiClientHealthCheck(managedKijiClient));

  // Remove all built-in Dropwizard ExceptionHandler.
  // Always depend on custom ones.
  // Inspired by Jeremy Whitlock's suggestion on thoughtspark.org.
  Set<Object> jerseyResources = environment.getJerseyResourceConfig().getSingletons();
  Iterator<Object> jerseyResourcesIterator = jerseyResources.iterator();
  while (jerseyResourcesIterator.hasNext()) {
    Object jerseyResource = jerseyResourcesIterator.next();
    if (jerseyResource instanceof ExceptionMapper
        && jerseyResource.getClass().getName().startsWith("com.yammer.dropwizard.jersey")) {
      jerseyResourcesIterator.remove();
    }
  }

  // Load admin task to manually refresh instances.
  environment.addTask(new RefreshInstancesTask(managedKijiClient));
  // Load admin task to manually close instances and tables.
  environment.addTask(new CloseTask(managedKijiClient));
  // Load admin task to manually shutdown the system.
  environment.addTask(new ShutdownTask(managedKijiClient, configuration));

  // Adds custom serializers.
  registerSerializers(environment.getObjectMapperFactory());

  // Adds exception mappers to print better exception messages to the client than what
  // Dropwizard does by default.
  environment.addProvider(new GeneralExceptionMapper());

  // Load resources.
  for (KijiRestPlugin plugin : Lookups.get(KijiRestPlugin.class)) {
    LOG.info("Loading plugin {}", plugin.getClass());
    plugin.install(managedKijiClient, configuration, environment);
  }

  // Allow global CORS filter. CORS off by default.
  if (configuration.getCORS()) {
    environment.addFilter(
        CrossOriginFilter.class,
        configuration.getHttpConfiguration().getRootPath());
    LOG.info("Global cross-origin resource sharing is allowed.");
  }
}
 

import com.yammer.dropwizard.config.Environment; //导入方法依赖的package包/类
@Override
public void run(TestConfiguration configuration, Environment environment) throws Exception {

    for (NagiosCheckTask checkTask : configuration.getCheckTasks()){

        environment.addTask(checkTask);
    }

    for (HealthCheck healthCheck : configuration.getWrappedCheckTasks()){

        environment.addHealthCheck(healthCheck);
    }

    environment.addResource(FakeResource.class);
}