use of orgomg.cwm.resource.relational.Schema in project tdi-studio-se by Talend.
the class MergeTosMetadataMigrationTask method execute.
@Override
public ExecutionResult execute(Item item) {
if (item instanceof ConnectionItem) {
try {
URI itemURI = getItemURI(item);
if (itemURI != null) {
URI itemResourceURI = getItemResourceURI(itemURI);
if (metadata400to410 == null) {
metadata400to410 = new TosMetadataMigrationFrom400to410();
}
Resource migratedResource = metadata400to410.migrate(itemResourceURI.toString(), new NullProgressMonitor());
HashMap<String, Object> xmlSaveOtions = XML_SAVE_OTIONS_1_0;
if (migratedResource != null) {
// check for DB connection caus we need to setup Schema and Catalog properly
EObject content = migratedResource.getContents().get(0);
if (content != null && "DatabaseConnection".equals(content.eClass().getName())) {
// resource is dynamic EMF so convert it to static model by serialising it and reloading it
ByteArrayOutputStream tempStream = new ByteArrayOutputStream();
try {
// serialize into memory
try {
migratedResource.save(tempStream, XML_SAVE_OTIONS_1_0);
} catch (Exception e) {
// try with version 1.1
tempStream = new ByteArrayOutputStream();
xmlSaveOtions = XML_SAVE_OTIONS_1_1;
migratedResource.save(tempStream, xmlSaveOtions);
} finally {
tempStream.close();
}
// create a resource to laod the inmemory resource that should be a static EMF model
migratedResource = resourceSet.createResource(URI.createURI(//$NON-NLS-1$
"http://talend/dummy_static.metadata"));
migratedResource.load(new ByteArrayInputStream(tempStream.toByteArray()), xmlSaveOtions);
// check that DBConnection is firdt element
DatabaseConnection databaseConnection = SwitchHelpers.DATABASECONNECTION_SWITCH.doSwitch(migratedResource.getContents().get(0));
// do not check for null caus DB connection is already check above
String databaseType = databaseConnection.getDatabaseType();
databaseConnection.setDriverClass(ExtractMetaDataUtils.getInstance().getDriverClassByDbType(databaseType));
EDatabaseTypeName currentType = EDatabaseTypeName.getTypeFromDbType(databaseType);
EDatabaseSchemaOrCatalogMapping curCatalog = currentType.getCatalogMappingField();
EDatabaseSchemaOrCatalogMapping curSchema = currentType.getSchemaMappingField();
// all the DB connection are migrated with a Schema by default
if (!curCatalog.equals(EDatabaseSchemaOrCatalogMapping.None)) {
List<Schema> schemas = ConnectionHelper.getSchema(databaseConnection);
if (!curSchema.equals(EDatabaseSchemaOrCatalogMapping.None)) {
// we need to place the current schemas into a catalogs
ConnectionHelper.removeSchemas(schemas, databaseConnection);
for (Schema schema : schemas) {
// compute the name of the schema and the catalogs
String schemaName = computeSchemaName(schema, databaseConnection, curSchema);
String catalogName = computeCatalogName(databaseConnection, curCatalog);
schema.setName(schemaName);
Catalog catalog = RelationalFactory.eINSTANCE.createCatalog();
// catalogs are not in a contained reference
migratedResource.getContents().add(catalog);
catalog.setName(catalogName);
// add the schema to the catalog and the the catalog to the connection
CatalogHelper.addSchemas(Collections.singleton(schema), catalog);
ConnectionHelper.addCatalog(catalog, databaseConnection);
}
} else {
// we need to replace the Schemas with a Catalogs
for (Schema schema : schemas) {
// compute the name the catalog
String catalogName = computeCatalogName(databaseConnection, curCatalog);
// use owned elements to get everything regardless of tables or views or
// else
Catalog catalog = RelationalFactory.eINSTANCE.createCatalog();
// catalogs are not in a contained reference
migratedResource.getContents().add(catalog);
catalog.setName(catalogName);
catalog.getOwnedElement().addAll(schema.getOwnedElement());
ConnectionHelper.addCatalog(catalog, databaseConnection);
ConnectionHelper.removeSchemas(Collections.singleton(schema), databaseConnection);
}
}
} else if (!curSchema.equals(EDatabaseSchemaOrCatalogMapping.None)) {
List<Schema> schemas = ConnectionHelper.getSchema(databaseConnection);
for (Schema schema : schemas) {
String schemaName = computeSchemaName(schema, databaseConnection, curSchema);
schema.setName(schemaName);
}
}
// else no catalog so we keep the schema as is
} catch (Exception e) {
// we have an exception finalising the migration but we trap it caus we still try to
// save it
log.error("Cannot complete merge metadata migration on file:" + itemResourceURI.toString(), e);
ExceptionHandler.process(e);
} finally {
tempStream.close();
}
}
// else not a DB connection so persist
OutputStream outputStream = item.eResource().getResourceSet().getURIConverter().createOutputStream(itemResourceURI, null);
try {
migratedResource.save(outputStream, xmlSaveOtions);
} finally {
outputStream.close();
}
}
return ExecutionResult.SUCCESS_WITH_ALERT;
}
} catch (ATLCoreException e) {
log.error(e);
ExceptionHandler.process(e);
return ExecutionResult.SUCCESS_NO_ALERT;
} catch (IOException e) {
log.error(e);
ExceptionHandler.process(e);
return ExecutionResult.SUCCESS_NO_ALERT;
} finally {
resourceSet.getResources().clear();
}
}
return ExecutionResult.NOTHING_TO_DO;
}
use of orgomg.cwm.resource.relational.Schema in project tdi-studio-se by Talend.
the class MetadataColumnComparator method restoreConnection.
/**
* dev Comment method "restoreConnection".
*
* @param connection
* @param tablesFromEMF
*/
@SuppressWarnings("unchecked")
private void restoreConnection(DatabaseConnection connection, List<MetadataTable> tablesFromEMF) {
tablesFromEMF = sortTableColumn(tablesFromEMF);
// bug when do refresh twice it will add all tables
if (tList == null) {
tList = new ArrayList<MetadataTable>();
tList.addAll(ConnectionHelper.getTables(connection));
}
Catalog c = (Catalog) ConnectionHelper.getPackage(connection.getSID(), connection, Catalog.class);
Schema s = (Schema) ConnectionHelper.getPackage(connection.getSID(), connection, Schema.class);
Schema schema = (Schema) ConnectionHelper.getPackage(connection.getUiSchema(), connection, Schema.class);
if (c != null) {
PackageHelper.addMetadataTable(tablesFromEMF, c);
} else if (s != null) {
PackageHelper.addMetadataTable(tablesFromEMF, s);
} else if (schema != null) {
PackageHelper.addMetadataTable(tablesFromEMF, schema);
} else {
Schema defaultSchema = null;
List<Schema> schemas = ConnectionHelper.getSchema(connection);
if (schemas.size() > 0) {
for (Schema sch : schemas) {
if (" ".equals(sch.getName())) {
//$NON-NLS-1$
defaultSchema = sch;
break;
}
}
}
if (defaultSchema == null) {
//$NON-NLS-1$
defaultSchema = SchemaHelper.createSchema(" ");
ConnectionHelper.addSchema(defaultSchema, connection);
}
PackageHelper.addMetadataTable(tablesFromEMF, defaultSchema);
}
}
use of orgomg.cwm.resource.relational.Schema in project tdi-studio-se by Talend.
the class MetadataColumnComparator method deleteNouseTables.
public void deleteNouseTables(Connection connection) {
if (!(connection instanceof DatabaseConnection)) {
return;
}
if (tList == null) {
return;
}
List<MetadataTable> tableList = new ArrayList<MetadataTable>(ConnectionHelper.getTables(connection));
tableList.removeAll(tList);
Catalog catalog = (Catalog) ConnectionHelper.getPackage(((DatabaseConnection) connection).getSID(), connection, Catalog.class);
Schema schema = (Schema) ConnectionHelper.getPackage(((DatabaseConnection) connection).getUiSchema(), connection, Schema.class);
String c = "";
String s = "";
if (catalog != null) {
c = catalog.getName();
}
if (schema != null) {
s = schema.getName();
}
ProjectNodeHelper.removeTablesFromCurrentCatalogOrSchema(c, s, (DatabaseConnection) connection, tableList);
tList = null;
}
use of orgomg.cwm.resource.relational.Schema in project tdi-studio-se by Talend.
the class QueryGuessCommand method getDefaultSchema.
private String getDefaultSchema(String realTableName) {
String schema = "";
List<Schema> schemas = ConnectionHelper.getSchema(this.conn);
Set<Catalog> catalog = ConnectionHelper.getAllCatalogs(this.conn);
for (Schema deScha : schemas) {
for (ModelElement ele : deScha.getOwnedElement()) {
String childeleName = TalendTextUtils.addQuotesWithSpaceFieldForSQLStringForce(TalendTextUtils.declareString(ele.getName()), dbType, true);
if (childeleName.startsWith(TalendTextUtils.QUOTATION_MARK) && childeleName.endsWith(TalendTextUtils.QUOTATION_MARK) && childeleName.length() > 2) {
childeleName = childeleName.substring(1, childeleName.length() - 1);
}
if (childeleName.equals(realTableName)) {
return deScha.getName();
}
}
}
for (Catalog cata : catalog) {
for (ModelElement ele : cata.getOwnedElement()) {
if (ele instanceof Schema) {
for (ModelElement child : ((Schema) ele).getOwnedElement()) {
String childeleName = TalendTextUtils.addQuotesWithSpaceFieldForSQLStringForce(TalendTextUtils.declareString(child.getName()), dbType, true);
if (childeleName.startsWith(TalendTextUtils.QUOTATION_MARK) && childeleName.endsWith(TalendTextUtils.QUOTATION_MARK) && childeleName.length() > 2) {
childeleName = childeleName.substring(1, childeleName.length() - 1);
}
if (childeleName.equals(realTableName)) {
return ele.getName();
}
}
} else {
String eleName = TalendTextUtils.addQuotesWithSpaceFieldForSQLStringForce(TalendTextUtils.declareString(ele.getName()), dbType, true);
if (eleName.startsWith(TalendTextUtils.QUOTATION_MARK) && eleName.endsWith(TalendTextUtils.QUOTATION_MARK) && eleName.length() > 2) {
eleName = eleName.substring(1, eleName.length() - 1);
}
if (eleName.equals(realTableName)) {
return cata.getName();
}
}
}
}
return schema;
}
Aggregations