use of org.talend.components.netsuite.client.MetaDataSource in project components by Talend.
the class NetSuiteDatasetRuntimeTest method testGetSchemaForRecordWithCustomFields.
@Test
public void testGetSchemaForRecordWithCustomFields() throws Exception {
final CustomMetaDataSource customMetaDataSource = new EmptyCustomMetaDataSource() {
@Override
public Map<String, CustomFieldDesc> getCustomFields(RecordTypeInfo recordTypeInfo) {
try {
if (recordTypeInfo.getName().equals("Account")) {
JsonNode fieldListNode = objectMapper.readTree(NetSuiteDatasetRuntimeTest.class.getResource("/test-data/customFields-1.json"));
Map<String, CustomFieldDesc> customFieldDescMap = TestUtils.readCustomFields(fieldListNode);
return customFieldDescMap;
}
return null;
} catch (IOException e) {
throw new NetSuiteException(e.getMessage(), e);
}
}
};
MetaDataSource metaDataSource = clientService.createDefaultMetaDataSource();
metaDataSource.setCustomMetaDataSource(customMetaDataSource);
NetSuiteDatasetRuntimeImpl datasetRuntime = new NetSuiteDatasetRuntimeImpl(metaDataSource);
TypeDesc typeDesc = metaDataSource.getTypeInfo("Account");
Schema s = datasetRuntime.getSchema(typeDesc.getTypeName());
assertThat(s.getType(), is(Schema.Type.RECORD));
assertThat(s.getName(), is("Account"));
assertThat(s.getFields(), hasSize(typeDesc.getFields().size()));
assertThat(s.getObjectProps().keySet(), Matchers.<String>empty());
CustomFieldDesc fieldDesc = (CustomFieldDesc) typeDesc.getField("custom_field_1");
Schema.Field f = getNsFieldByName(s, fieldDesc.getName());
assertUnionType(f.schema(), Arrays.asList(Schema.Type.STRING, Schema.Type.NULL));
assertThat(f.getObjectProps().keySet(), containsInAnyOrder(SchemaConstants.TALEND_COLUMN_DB_COLUMN_NAME, SchemaConstants.TALEND_COLUMN_DB_TYPE, NetSuiteSchemaConstants.NS_CUSTOM_FIELD, NetSuiteSchemaConstants.NS_CUSTOM_FIELD_SCRIPT_ID, NetSuiteSchemaConstants.NS_CUSTOM_FIELD_INTERNAL_ID, NetSuiteSchemaConstants.NS_CUSTOM_FIELD_CUSTOMIZATION_TYPE, NetSuiteSchemaConstants.NS_CUSTOM_FIELD_TYPE));
assertThat(f.getProp(SchemaConstants.TALEND_COLUMN_DB_COLUMN_NAME), is(fieldDesc.getName()));
CustomFieldDesc customFieldDesc = NetSuiteDatasetRuntimeImpl.readCustomField(f);
assertThat(customFieldDesc.getName(), is(fieldDesc.getName()));
}
use of org.talend.components.netsuite.client.MetaDataSource in project components by Talend.
the class NetSuiteEndpoint method connect.
/**
* Creates new NetSuite client and connects to NetSuite remote endpoint.
*
* @param connectionConfig connection configuration
* @return client
* @throws NetSuiteException if an error occurs during connecting
*/
private NetSuiteClientService<?> connect(ConnectionConfig connectionConfig) throws NetSuiteException {
NetSuiteClientService<?> clientService = clientFactory.createClient();
clientService.setEndpointUrl(connectionConfig.getEndpointUrl().toString());
clientService.setCredentials(connectionConfig.getCredentials());
clientService.setBodyFieldsOnly(connectionConfig.isBodyFieldsOnly());
MetaDataSource metaDataSource = clientService.getMetaDataSource();
metaDataSource.setCustomizationEnabled(connectionConfig.isCustomizationEnabled());
clientService.login();
return clientService;
}
use of org.talend.components.netsuite.client.MetaDataSource in project components by Talend.
the class NetSuiteSearchInputReader method start.
@Override
public boolean start() throws IOException {
try {
schema = properties.module.main.schema.getValue();
clientService = ((NetSuiteSource) getCurrentSource()).getClientService();
// Set up MetaDataSource which retrieves customization meta data from schema.
// We use MetaDataSource from NetSuite client as base source.
MetaDataSource originalMetaDataSource = clientService.getMetaDataSource();
metaDataSource = clientService.createDefaultMetaDataSource();
metaDataSource.setCustomizationEnabled(originalMetaDataSource.isCustomizationEnabled());
SchemaCustomMetaDataSource schemaCustomMetaDataSource = new SchemaCustomMetaDataSource(clientService.getBasicMetaData(), originalMetaDataSource.getCustomMetaDataSource(), schema);
metaDataSource.setCustomMetaDataSource(schemaCustomMetaDataSource);
// Execute search and get search result set.
resultSet = search();
return advance();
} catch (NetSuiteException e) {
throw new IOException(e);
}
}
use of org.talend.components.netsuite.client.MetaDataSource in project components by Talend.
the class NetSuiteWriteOperation method createWriter.
@Override
public Writer<Result> createWriter(RuntimeContainer adaptor) {
NetSuiteClientService clientService = sink.getClientService();
OutputAction action = properties.module.action.getValue();
Schema schema = properties.module.main.schema.getValue();
MetaDataSource originalMetaDataSource = clientService.getMetaDataSource();
MetaDataSource metaDataSource = clientService.createDefaultMetaDataSource();
metaDataSource.setCustomizationEnabled(originalMetaDataSource.isCustomizationEnabled());
SchemaCustomMetaDataSource schemaCustomMetaDataSource = new SchemaCustomMetaDataSource(clientService.getBasicMetaData(), originalMetaDataSource.getCustomMetaDataSource(), schema);
metaDataSource.setCustomMetaDataSource(schemaCustomMetaDataSource);
NetSuiteOutputWriter<?, ?> writer;
switch(action) {
case ADD:
writer = new NetSuiteAddWriter<>(this, adaptor, metaDataSource);
break;
case UPDATE:
writer = new NetSuiteUpsertWriter<>(this, adaptor, metaDataSource);
break;
case UPSERT:
writer = new NetSuiteUpsertWriter<>(this, adaptor, metaDataSource);
Boolean useNativeUpsert = properties.module.useNativeUpsert.getValue();
if (useNativeUpsert != null) {
((NetSuiteUpsertWriter) writer).setUseNativeUpsert(useNativeUpsert);
}
break;
case DELETE:
writer = new NetSuiteDeleteWriter<>(this, adaptor, metaDataSource);
break;
default:
throw new NetSuiteException(new NetSuiteErrorCode(NetSuiteErrorCode.INTERNAL_ERROR), NetSuiteRuntimeI18n.MESSAGES.getMessage("error.outputOperationNotImplemented", action));
}
Integer batchSize = properties.batchSize.getValue();
if (batchSize != null) {
writer.setBatchSize(batchSize);
}
Boolean dieOnError = properties.dieOnError.getValue();
if (dieOnError != null) {
writer.setExceptionForErrors(dieOnError);
}
return writer;
}
Aggregations