How to use

co.cask.cdap.data.runtime.DataSetsModules

constructor

private void addOnPremiseModules(List<Module> modules) {
 String instanceId = systemArgs.getOption(ProgramOptionConstants.INSTANCE_ID);
 modules.add(new LocationRuntimeModule().getDistributedModules());
 modules.add(new KafkaClientModule());
 modules.add(new LoggingModules().getDistributedModules());
 modules.add(new DataFabricModules(generateClientId(programRunId, instanceId)).getDistributedModules());
 modules.add(new DataSetsModules().getDistributedModules());
 modules.add(new NamespaceClientRuntimeModule().getDistributedModules());
 modules.add(new DistributedProgramStreamModule());
}

private void addOnPremiseModules(List<Module> modules) {
 String instanceId = systemArgs.getOption(ProgramOptionConstants.INSTANCE_ID);
 modules.add(new DFSLocationModule());
 modules.add(new KafkaClientModule());
 modules.add(new KafkaLogAppenderModule());
 modules.add(new DataFabricModules(generateClientId(programRunId, instanceId)).getDistributedModules());
 modules.add(new DataSetsModules().getDistributedModules());
 modules.add(new NamespaceQueryAdminModule());
 modules.add(new AbstractModule() {
  @Override
  protected void configure() {
   // bind explore client to ProgramDiscoveryExploreClient which is aware of the programId
   bind(ExploreClient.class).to(ProgramDiscoveryExploreClient.class).in(Scopes.SINGLETON);
  }
 });
}

@VisibleForTesting
public static Injector createInjector(CConfiguration cConf, Configuration hConf) {
 return Guice.createInjector(
  new ConfigModule(cConf, hConf),
  new IOModule(),
  new ZKClientModule(),
  new ZKDiscoveryModule(),
  new KafkaClientModule(),
  new DFSLocationModule(),
  new DataFabricModules(HBaseTableExporter.class.getName()).getDistributedModules(),
  new DataSetsModules().getDistributedModules(),
  new MessagingClientModule(),
  new MetricsClientRuntimeModule().getDistributedModules(),
  new KafkaLogAppenderModule(),
  new ExploreClientModule(),
  new AuthorizationModule(),
  new AuthorizationEnforcementModule().getMasterModule(),
  new AuthenticationContextModules().getMasterModule(),
  new NamespaceQueryAdminModule(),
  new SecureStoreServerModule(),
  new AbstractModule() {
   @Override
   protected void configure() {
    bind(UGIProvider.class).to(RemoteUGIProvider.class);
    bind(OwnerAdmin.class).to(DefaultOwnerAdmin.class);
   }
  }
 );
}

@VisibleForTesting
static Injector createInjector(CConfiguration cConf, Configuration hConf) {
 return Guice.createInjector(
  new ConfigModule(cConf, hConf),
  new ZKClientModule(),
  new DFSLocationModule(),
  new NamespaceQueryAdminModule(),
  new ZKDiscoveryModule(),
  new DataFabricModules("cdap.explore.ContextManager").getDistributedModules(),
  new DataSetsModules().getDistributedModules(),
  new ExploreClientModule(),
  new KafkaClientModule(),
  new AuditModule().getDistributedModules(),
  new AuthorizationEnforcementModule().getDistributedModules(),
  new SecureStoreClientModule(),
  new AuthenticationContextModules().getMasterModule(),
  new MessagingClientModule(),
  new AbstractModule() {
   @Override
   protected void configure() {
    bind(UGIProvider.class).to(RemoteUGIProvider.class).in(Scopes.SINGLETON);
    bind(MetricsCollectionService.class).to(NoOpMetricsCollectionService.class).in(Scopes.SINGLETON);
    // bind PrivilegesManager to a remote implementation, so it does not need to instantiate the authorizer
    bind(PrivilegesManager.class).to(RemotePrivilegesManager.class);
    bind(OwnerAdmin.class).to(DefaultOwnerAdmin.class);
   }
  }
 );
}

 @VisibleForTesting
 static Injector createGuiceInjector(CConfiguration cConf, Configuration hConf, TwillContext twillContext) {
  String txClientId = String.format("cdap.service.%s.%d", Constants.Service.LOGSAVER, twillContext.getInstanceId());
  return Guice.createInjector(
   new ConfigModule(cConf, hConf),
   new IOModule(),
   new ZKClientModule(),
   new ZKDiscoveryModule(),
   new KafkaClientModule(),
   new MetricsClientRuntimeModule().getDistributedModules(),
   new DFSLocationModule(),
   new NamespaceQueryAdminModule(),
   new DataFabricModules(txClientId).getDistributedModules(),
   new DataSetsModules().getDistributedModules(),
   new DistributedLogFrameworkModule(twillContext),
   new KafkaLogAppenderModule(),
   new AuditModule().getDistributedModules(),
   new AuthorizationEnforcementModule().getDistributedModules(),
   new AuthenticationContextModules().getMasterModule(),
   new MessagingClientModule(),
   new AbstractModule() {
    @Override
    protected void configure() {
     bind(OwnerAdmin.class).to(DefaultOwnerAdmin.class);
     bind(UGIProvider.class).to(RemoteUGIProvider.class).in(Scopes.SINGLETON);
    }
   }
  );
 }
}

