How to use

withApplicationTaskContextFactory

method

in

org.apache.samza.application.descriptors.StreamApplicationDescriptor

/**
 * For unit testing only
 */
@VisibleForTesting
void translate(SamzaSqlQueryParser.QueryInfo queryInfo, StreamApplicationDescriptor appDesc, int queryId) {
 QueryPlanner planner =
   new QueryPlanner(sqlConfig.getRelSchemaProviders(), sqlConfig.getInputSystemStreamConfigBySource(),
     sqlConfig.getUdfMetadata());
 final RelRoot relRoot = planner.plan(queryInfo.getSelectQuery());
 SamzaSqlExecutionContext executionContext = new SamzaSqlExecutionContext(sqlConfig);
 TranslatorContext translatorContext = new TranslatorContext(appDesc, relRoot, executionContext);
 translate(relRoot, sqlConfig.getOutputSystemStreams().get(queryId), translatorContext, queryId);
 Map<Integer, TranslatorContext> translatorContexts = new HashMap<>();
 translatorContexts.put(queryId, translatorContext.clone());
 appDesc.withApplicationTaskContextFactory(new ApplicationTaskContextFactory<SamzaSqlApplicationContext>() {
  @Override
  public SamzaSqlApplicationContext create(ExternalContext externalContext, JobContext jobContext,
    ContainerContext containerContext, TaskContext taskContext,
    ApplicationContainerContext applicationContainerContext) {
   return new SamzaSqlApplicationContext(translatorContexts);
  }
 });
}

appDescriptor.withApplicationTaskContextFactory(new ApplicationTaskContextFactory<SamzaSqlApplicationContext>() {
 @Override
 public SamzaSqlApplicationContext create(ExternalContext externalContext, JobContext jobContext,

appDescriptor.withApplicationTaskContextFactory((jobContext,
  containerContext,
  taskContext,

@Test
public void testApplicationTaskContextFactory() {
 ApplicationTaskContextFactory factory = mock(ApplicationTaskContextFactory.class);
 StreamApplication testApp = appDesc -> appDesc.withApplicationTaskContextFactory(factory);
 StreamApplicationDescriptorImpl appSpec = new StreamApplicationDescriptorImpl(testApp, getConfig());
 assertEquals(appSpec.getApplicationTaskContextFactory(), Optional.of(factory));
}

/**
 * For unit testing only
 */
@VisibleForTesting
public void translate(SamzaSqlQueryParser.QueryInfo queryInfo, StreamApplicationDescriptor appDesc) {
 QueryPlanner planner =
   new QueryPlanner(sqlConfig.getRelSchemaProviders(), sqlConfig.getSystemStreamConfigsBySource(),
     sqlConfig.getUdfMetadata());
 final RelRoot relRoot = planner.plan(queryInfo.getSql());
 int queryId = 1;
 SamzaSqlExecutionContext executionContext = new SamzaSqlExecutionContext(sqlConfig);
 Map<String, SamzaRelConverter> converters = sqlConfig.getSamzaRelConverters();
 TranslatorContext translatorContext = new TranslatorContext(appDesc, relRoot, executionContext, converters);
 translate(relRoot, translatorContext, queryId);
 Map<Integer, TranslatorContext> translatorContexts = new HashMap<>();
 translatorContexts.put(queryId, translatorContext.clone());
 appDesc.withApplicationTaskContextFactory((jobContext,
   containerContext,
   taskContext,
   applicationContainerContext) ->
   new SamzaSqlApplicationContext(translatorContexts));
}