Search in sources :

Example 6 with GenericMetaObject

use of org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaObject in project dbeaver by serge-rider.

the class GenericTable method loadReferences.

private synchronized List<GenericTableForeignKey> loadReferences(DBRProgressMonitor monitor) throws DBException {
    if (!isPersisted() || !getDataSource().getInfo().supportsReferentialIntegrity()) {
        return new ArrayList<>();
    }
    try (JDBCSession session = DBUtils.openMetaSession(monitor, getDataSource(), "Load table relations")) {
        // Read foreign keys in two passes
        // First read entire resultset to prevent recursive metadata requests
        // some drivers don't like it
        final GenericMetaObject fkObject = getDataSource().getMetaObject(GenericConstants.OBJECT_FOREIGN_KEY);
        final List<ForeignKeyInfo> fkInfos = new ArrayList<>();
        JDBCDatabaseMetaData metaData = session.getMetaData();
        // Load indexes
        try (JDBCResultSet dbResult = metaData.getExportedKeys(getCatalog() == null ? null : getCatalog().getName(), getSchema() == null ? null : getSchema().getName(), getName())) {
            while (dbResult.next()) {
                ForeignKeyInfo fkInfo = new ForeignKeyInfo();
                fkInfo.pkColumnName = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.PKCOLUMN_NAME);
                fkInfo.fkTableCatalog = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.FKTABLE_CAT);
                fkInfo.fkTableSchema = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.FKTABLE_SCHEM);
                fkInfo.fkTableName = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.FKTABLE_NAME);
                fkInfo.fkColumnName = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.FKCOLUMN_NAME);
                fkInfo.keySeq = GenericUtils.safeGetInt(fkObject, dbResult, JDBCConstants.KEY_SEQ);
                fkInfo.updateRuleNum = GenericUtils.safeGetInt(fkObject, dbResult, JDBCConstants.UPDATE_RULE);
                fkInfo.deleteRuleNum = GenericUtils.safeGetInt(fkObject, dbResult, JDBCConstants.DELETE_RULE);
                fkInfo.fkName = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.FK_NAME);
                fkInfo.pkName = GenericUtils.safeGetStringTrimmed(fkObject, dbResult, JDBCConstants.PK_NAME);
                fkInfo.defferabilityNum = GenericUtils.safeGetInt(fkObject, dbResult, JDBCConstants.DEFERRABILITY);
                fkInfos.add(fkInfo);
            }
        }
        List<GenericTableForeignKey> fkList = new ArrayList<>();
        Map<String, GenericTableForeignKey> fkMap = new HashMap<>();
        for (ForeignKeyInfo info : fkInfos) {
            DBSForeignKeyModifyRule deleteRule = JDBCUtils.getCascadeFromNum(info.deleteRuleNum);
            DBSForeignKeyModifyRule updateRule = JDBCUtils.getCascadeFromNum(info.updateRuleNum);
            DBSForeignKeyDefferability defferability;
            switch(info.defferabilityNum) {
                case DatabaseMetaData.importedKeyInitiallyDeferred:
                    defferability = DBSForeignKeyDefferability.INITIALLY_DEFERRED;
                    break;
                case DatabaseMetaData.importedKeyInitiallyImmediate:
                    defferability = DBSForeignKeyDefferability.INITIALLY_IMMEDIATE;
                    break;
                case DatabaseMetaData.importedKeyNotDeferrable:
                    defferability = DBSForeignKeyDefferability.NOT_DEFERRABLE;
                    break;
                default:
                    defferability = DBSForeignKeyDefferability.UNKNOWN;
                    break;
            }
            if (info.fkTableName == null) {
                log.debug("Null FK table name");
                continue;
            }
            //String fkTableFullName = DBUtils.getFullyQualifiedName(getDataSource(), info.fkTableCatalog, info.fkTableSchema, info.fkTableName);
            GenericTable fkTable = getDataSource().findTable(monitor, info.fkTableCatalog, info.fkTableSchema, info.fkTableName);
            if (fkTable == null) {
                log.warn("Can't find FK table " + info.fkTableName);
                continue;
            }
            GenericTableColumn pkColumn = this.getAttribute(monitor, info.pkColumnName);
            if (pkColumn == null) {
                log.warn("Can't find PK column " + info.pkColumnName);
                continue;
            }
            GenericTableColumn fkColumn = fkTable.getAttribute(monitor, info.fkColumnName);
            if (fkColumn == null) {
                log.warn("Can't find FK table " + fkTable.getFullyQualifiedName(DBPEvaluationContext.DDL) + " column " + info.fkColumnName);
                continue;
            }
            // Find PK
            GenericPrimaryKey pk = null;
            if (!CommonUtils.isEmpty(info.pkName)) {
                pk = DBUtils.findObject(this.getConstraints(monitor), info.pkName);
                if (pk == null) {
                    log.debug("Unique key '" + info.pkName + "' not found in table " + this.getFullyQualifiedName(DBPEvaluationContext.DDL));
                }
            }
            if (pk == null) {
                Collection<GenericPrimaryKey> uniqueKeys = this.getConstraints(monitor);
                if (uniqueKeys != null) {
                    for (GenericPrimaryKey pkConstraint : uniqueKeys) {
                        if (pkConstraint.getConstraintType().isUnique() && DBUtils.getConstraintAttribute(monitor, pkConstraint, pkColumn) != null) {
                            pk = pkConstraint;
                            break;
                        }
                    }
                }
            }
            if (pk == null) {
                log.warn("Can't find unique key for table " + this.getFullyQualifiedName(DBPEvaluationContext.DDL) + " column " + pkColumn.getName());
                // Too bad. But we have to create new fake PK for this FK
                //String pkFullName = getFullyQualifiedName() + "." + info.pkName;
                pk = new GenericPrimaryKey(this, info.pkName, null, DBSEntityConstraintType.PRIMARY_KEY, true);
                pk.addColumn(new GenericTableConstraintColumn(pk, pkColumn, info.keySeq));
                // Add this fake constraint to it's owner
                this.addUniqueKey(pk);
            }
            // Find (or create) FK
            GenericTableForeignKey fk;
            if (CommonUtils.isEmpty(info.fkName)) {
                // Make fake FK name
                info.fkName = info.fkTableName.toUpperCase() + "_FK" + info.keySeq;
                fk = DBUtils.findObject(fkTable.getAssociations(monitor), info.fkName);
            } else {
                fk = DBUtils.findObject(fkTable.getAssociations(monitor), info.fkName);
                if (fk == null) {
                    log.warn("Can't find foreign key '" + info.fkName + "' for table " + fkTable.getFullyQualifiedName(DBPEvaluationContext.DDL));
                // No choice, we have to create fake foreign key :(
                }
            }
            if (fk != null && !fkList.contains(fk)) {
                fkList.add(fk);
            }
            if (fk == null) {
                fk = fkMap.get(info.fkName);
                if (fk == null) {
                    fk = new GenericTableForeignKey(fkTable, info.fkName, null, pk, deleteRule, updateRule, defferability, true);
                    fkMap.put(info.fkName, fk);
                    fkList.add(fk);
                }
                GenericTableForeignKeyColumnTable fkColumnInfo = new GenericTableForeignKeyColumnTable(fk, fkColumn, info.keySeq, pkColumn);
                fk.addColumn(fkColumnInfo);
            }
        }
        return fkList;
    } catch (SQLException ex) {
        throw new DBException(ex, getDataSource());
    }
}
Also used : JDBCSession(org.jkiss.dbeaver.model.exec.jdbc.JDBCSession) DBException(org.jkiss.dbeaver.DBException) GenericMetaObject(org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaObject) SQLException(java.sql.SQLException) JDBCDatabaseMetaData(org.jkiss.dbeaver.model.exec.jdbc.JDBCDatabaseMetaData) DBSForeignKeyModifyRule(org.jkiss.dbeaver.model.struct.rdb.DBSForeignKeyModifyRule) JDBCResultSet(org.jkiss.dbeaver.model.exec.jdbc.JDBCResultSet) DBSForeignKeyDefferability(org.jkiss.dbeaver.model.struct.rdb.DBSForeignKeyDefferability)

