c# - 通过 GetSchema 确定主键列

Question

有没有办法使用 ADO.NET 方法确定列是否是主键GetSchema？

这是我到目前为止所得到的：

public IEnumerable<DbColumnInfo> GetColumns(string providerName, string connectionString, string tableName)
{
    DbProviderFactory factory = DbProviderFactories.GetFactory(providerName);
    using (DbConnection connection = factory.CreateConnection())
    using (DbCommand command = factory.CreateCommand())
    {
        connection.ConnectionString = connectionString;
        connection.Open();
        command.Connection = connection;

        var columns = connection.GetSchema("Columns", tableName.Split('.'));
        foreach (DataRow row in columns.Rows)
        {
            yield return new DbColumnInfo()
            {
                Name = row.Field<string>(3),
                OrdinalPosition = row.Field<short>(4),
                DataType = this.FormatDataType(row),
                IsNullable = string.Equals(row.Field<string>(6), "yes", StringComparison.InvariantCultureIgnoreCase),
                IsPrimaryKey = // ... ?
            };
        }
    }
}

score 4 · Accepted Answer

恐怕您无法确定 connection.GetSchema() ...

但作为一种解决方法，您可以尝试使用 dataadapter，如果它适合您：

    var da = factory.CreateDataAdapter();
    command.CommandText = "select * from Employees";
    da.SelectCommand = command;
    da.MissingSchemaAction = MissingSchemaAction.AddWithKey;

    var dtab = new DataTable();
    da.FillSchema(dtab, SchemaType.Source);

    foreach (DataColumn col in dtab.Columns)
    {
        string name = col.ColumnName;
        bool isNull = col.AllowDBNull;
        bool isPrimary = dtab.PrimaryKey.Contains(col);
    }

score 3 · Accepted Answer

自从qes询问以来，我目前使用的（自从Laszlo提供了他的工作解决方案以来，经过几个版本重构代码之后）并不像我希望的那样优雅，但它更有效并且满足了我的需求。基本上，我提供了一个DbInfoProvider类，该类旨在为给定的提供者生产DbTableInfo和DbColumnInfo服务。SqlDbInfoProvider仅用于获取 SQL Server 的列信息：

public IEnumerable<DbColumnInfo> GetColumns(string connectionString, DbTableInfo table)
{
    DbProviderFactory factory = DbProviderFactories.GetFactory(this.providerName);
    using (DbConnection connection = factory.CreateConnection())
    using (DbCommand command = factory.CreateCommand())
    {
        connection.ConnectionString = connectionString;
        connection.Open();
        command.Connection = connection;
        command.CommandText = ColumnInfoQuery;
        command.CommandType = CommandType.Text;
        var tableSchema = factory.CreateParameter();
        tableSchema.ParameterName = "@tableSchema";
        tableSchema.DbType = DbType.String;
        tableSchema.Value = table.Schema;
        command.Parameters.Add(tableSchema);
        var tableName = factory.CreateParameter();
        tableName.ParameterName = "@tableName";
        tableName.DbType = DbType.String;
        tableName.Value = table.Name;
        command.Parameters.Add(tableName);

        var dataTable = new DataTable();
        using (var reader = command.ExecuteReader())
        {
            while (reader.Read())
            {
                yield return new DbColumnInfo()
                {
                    Name = reader.GetString(0),
                    OrdinalPosition = reader.GetInt32(1),
                    DataType = reader.GetString(2),
                    IsNullable = reader.GetBoolean(3),
                    IsPrimaryKey = reader.GetBoolean(4),
                    IsForeignKey = reader.GetBoolean(5),
                    IsUnique = reader.GetBoolean(6),
                };
            }
        }
    }
}

ColumnInfoQuery像这样的静态字符串在哪里：