@BeforeClass
public static void init() throws Exception {
 cConf = CConfiguration.create();
 cConf.set(Constants.CFG_LOCAL_DATA_DIR, tmpFolder.newFolder().getAbsolutePath());
 Injector injector = Guice.createInjector(
  new ConfigModule(cConf),
  new NonCustomLocationUnitTestModule(),
  new InMemoryDiscoveryModule(),
  new DataSetsModules().getStandaloneModules(),
  new DataFabricLevelDBModule(),
  new TransactionMetricsModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getStandaloneModules(),
  new AuthenticationContextModules().getMasterModule());
 service = injector.getInstance(LevelDBTableService.class);
}

@BeforeClass
public static void createInjector() throws IOException {
 CConfiguration cConf = CConfiguration.create();
 cConf.set(Constants.CFG_LOCAL_DATA_DIR, TEMP_FOLDER.newFolder().getAbsolutePath());
 Injector injector = Guice.createInjector(
  new ConfigModule(cConf),
  new DataSetsModules().getInMemoryModules(),
  new LocalLocationModule(),
  new TransactionInMemoryModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new NamespaceAdminTestModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getInMemoryModules(),
  new AuthenticationContextModules().getMasterModule()
 );
 TransactionManager txManager = injector.getInstance(TransactionManager.class);
 txManager.startAndWait();
 ownerStore = injector.getInstance(OwnerStore.class);
}

@BeforeClass
public static void init() throws Exception {
 CConfiguration conf = CConfiguration.create();
 conf.set(Constants.CFG_LOCAL_DATA_DIR, tmpFolder.newFolder().getAbsolutePath());
 injector = Guice.createInjector(
  new ConfigModule(conf),
  new NonCustomLocationUnitTestModule(),
  new InMemoryDiscoveryModule(),
  new DataSetsModules().getStandaloneModules(),
  new DataFabricLevelDBModule(),
  new TransactionMetricsModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getStandaloneModules(),
  new AuthenticationContextModules().getMasterModule());
 service = injector.getInstance(LevelDBTableService.class);
}

@VisibleForTesting
static Injector createGuiceInjector(CConfiguration cConf, Configuration hConf, String txClientId,
                  TwillContext twillContext) {
 return Guice.createInjector(
  new ConfigModule(cConf, hConf),
  new IOModule(),
  new ZKClientModule(),
  new ZKDiscoveryModule(),
  new KafkaClientModule(),
  new MessagingClientModule(),
  new MetricsClientRuntimeModule().getDistributedModules(),
  new MetricsStoreModule(),
  new KafkaLogAppenderModule(),
  new DFSLocationModule(),
  new NamespaceQueryAdminModule(),
  new DataFabricModules(txClientId).getDistributedModules(),
  new DataSetsModules().getDistributedModules(),
  new MetricsProcessorModule(twillContext),
  new MetricsProcessorStatusServiceModule(),
  new AuditModule().getDistributedModules(),
  new AuthorizationEnforcementModule().getDistributedModules(),
  new AuthenticationContextModules().getMasterModule(),
  new AbstractModule() {
   @Override
   protected void configure() {
    bind(OwnerAdmin.class).to(DefaultOwnerAdmin.class);
   }
  }
 );
}

new NamespaceQueryAdminModule(),
new DataFabricModules(txClientId).getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new KafkaLogAppenderModule(),
new ExploreRuntimeModule().getDistributedModules(),

@BeforeClass
public static void setup() throws Exception {
 CConfiguration conf = CConfiguration.create();
 conf.set(Constants.CFG_LOCAL_DATA_DIR, tmpFolder.newFolder().getAbsolutePath());
 Injector injector = Guice.createInjector(
  new ConfigModule(conf),
  new NonCustomLocationUnitTestModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getInMemoryModules(),
  new AuthenticationContextModules().getMasterModule(),
  new InMemoryDiscoveryModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new DataSetsModules().getInMemoryModules(),
  new DataFabricLevelDBModule(),
  new TransactionMetricsModule());
 dsFramework = injector.getInstance(DatasetFramework.class);
}

