com.hubspot.singularity.data.SingularityDataModule Maven / Gradle / Ivy

Go to download
package com.hubspot.singularity.data;

import com.codahale.metrics.MetricRegistry;
import com.google.inject.AbstractModule;
import com.google.inject.Provides;
import com.google.inject.Scopes;
import com.google.inject.Singleton;
import com.hubspot.singularity.SingularityDeploy;
import com.hubspot.singularity.SingularityTask;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.usage.UsageManager;
import com.hubspot.singularity.helpers.RequestHelper;

public class SingularityDataModule extends AbstractModule {
  private final SingularityConfiguration configuration;

  public SingularityDataModule(final SingularityConfiguration configuration) {
    this.configuration = configuration;
  }

  @Override
  protected void configure() {
    bind(RequestHelper.class).in(Scopes.SINGLETON);

    bind(MetadataManager.class).in(Scopes.SINGLETON);
    bind(StateManager.class).in(Scopes.SINGLETON);
    bind(TaskManager.class).in(Scopes.SINGLETON);
    bind(DeployManager.class).in(Scopes.SINGLETON);
    bind(RackManager.class).in(Scopes.SINGLETON);
    bind(RequestManager.class).in(Scopes.SINGLETON);
    bind(AgentManager.class).in(Scopes.SINGLETON);
    bind(InactiveAgentManager.class).in(Scopes.SINGLETON);
    bind(TaskRequestManager.class).in(Scopes.SINGLETON);
    bind(SandboxManager.class).in(Scopes.SINGLETON);
    bind(SingularityValidator.class).in(Scopes.SINGLETON);
    bind(UserManager.class).in(Scopes.SINGLETON);
    bind(UsageManager.class).in(Scopes.SINGLETON);
    bind(ShuffleConfigurationManager.class).in(Scopes.SINGLETON);

    bind(WebhookManager.class).in(Scopes.SINGLETON);

    bind(NotificationsManager.class).in(Scopes.SINGLETON);

    bind(SingularityWebCache.class).in(Scopes.SINGLETON);

    bind(ExecutorIdGenerator.class).asEagerSingleton();
    bind(WebhookManager.class).in(Scopes.SINGLETON);
    bind(DisasterManager.class).in(Scopes.SINGLETON);
    bind(PriorityManager.class).in(Scopes.SINGLETON);
    bind(RequestGroupManager.class).in(Scopes.SINGLETON);
    bind(AuthTokenManager.class).in(Scopes.SINGLETON);
  }

  @Provides
  @Singleton
  public ZkCache taskCache(
    SingularityConfiguration configuration,
    MetricRegistry registry
  ) {
    return new ZkCache<>(
      configuration.getCacheTasksMaxSize(),
      configuration.getCacheTasksInitialSize(),
      configuration.getCacheTasksForMillis(),
      registry,
      "tasks"
    );
  }

  @Provides
  @Singleton
  public ZkCache deployCache(
    SingularityConfiguration configuration,
    MetricRegistry registry
  ) {
    return new ZkCache<>(
      configuration.getCacheDeploysMaxSize(),
      configuration.getCacheDeploysInitialSize(),
      configuration.getCacheDeploysForMillis(),
      registry,
      "deploys"
    );
  }
}