Example 7 with GenericMetaObject

use of org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaObject in project dbeaver by serge-rider.

the class GenericUtils method getColumn.

public static Object getColumn(GenericDataSource dataSource, String objectType, String columnId) {
    GenericMetaObject object = dataSource.getMetaObject(objectType);
    if (object == null) {
        return columnId;
    }
    GenericMetaColumn column = object.getColumn(columnId);
    if (column == null || !column.isSupported()) {
        return columnId;
    }
    return column.getColumnIdentifier();
}
Also used : GenericMetaObject(org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaObject) GenericMetaColumn(org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaColumn)

Aggregations

GenericMetaObject (org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaObject)7 DBException (org.jkiss.dbeaver.DBException)4 JDBCResultSet (org.jkiss.dbeaver.model.exec.jdbc.JDBCResultSet)4 SQLException (java.sql.SQLException)2 ArrayList (java.util.ArrayList)2 JDBCSession (org.jkiss.dbeaver.model.exec.jdbc.JDBCSession)2 DBRProgressMonitor (org.jkiss.dbeaver.model.runtime.DBRProgressMonitor)2 Matcher (java.util.regex.Matcher)1 GenericMetaColumn (org.jkiss.dbeaver.ext.generic.model.meta.GenericMetaColumn)1 JDBCDatabaseMetaData (org.jkiss.dbeaver.model.exec.jdbc.JDBCDatabaseMetaData)1 DBSForeignKeyDefferability (org.jkiss.dbeaver.model.struct.rdb.DBSForeignKeyDefferability)1 DBSForeignKeyModifyRule (org.jkiss.dbeaver.model.struct.rdb.DBSForeignKeyModifyRule)1 DBSProcedureParameterKind (org.jkiss.dbeaver.model.struct.rdb.DBSProcedureParameterKind)1