@BeforeClass
public static void setup() {
 Injector injector = Guice.createInjector(
  new ConfigModule(),
  Modules.override(
   new DataSetsModules().getInMemoryModules()).with(new AbstractModule() {
   @Override
   protected void configure() {
    // Need the distributed metadata store.
    bind(MetadataStore.class).to(DefaultMetadataStore.class);
   }
  }),
  new LocalLocationModule(),
  new TransactionInMemoryModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new NamespaceAdminTestModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getInMemoryModules(),
  new AuthenticationContextModules().getMasterModule()
 );
 txManager = injector.getInstance(TransactionManager.class);
 txManager.startAndWait();
 storage = injector.getInstance(DatasetMetadataStorage.class);
}

new MessagingClientModule(),
new DataFabricModules(txClientId).getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new DFSLocationModule(),
new NamespaceQueryAdminModule(),

new MetricsClientRuntimeModule().getInMemoryModules(),
new DataFabricModules().getInMemoryModules(),
new DataSetsModules().getStandaloneModules(),
new DataSetServiceModules().getInMemoryModules(),
new ExploreClientModule(),

new MessagingClientModule(),
new DataFabricModules(txClientId).getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new DFSLocationModule(),
new NamespaceQueryAdminModule(),

new NamespaceQueryAdminModule(),
new DataFabricModules(txClientId).getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new DataSetServiceModules().getDistributedModules(),
new KafkaLogAppenderModule(),

new DiscoveryRuntimeModule().getDistributedModules(),
new DataFabricModules("cdap.explore.ContextManager").getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new ExploreClientModule(),
new ViewAdminModules().getDistributedModules(),

@BeforeClass
public static void setupDefaultStore() {
 injector = Guice.createInjector(
  new ConfigModule(),
  Modules.override(
   new DataSetsModules().getInMemoryModules()).with(new AbstractModule() {
   @Override
   protected void configure() {
    // Need the distributed metadata store.
    bind(MetadataStore.class).to(DefaultMetadataStore.class);
   }
  }),
  new LocalLocationModule(),
  new TransactionInMemoryModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new NamespaceAdminTestModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getInMemoryModules(),
  new AuthenticationContextModules().getMasterModule(),
  new AuditModule().getInMemoryModules()
 );
 txManager = injector.getInstance(TransactionManager.class);
 txManager.startAndWait();
 AbstractMetadataStoreTest.commonSetup();
}

@BeforeClass
public static void setupDefaultStore() {
 injector = Guice.createInjector(
  new ConfigModule(),
  Modules.override(
   new DataSetsModules().getInMemoryModules()).with(new AbstractModule() {
   @Override
   protected void configure() {
    // Need the distributed metadata store.
    bind(MetadataStore.class).to(StorageProviderMetadataStore.class);
    bind(MetadataStorage.class).to(DatasetMetadataStorage.class).in(Scopes.SINGLETON);
   }
  }),
  new LocalLocationModule(),
  new TransactionInMemoryModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new NamespaceAdminTestModule(),
  new AuthorizationTestModule(),
  new AuthorizationEnforcementModule().getInMemoryModules(),
  new AuthenticationContextModules().getMasterModule(),
  new AuditModule().getInMemoryModules()
 );
 txManager = injector.getInstance(TransactionManager.class);
 txManager.startAndWait();
 AbstractMetadataStoreTest.commonSetup();
}

 @Override
 protected void configure() {
  install(new DataFabricModules().getInMemoryModules());
  install(new DataSetsModules().getStandaloneModules());
  install(new TransactionExecutorModule());
  install(new DataSetServiceModules().getInMemoryModules());
  install(new ConfigModule(cConf, hConf, sConf));
  install(new IOModule());
  install(new InMemoryDiscoveryModule());
  install(new AppFabricServiceRuntimeModule().getInMemoryModules());
  install(new ServiceStoreModules().getInMemoryModules());
  install(new ProgramRunnerRuntimeModule().getInMemoryModules());
  install(new NonCustomLocationUnitTestModule());
  install(new LocalLogAppenderModule());
  install(new LogReaderRuntimeModules().getInMemoryModules());
  install(new LogQueryServerModule());
  install(new MetricsHandlerModule());
  install(new MetricsClientRuntimeModule().getInMemoryModules());
  install(new ExploreClientModule());
  install(new ConfigStoreModule().getInMemoryModule());
  install(new NamespaceStoreModule().getStandaloneModules());
  install(new MetadataServiceModule());
  install(new AuthorizationModule());
  install(new AuthorizationEnforcementModule().getStandaloneModules());
  install(new SecureStoreServerModule());
  install(new MetadataReaderWriterModules().getInMemoryModules());
  install(new MessagingServerRuntimeModule().getInMemoryModules());
  install(new MockProvisionerModule());
 }
}