public void generateKafkaTopics(String cid) {
stringBuffer.append(TEXT_18);
stringBuffer.append(cid);
stringBuffer.append(TEXT_19);
if (tKafkaInputUtil.getKafkaTopics().isEmpty()) {
stringBuffer.append(TEXT_20);
} else {
for (String kafkaTopic : tKafkaInputUtil.getKafkaTopics()) {
stringBuffer.append(TEXT_21);
stringBuffer.append(cid);
stringBuffer.append(TEXT_22);
stringBuffer.append(kafkaTopic);
stringBuffer.append(TEXT_23);
} // end for
} // end else
} // end generateKafkaTopics
public void generateKafkaProperties(String cid) {
stringBuffer.append(TEXT_3);
stringBuffer.append(cid);
stringBuffer.append(TEXT_4);
if (tKafkaInputUtil.getBrokerList() == null || "".equals(tKafkaInputUtil.getBrokerList())) {
stringBuffer.append(TEXT_5);
} else {
stringBuffer.append(TEXT_6);
stringBuffer.append(cid);
stringBuffer.append(TEXT_7);
stringBuffer.append(tKafkaInputUtil.getBrokerList());
stringBuffer.append(TEXT_8);
for (Entry<String, String> kafkaProperty :
tKafkaInputUtil.getKafkaConsumerProperties().entrySet()) {
stringBuffer.append(TEXT_9);
stringBuffer.append(cid);
stringBuffer.append(TEXT_10);
stringBuffer.append(kafkaProperty.getKey());
stringBuffer.append(TEXT_11);
stringBuffer.append(kafkaProperty.getValue());
stringBuffer.append(TEXT_12);
} // end for
} // end else
stringBuffer.append(TEXT_13);
stringBuffer.append(cid);
stringBuffer.append(TEXT_14);
stringBuffer.append(tKafkaInputUtil.getEncoding());
stringBuffer.append(TEXT_15);
stringBuffer.append(cid);
stringBuffer.append(TEXT_16);
stringBuffer.append(tKafkaInputUtil.getAutoOffsetReset());
stringBuffer.append(TEXT_17);
} // end generateKafkaProperties
public String generate(Object argument) {
final StringBuffer stringBuffer = new StringBuffer();
stringBuffer.append(TEXT_1);
stringBuffer.append(TEXT_2);
/**
* This is a Javajet helper class meant to mutualize code generation for tKafkaXXXInput
* components. These components share several parameters : - the Kafka consumer properties - the
* list of Kafka topics to read from
*
* <p>Each tKafkaXXXInput_sparkstreamingconfig.javajet should at least import this file to avoid
* code duplication.
*/
final class TKafkaInputHelper {
private TKafkaInputUtil tKafkaInputUtil;
public TKafkaInputHelper(TKafkaInputUtil util) {
tKafkaInputUtil = util;
}
public void generateKafkaProperties(String cid) {
stringBuffer.append(TEXT_3);
stringBuffer.append(cid);
stringBuffer.append(TEXT_4);
if (tKafkaInputUtil.getBrokerList() == null || "".equals(tKafkaInputUtil.getBrokerList())) {
stringBuffer.append(TEXT_5);
} else {
stringBuffer.append(TEXT_6);
stringBuffer.append(cid);
stringBuffer.append(TEXT_7);
stringBuffer.append(tKafkaInputUtil.getBrokerList());
stringBuffer.append(TEXT_8);
for (Entry<String, String> kafkaProperty :
tKafkaInputUtil.getKafkaConsumerProperties().entrySet()) {
stringBuffer.append(TEXT_9);
stringBuffer.append(cid);
stringBuffer.append(TEXT_10);
stringBuffer.append(kafkaProperty.getKey());
stringBuffer.append(TEXT_11);
stringBuffer.append(kafkaProperty.getValue());
stringBuffer.append(TEXT_12);
} // end for
} // end else
stringBuffer.append(TEXT_13);
stringBuffer.append(cid);
stringBuffer.append(TEXT_14);
stringBuffer.append(tKafkaInputUtil.getEncoding());
stringBuffer.append(TEXT_15);
stringBuffer.append(cid);
stringBuffer.append(TEXT_16);
stringBuffer.append(tKafkaInputUtil.getAutoOffsetReset());
stringBuffer.append(TEXT_17);
} // end generateKafkaProperties
public void generateKafkaTopics(String cid) {
stringBuffer.append(TEXT_18);
stringBuffer.append(cid);
stringBuffer.append(TEXT_19);
if (tKafkaInputUtil.getKafkaTopics().isEmpty()) {
stringBuffer.append(TEXT_20);
} else {
for (String kafkaTopic : tKafkaInputUtil.getKafkaTopics()) {
stringBuffer.append(TEXT_21);
stringBuffer.append(cid);
stringBuffer.append(TEXT_22);
stringBuffer.append(kafkaTopic);
stringBuffer.append(TEXT_23);
} // end for
} // end else
} // end generateKafkaTopics
} // end class TKafkaInputHelper
stringBuffer.append(TEXT_24);
BigDataCodeGeneratorArgument codeGenArgument = (BigDataCodeGeneratorArgument) argument;
INode node = (INode) codeGenArgument.getArgument();
String cid = node.getUniqueName();
TKafkaInputUtil tKafkaInputUtil = new TKafkaInputUtil(node);
TKafkaInputHelper tKafkaInputHelper = new TKafkaInputHelper(tKafkaInputUtil);
tKafkaInputHelper.generateKafkaProperties(cid);
tKafkaInputHelper.generateKafkaTopics(cid);
boolean useHierarchical =
"true".equals(ElementParameterParser.getValue(node, "__USE_HIERARCHICAL__"));
if (useHierarchical) {
String schemaFileName = ElementParameterParser.getValue(node, "__SCHEMA_FILENAME__");
stringBuffer.append(TEXT_25);
stringBuffer.append(cid);
stringBuffer.append(TEXT_26);
stringBuffer.append(schemaFileName);
stringBuffer.append(TEXT_27);
stringBuffer.append(TEXT_28);
stringBuffer.append(cid);
stringBuffer.append(TEXT_29);
stringBuffer.append(cid);
stringBuffer.append(TEXT_30);
}
stringBuffer.append(TEXT_31);
stringBuffer.append(tKafkaInputUtil.getOutgoingConnection().getName());
stringBuffer.append(TEXT_32);
stringBuffer.append(cid);
stringBuffer.append(TEXT_33);
stringBuffer.append(cid);
stringBuffer.append(TEXT_34);
stringBuffer.append(cid);
stringBuffer.append(TEXT_35);
stringBuffer.append(cid);
stringBuffer.append(TEXT_36);
stringBuffer.append(TEXT_37);
return stringBuffer.toString();
}
