How to use

co.cask.cdap.data.runtime.DataFabricModules

constructor

private static Module createDataFabricModule() {
 return Modules.override(new DataFabricModules().getInMemoryModules())
  .with(new AbstractModule() {
   @Override
   protected void configure() {
    // we inject a TxSystemClient that creates transaction objects with additional fields for validation
    bind(InMemoryTxSystemClient.class).in(Scopes.SINGLETON);
    bind(TransactionSystemClient.class).to(RevealingTxSystemClient.class).in(Scopes.SINGLETON);
   }
  });
}

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);
  }
 });
}

modules.add(new DataSetsModules().getStandaloneModules());
modules.add(new DataSetServiceModules().getStandaloneModules());
modules.add(Modules.override(new DataFabricModules().getInMemoryModules()).with(new AbstractModule() {
 @Override
 protected void configure() {

private void addIsolatedModules(List<Module> modules) {
 modules.add(new DataSetsModules().getStandaloneModules());
 modules.add(new DataSetServiceModules().getStandaloneModules());
 modules.add(Modules.override(new DataFabricModules().getInMemoryModules()).with(new AbstractModule() {
  @Override
  protected void configure() {

private static Module createDataFabricModule() {
 return Modules.override(new DataFabricModules().getInMemoryModules(),
             new ViewAdminModules().getInMemoryModules(),
             new StreamAdminModules().getInMemoryModules())
  .with(new AbstractModule() {
   @Override
   protected void configure() {
    bind(StreamConsumerStateStoreFactory.class)
     .to(LevelDBStreamConsumerStateStoreFactory.class).in(Singleton.class);
    bind(StreamConsumerFactory.class).to(LevelDBStreamFileConsumerFactory.class).in(Singleton.class);
    bind(StreamFileWriterFactory.class).to(LocationStreamFileWriterFactory.class).in(Singleton.class);
    // we inject a TxSystemClient that creates transaction objects with additional fields for validation
    bind(InMemoryTxSystemClient.class).in(Scopes.SINGLETON);
    bind(TransactionSystemClient.class).to(RevealingTxSystemClient.class).in(Scopes.SINGLETON);
   }
  });
}

@BeforeClass
public static void setup() {
 Injector injector = Guice.createInjector(
  new ConfigModule(),
  new NonCustomLocationUnitTestModule(),
  new InMemoryDiscoveryModule(),
  new DataFabricModules().getInMemoryModules(),
  new TransactionMetricsModule(),
  new SystemDatasetRuntimeModule().getInMemoryModules(),
  new PrivateModule() {
   @Override
   protected void configure() {
    bind(DatasetDefinitionRegistryFactory.class)
     .to(DefaultDatasetDefinitionRegistryFactory.class).in(Scopes.SINGLETON);
    bind(DatasetFramework.class).to(InMemoryDatasetFramework.class);
    expose(DatasetFramework.class);
   }
  });
 dsFramework = injector.getInstance(DatasetFramework.class);
}

@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);
    }
   }
  );
 }
}

@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 DFSLocationModule(),
new NamespaceQueryAdminModule(),
new DataFabricModules(txClientId).getDistributedModules(),
new DataSetsModules().getDistributedModules(),
new KafkaLogAppenderModule(),

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

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

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

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

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

new LocalLocationModule(),
new ProgramRunnerRuntimeModule().getStandaloneModules(),
new DataFabricModules(StandaloneMain.class.getName()).getStandaloneModules(),
new DataSetsModules().getStandaloneModules(),
new DataSetServiceModules().getStandaloneModules(),

new ConfigModule(cConf, new Configuration()),
new IOModule(),
new DataFabricModules().getInMemoryModules(),
new DataSetsModules().getStandaloneModules(),
new TransactionExecutorModule(),

 @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());
 }
}