SELECT c.[column_name]
     , CAST(c.[ordinal_position] AS int) [ordinal_position]
     , CASE WHEN c.[data_type] IN ( 'bit', 'date', 'datetime', 'smalldatetime', 'int', 'bigint', 'smallint', 'tinyint', 'real', 'money', 'smallmoney', 'image', 'text', 'ntext' )  THEN c.[data_type]
            WHEN c.[character_maximum_length] IS NOT NULL
            THEN c.[data_type] + '(' + CAST(c.[character_maximum_length] AS VARCHAR(30)) + ')'
            WHEN c.[datetime_precision] IS NOT NULL 
            THEN c.[data_type] + '(' + CAST(c.[datetime_precision] AS VARCHAR(30)) + ')'
            WHEN c.[numeric_scale] IS NOT NULL 
            THEN c.[data_type] + '(' + CAST(c.[numeric_precision] AS VARCHAR(30)) + ',' + CAST(c.[numeric_scale] AS VARCHAR(30)) + ')'
            WHEN c.[numeric_precision] IS NOT NULL 
            THEN c.[data_type] + '(' + CAST(c.[numeric_precision] AS VARCHAR(30)) + ')'
            ELSE c.[data_type]
       END [data_type]
     , CAST(MAX(CASE c.[is_nullable] WHEN 'YES' THEN 1 ELSE 0 END) AS bit) [is_nullable]
     , CAST(MAX(CASE WHEN pk.[constraint_type] = 'PRIMARY KEY' THEN 1 ELSE 0 END) AS bit) [is_primary_key]
     , CAST(MAX(CASE WHEN pk.[constraint_type] = 'FOREIGN KEY' THEN 1 ELSE 0 END) AS bit) [is_foreign_key]
     , CAST(MAX(CASE WHEN pk.[constraint_type] = 'FOREIGN KEY' THEN 0 ELSE 1 END) AS bit) [is_unique]
FROM information_schema.columns c
LEFT JOIN information_schema.constraint_column_usage ccu 
        ON c.[column_name] = ccu.[column_name] 
        AND c.[table_name] = ccu.[table_name] 
        AND c.[table_schema] = ccu.[table_schema]
        AND c.[table_catalog] = ccu.[table_catalog]
LEFT JOIN information_schema.table_constraints pk 
        ON pk.[constraint_name] = ccu.[constraint_name]
        AND pk.[table_name] = ccu.[table_name] 
        AND pk.[constraint_schema] = ccu.[table_schema]
        AND pk.[constraint_catalog] = ccu.[table_catalog]
        AND pk.[constraint_type] IN ( 'PRIMARY KEY', 'FOREIGN KEY', 'UNIQUE' )
WHERE c.[table_schema] = @tableSchema
        AND c.[table_name] = @tableName
GROUP BY c.[table_schema], c.[table_name], c.[column_name], c.[ordinal_position]
       , c.[data_type], c.[character_maximum_length], c.[datetime_precision]
       , c.[numeric_precision], c.[numeric_scale], c.[is_nullable]

score 2 · Accepted Answer

Yes, you can determine which columns are primary keys by requesting the schema for the Indexes. Then search the indexes for the specific colum / table

DataTable indexes = conn.GetSchema("Indexes");
List<string> PrimaryKeys = new List<string>();
foreach (DataRow row in indexes.Rows)
  if (Convert.ToBoolean(row["PRIMARY_KEY"]))
    PrimaryKeys.Add(row["TABLE_NAME"] + "." + row["COLUMN_NAME"]);

the PrimaryKeys will contain the list of primary keys in the database. simply check if your [table].[column] is in this list.

score 0 · Accepted Answer

至少对于 Oracle ODP.Net 提供程序，GetSchema() 有一个“PrimaryKeys”选项。我建议您使用不带参数的 GetSchema() 来查看为您的实现定义了哪些其他调用。此代码检索描述 Oracle 提供程序的其他已实现 GetSchema() 调用的 GetSchema() 元定义。

    private void GetSchemaMetaInfo(DbConnection connection)
    {
        var metaDataCollections = connection.GetSchema("MetaDataCollections");
        var dataSourceInformation = connection.GetSchema("DataSourceInformation");
        var dataTypes = connection.GetSchema("DataTypes");
        var restrictions = connection.GetSchema("Restrictions");
        var reservedWords = connection.GetSchema("ReservedWords");
    }

score 0 · Accepted Answer

这适用于 SQLITE 和可能的其他数据库：-

            string[] restrictions = new string[] { null, null,   strTable };
            DataTable tableInfo = Connection.GetSchema("IndexColumns", restrictions);

            if (tableInfo == null)
                throw new Exception("TableInfo null Error");

            foreach (DataRow test in tableInfo.Rows)
            {
                Console.WriteLine(test["column_name"]);             
            }

score 0 · Accepted Answer

可以使用 PrimaryKeys 集合来确定特定表的主键，然后检索相应的索引名称，然后从 IndexColumns 集合中获取构成索引和主键的列。

String pkIndxNm = null;
List<String> lstPkColNms = new List<String>();

DataTable dt = dbConn.GetSchema("PrimaryKeys", new[] { owner, tblNm });
if (dt.Rows.Count == 1) {
    DataRow pkRow = dt.Rows[0];
    pkIndxNm = pkRow["INDEX_NAME"] as String;

    // Now use the IndexColumns collection to pick up the names of the 
    // columns which constitute the primary key.
    //
    dt = dbConn.GetSchema("IndexColumns", new[] { owner, pkIndxNm });
    foreach (DataRow icRow in dt.Rows) {
        String colNm = icRow["COLUMN_NAME"] as String;
        lstPkColNms.Add(colNm);
    }
}

