C#连接MySQL时如何判断列是主键PK、外键FK还是普通列？

前置依赖 #

请先安装 MySQL 官方 .NET 驱动，推荐使用 MySqlConnector，旧版 MySql.Data 驱动的实现逻辑和下方示例兼容。

实现方案 #

方案1：查询 MySQL 系统表获取元数据 #

兼容性最强，不需要依赖本地数据对象，直接从 MySQL 的 information_schema 系统表读取元数据，适配所有 MySQL 5.7+ 及 8.x 版本。
核心逻辑：

• 主键列满足 KEY_COLUMN_USAGE.CONSTRAINT_NAME = 'PRIMARY'
• 外键列满足 KEY_COLUMN_USAGE.REFERENCED_TABLE_NAME 不为空
• 都不满足即为普通列

示例代码：

using MySqlConnector;

public enum ColumnType
{
    Normal,
    PrimaryKey,
    ForeignKey
}

public ColumnType GetColumnType(string databaseName, string tableName, string columnName)
{
    // 替换为你的业务连接字符串
    string connectionString = "server=127.0.0.1;user=root;password=你的密码;database=你的库名;";
    using var conn = new MySqlConnection(connectionString);
    conn.Open();

    var cmd = new MySqlCommand(@"
        SELECT 
            CASE 
                WHEN CONSTRAINT_NAME = 'PRIMARY' THEN 'PK'
                WHEN REFERENCED_TABLE_NAME IS NOT NULL THEN 'FK'
                ELSE 'NORMAL'
            END AS column_type
        FROM information_schema.KEY_COLUMN_USAGE 
        WHERE TABLE_SCHEMA = @dbName 
        AND TABLE_NAME = @tableName 
        AND COLUMN_NAME = @columnName
        LIMIT 1
    ", conn);
    cmd.Parameters.AddWithValue("@dbName", databaseName);
    cmd.Parameters.AddWithValue("@tableName", tableName);
    cmd.Parameters.AddWithValue("@columnName", columnName);

    var result = cmd.ExecuteScalar()?.ToString();
    return result switch
    {
        "PK" => ColumnType.PrimaryKey,
        "FK" => ColumnType.ForeignKey,
        _ => ColumnType.Normal
    };
}

注意：如果你的 MySQL 开启了大小写敏感配置，需要保证传入的库名、表名、列名和创建时的大小写完全一致。

方案2：通过 DataTable 内置属性获取 #

如果你已经在业务中查询了对应表的 DataTable，可以直接从本地对象的元数据读取属性，不需要额外发请求到数据库。
示例代码：

using MySqlConnector;
using System.Data;
using System.Linq;

public ColumnType GetColumnTypeFromDataTable(DataTable table, string columnName)
{
    var targetColumn = table.Columns[columnName];
    if (targetColumn == null)
        throw new KeyNotFoundException("指定列不存在于当前表中");

    // 判断是否为主键
    if (table.PrimaryKey.Contains(targetColumn))
        return ColumnType.PrimaryKey;

    // 判断是否为外键
    if (table.Constraints.OfType<ForeignKeyConstraint>().Any(fk => fk.Columns.Contains(targetColumn)))
        return ColumnType.ForeignKey;

    return ColumnType.Normal;
}

// 调用示例
public void UsageDemo()
{
    string connectionString = "你的MySQL连接字符串";
    using var conn = new MySqlConnection(connectionString);
    conn.Open();
    var cmd = new MySqlCommand("SELECT * FROM 你的表名 LIMIT 1", conn);
    var adapter = new MySqlDataAdapter(cmd);
    var table = new DataTable();
    // 必须调用FillSchema才能获取主键、外键的元数据
    adapter.FillSchema(table, SchemaType.Mapped);
    adapter.Fill(table);

    var columnType = GetColumnTypeFromDataTable(table, "要判断的列名");
}

方案选型建议 #

• 仅需要判断单个列属性、没有提前查询表数据的场景，优先选方案1，单次查询开销极低
• 已经在操作对应表的 DataTable 场景，优先选方案2，无额外数据库请求

内容的提问来源于stack exchange，提问作者ProWily