从DataTable获取字段最大长度及VB.NET适配PostgreSQL表创建问题

我完全懂你现在的处境——把Gist里的代码转成VB.NET适配PostgreSQL，用GetSchemaTable确实能拿到字段的基础定义，但它的通用ADO.NET结构没法直接完美匹配PostgreSQL的特定语法和类型，比如类型映射、约束细节这些肯定让你头疼了。下面我给你一套针对性的解决方案，直接落地到VB.NET代码里：

核心问题拆解 #

GetSchemaTable返回的是标准的元数据，但PostgreSQL有自己的专属类型体系（比如TEXT/TIMESTAMPTZ/NUMERIC），还有约束规则（主键、自增、默认值），这些都需要我们在通用元数据的基础上做针对性转换。

具体实现步骤 #

1. 建立.NET类型到PostgreSQL类型的映射字典 #

首先我们需要把DataTable里的.NET类型对应到合适的PG类型，这是最关键的一步：

Private Shared ReadOnly TypeMapping As New Dictionary(Of Type, String) From {
    {GetType(String), "TEXT"},
    {GetType(Int32), "INTEGER"},
    {GetType(Int64), "BIGINT"},
    {GetType(Decimal), "NUMERIC"},
    {GetType(DateTime), "TIMESTAMPTZ"},
    {GetType(Boolean), "BOOLEAN"},
    {GetType(Double), "DOUBLE PRECISION"},
    {GetType(Byte()), "BYTEA"}
}

注：如果需要指定VARCHAR长度，可以单独处理String类型，结合Column.MaxLength生成VARCHAR(n)格式

2. 从GetSchemaTable中提取完整列定义 #

拿到DataReader的SchemaTable后，我们要逐个字段提取类型、约束、精度等信息：

Private Function BuildColumnDefinitions(schemaTable As DataTable) As List(Of String)
    Dim columnDefs As New List(Of String)()
    PrimaryKeyColumns.Clear()
    
    For Each row As DataRow In schemaTable.Rows
        Dim columnName As String = row("ColumnName").ToString()
        Dim netType As Type = DirectCast(row("DataType"), Type)
        Dim isNullable As Boolean = Convert.ToBoolean(row("AllowDBNull"))
        Dim isPrimaryKey As Boolean = Convert.ToBoolean(row("IsKey"))
        Dim precision As Integer = If(row("NumericPrecision") Is DBNull.Value, 0, Convert.ToInt32(row("NumericPrecision")))
        Dim scale As Integer = If(row("NumericScale") Is DBNull.Value, 0, Convert.ToInt32(row("NumericScale")))
        
        ' 匹配PG类型
        Dim pgType As String = If(TypeMapping.ContainsKey(netType), TypeMapping(netType), "TEXT")
        ' 处理数值类型的精度和刻度
        If netType = GetType(Decimal) AndAlso precision > 0 Then
            pgType = $"NUMERIC({precision}, {scale})"
        End If
        
        ' 构建基础列定义
        Dim columnDef As String = $"""{columnName}"" {pgType}"
        If Not isNullable Then
            columnDef &= " NOT NULL"
        End If
        
        columnDefs.Add(columnDef)
        
        ' 收集主键字段（最后统一添加约束）
        If isPrimaryKey Then
            PrimaryKeyColumns.Add(columnName)
        End If
    Next
    
    Return columnDefs
End Function

3. 完整的建表函数 #

把上面的逻辑整合起来，加上主键约束和PostgreSQL的连接执行逻辑：

Private PrimaryKeyColumns As New List(Of String)()

Public Sub CreatePostgreSQLTable(sourceData As DataTable, tableName As String, connectionString As String)
    ' 通过DataReader获取SchemaTable
    Using reader As IDataReader = sourceData.CreateDataReader()
        Dim schemaTable As DataTable = reader.GetSchemaTable()
        
        ' 构建列定义
        Dim columnDefs As List(Of String) = BuildColumnDefinitions(schemaTable)
        
        ' 添加主键约束（如果存在主键）
        If PrimaryKeyColumns.Count > 0 Then
            Dim pkConstraint As String = $"PRIMARY KEY ({String.Join(", ", PrimaryKeyColumns.Select(Function(c) $"""{c}"""))})"
            columnDefs.Add(pkConstraint)
        End If
        
        ' 构建CREATE TABLE语句
        Dim createTableSql As String = $"CREATE TABLE IF NOT EXISTS ""{tableName}"" ({String.Join(", ", columnDefs)})"
        
        ' 执行SQL语句
        Using conn As New NpgsqlConnection(connectionString)
            conn.Open()
            Using cmd As New NpgsqlCommand(createTableSql, conn)
                cmd.ExecuteNonQuery()
            End Using
        End Using
    End Using
End Sub

额外需要注意的点 #

• 自增列处理：如果DataTable里有自增列（AutoIncrement = True），需要把PG类型改成SERIAL或者GENERATED ALWAYS AS IDENTITY，可以在BuildColumnDefinitions里判断row("IsAutoIncrement")来特殊处理。
• 特殊类型支持：比如数组、JSONB这些PG特有类型，需要在TypeMapping里添加对应映射，或者单独处理DataTable的列扩展属性。
• 标识符大小写：PostgreSQL默认对标识符大小写不敏感，但如果列名有特殊字符或需要区分大小写，一定要用双引号包裹（上面代码已经做了处理）。

这样应该能解决你从DataTable获取PG列定义的核心问题，你可以根据自己的业务需求调整类型映射和约束逻辑~

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