这适用于使用 System.Data.Common.DbConnection 类和底层 System.Data.OracleClient 连接的 Oracle 数据库。

对于其他数据库，您可能需要检查 Restrictions 集合以了解可以作为参数传递的内容。GetSchema 限制的文档似乎毫无用处。我发现仅转储 Restrictions 集合的 DataTable 是了解如何最好地查询每个集合的最佳方式：

  DataTable dt = dbConn.GetSchema("Restrictions");
  AppLog.Log.Info("CollectionName | RestrictionName | ParameterName | " +
                  "RestrictionDefault | RestrictionNumber");
  AppLog.Log.Info(" ");
  foreach (DataRow r in dt.Rows) {
      String s = r["CollectionName"] as String;
      s += " | " + r["RestrictionName"] as String;
      s += " | " + r["ParameterName"] as String;
      s += " | " + r["RestrictionDefault"] as String;
      s += " | " + r["RestrictionNumber"].ToString();
      AppLog.Log.Info(s);
  }

这是我得到的，从中我知道在数组的第一个和第二个位置传递所有者和 indexName 以将其约束到正确的索引。

用户 | 用户名 | 姓名 | 用户名 | 1

表 | 所有者 | 所有者 | 所有者 | 1

表 | 表 | 表名 | 表名 | 2

专栏 | 所有者 | 所有者 | 所有者 | 1

专栏 | 表 | 表名 | 表名 | 2

专栏 | 专栏 | 栏名 | COLUMN_NAME | 3

意见 | 所有者 | 所有者 | 所有者 | 1

意见 | 查看 | 视图名 | VIEW_NAME | 2

同义词 | 所有者 | 所有者 | 所有者 | 1

同义词 | 同义词 | 同义词 | SYNONYM_NAME | 2

序列 | 所有者 | 所有者 | SEQUENCE_OWNER | 1

序列 | 序列 | 序列 | SEQUENCE_NAME | 2

程序参数 | 所有者 | 所有者 | 所有者 | 1

程序参数 | 对象名 | 对象名 | 对象名 | 2

功能 | 所有者 | 所有者 | 所有者 | 1

功能 | 姓名 | 姓名 | 对象名 | 2

索引列 | 所有者 | 所有者 | 索引所有者 | 1

索引列 | 姓名 | 姓名 | 索引名称 | 2

索引列 | 桌主 | 桌主 | 表所有者 | 3

索引列 | 表名 | 表名 | 表名 | 4

索引列 | 专栏 | 栏名 | COLUMN_NAME | 5

索引 | 所有者 | 所有者 | 所有者 | 1

索引 | 姓名 | 姓名 | 索引名称 | 2

索引 | 桌主 | 桌主 | 表所有者 | 3

索引 | 表名 | 表名 | 表名 | 4

包 | 所有者 | 所有者 | 所有者 | 1

包 | 姓名 | 包名 | 对象名 | 2

封装体 | 所有者 | 所有者 | 所有者 | 1

封装体 | 姓名 | 姓名 | 对象名 | 2

论据 | 所有者 | 所有者 | 所有者 | 1

论据 | 包名 | 包名 | 包名 | 2

论据 | 对象名 | 对象名 | 对象名 | 3

论据 | 参数名称 | 参数名称 | ARGUMENT_NAME | 4

程序 | 所有者 | 所有者 | 所有者 | 1

程序 | 姓名 | 姓名 | 对象名 | 2

唯一键 | 所有者 | 所有者 | 所有者 | 1

唯一键 | 表名 | 表名 | 表名 | 2

唯一键 | 约束名称 | 约束名称 | 约束_名称 | 3

主键 | 所有者 | 所有者 | 所有者 | 1

主键 | 表名 | 表名 | 表名 | 2

主键 | 约束名称 | 约束名称 | 约束_名称 | 3

外键 | Foreign_Key_Owner | 所有者 | FKCON.OWNER | 1

外键 | Foreign_Key_Table_Name | 表名 | FKCON.TABLE_NAME | 2

外键 | Foreign_Key_Constraint_Name | 约束名称 | FKCON.CONSTRAINT_NAME | 3

外键列 | 所有者 | 所有者 | FKCOLS.OWNER | 1

外键列 | 表名 | 表名 | FKCOLS.TABLE_NAME | 2

外键列 | 约束名称 | 约束名称 | FKCOLS.CONSTRAINT_NAME | 3

c# - 通过 GetSchema 确定主键列

6 回答 6

Related

Reference