/** qzhang Comment method "initPerlArray". */
private void initPerlArray() {
List<Map<String, Object>> map = new ArrayList<Map<String, Object>>();
MetadataTable table = (MetadataTable) this.component.getMetadataList().get(0);
for (IMetadataColumn col : table.getListColumns()) {
MetadataColumnExt ext = (MetadataColumnExt) col;
Map<String, Object> value = new HashMap<String, Object>();
value.put(RowGeneratorComponent.COLUMN_NAME, ext.getLabel());
value.put(RowGeneratorComponent.ARRAY, FunctionManagerExt.getOneColData(ext));
map.add(value);
}
this.component.setTableElementParameter(map);
}
public String generate(Object argument) {
final StringBuffer stringBuffer = new StringBuffer();
final BigDataCodeGeneratorArgument codeGenArgument = (BigDataCodeGeneratorArgument) argument;
final INode node = (INode) codeGenArgument.getArgument();
final String cid = node.getUniqueName();
IMetadataTable metadata = null;
IConnection conn = null;
List<IMetadataColumn> columns = null;
List<IMetadataTable> metadatas = node.getMetadataList();
if ((metadatas != null) && (metadatas.size() > 0)) {
metadata = metadatas.get(0);
if (metadata != null) {
columns = metadata.getListColumns();
}
}
List<? extends IConnection> conns = node.getIncomingConnections();
if (conns != null
&& conns.size() > 0
&& conns.get(0).getLineStyle().hasConnectionCategory(IConnectionCategory.DATA)) {
conn = conns.get(0);
}
if (columns == null || columns.isEmpty() || conn == null) {
return "";
}
String inRowStruct = codeGenArgument.getRecordStructName(conn);
String connName = conn.getName();
String nodes = ElementParameterParser.getValue(node, "__NODES__");
String index = ElementParameterParser.getValue(node, "__INDEX__");
String type = ElementParameterParser.getValue(node, "__TYPE__");
boolean jsonDoc = "JSON".equals(ElementParameterParser.getValue(node, "__DOC_TYPE__"));
List<Map<String, String>> configuration =
(List<Map<String, String>>)
ElementParameterParser.getObjectValue(node, "__CONFIGURATION__");
List<Map<String, String>> docMetadatas =
(List<Map<String, String>>) ElementParameterParser.getObjectValue(node, "__DOC_METADATA__");
Map<String, String> docMetadataMapping = new HashMap<String, String>();
org.talend.core.model.metadata.MetadataTable docColumnTable =
new org.talend.core.model.metadata.MetadataTable(); // for non-metadata column
List<IMetadataColumn> docColumnList = new ArrayList<IMetadataColumn>();
docColumnTable.setListColumns(docColumnList);
for (Map<String, String> docMetadata : docMetadatas) {
if ("true".equals(docMetadata.get("AS_META"))) {
docMetadataMapping.put(docMetadata.get("SCHEMA_COLUMN"), docMetadata.get("TYPE"));
} else {
for (IMetadataColumn col : columns) {
if (col.getLabel().equals(docMetadata.get("SCHEMA_COLUMN"))) {
docColumnList.add(col);
}
}
}
}
final boolean definedDocMetadata = docMetadataMapping.size() > 0;
if (definedDocMetadata && !jsonDoc) {
codeGenArgument
.getRecordStructGenerator()
.generateRecordStruct("doc_" + connName, docColumnTable);
}
if (definedDocMetadata) {
stringBuffer.append(TEXT_1);
stringBuffer.append(connName);
stringBuffer.append(TEXT_2);
stringBuffer.append(connName);
stringBuffer.append(TEXT_3);
stringBuffer.append(cid);
stringBuffer.append(TEXT_4);
} else {
if (jsonDoc) {
stringBuffer.append(TEXT_5);
stringBuffer.append(connName);
stringBuffer.append(TEXT_6);
stringBuffer.append(connName);
stringBuffer.append(TEXT_7);
stringBuffer.append(cid);
stringBuffer.append(TEXT_8);
} else {
stringBuffer.append(TEXT_9);
stringBuffer.append(inRowStruct);
stringBuffer.append(TEXT_10);
stringBuffer.append(connName);
stringBuffer.append(TEXT_11);
stringBuffer.append(connName);
stringBuffer.append(TEXT_12);
}
}
stringBuffer.append(TEXT_13);
stringBuffer.append(TEXT_14);
stringBuffer.append(cid);
stringBuffer.append(TEXT_15);
stringBuffer.append(cid);
stringBuffer.append(TEXT_16);
stringBuffer.append(cid);
stringBuffer.append(TEXT_17);
stringBuffer.append(cid);
stringBuffer.append(TEXT_18);
stringBuffer.append(connName);
stringBuffer.append(TEXT_19);
stringBuffer.append(TEXT_20);
return stringBuffer.toString();
}
