如何用Biml动态获取源表主键及类型并生成仅含主键的Staging表

要实现自动生成仅包含源表主键字段的Staging表，且适配任意主键数据类型，核心是利用Biml的元数据查询能力，直接从源数据库获取主键列的名称和原生数据类型，而非硬编码int类型。

完整Biml脚本示例 #

<#@ template language="C#" hostspecific="true" #>
<#
// 获取已配置的源数据库连接元数据
var sourceDb = RootNode.Connections["SourceConnection"].GetDatabaseSchema();

// 遍历dbo schema下的所有表（可根据需求调整筛选条件）
foreach (var sourceTable in sourceDb.Tables.Where(t => t.Schema == "dbo"))
{
    // 获取当前表的主键约束
    var pkConstraint = sourceTable.Constraints.OfType<AstPrimaryKeyConstraintNode>().FirstOrDefault();
    if (pkConstraint != null)
    {
#>
<Table Name="<#=sourceTable.Name#>PK" SchemaName="Staging">
    <Columns>
        <#
        // 遍历主键列，动态生成列定义，完全复用源列的数据类型
        foreach (var pkColumnName in pkConstraint.Columns)
        {
            var sourceColumn = sourceTable.Columns[pkColumnName];
        #>
        <Column Name="<#=sourceColumn.Name#>" DataType="<#=sourceColumn.DataType#>" 
                <#= sourceColumn.Length.HasValue ? $"Length=\"{sourceColumn.Length.Value}\"" : "" #>
                <#= sourceColumn.Precision.HasValue ? $"Precision=\"{sourceColumn.Precision.Value}\"" : "" #>
                <#= sourceColumn.Scale.HasValue ? $"Scale=\"{sourceColumn.Scale.Value}\"" : "" #> />
        <#
        }
        #>
    </Columns>
    // 为Staging表添加主键约束，和源表一致
    <Constraints>
        <PrimaryKeyConstraint Name="PK_<#=sourceTable.Name#>PK" Columns="<#=string.Join(",", pkConstraint.Columns)#>" />
    </Constraints>
</Table>
<#
    }
}
#>

关键逻辑说明 #

1. 元数据获取：通过GetDatabaseSchema()拉取源数据库的完整结构信息，包括表、列、约束的详细属性。
2. 主键筛选：用OfType<AstPrimaryKeyConstraintNode>()定位表的主键约束，确保只处理存在主键的表。
3. 动态列生成：遍历主键列时，直接引用源列的DataType、Length、Precision、Scale属性，自动适配int、bigint、varchar、uniqueidentifier等任意数据类型。
4. 灵活调整：如果需要对特定数据类型做转换（比如将源表的nvarchar(max)转为nvarchar(4000)），可在列生成逻辑中添加条件判断，示例：

<#
var targetDataType = sourceColumn.DataType;
int? targetLength = sourceColumn.Length;

if (sourceColumn.DataType == DataType.StringMax)
{
    targetDataType = DataType.String;
    targetLength = 4000;
}
#>
<Column Name="<#=sourceColumn.Name#>" DataType="<#=targetDataType#>" Length="<#=targetLength#>" />

使用注意事项 #

• 替换脚本中的SourceConnection为你Biml项目中实际配置的源数据库连接名称。
• 调整Where(t => t.Schema == "dbo")的筛选条件，可指定单个表（如t.Name == "Address"）或其他schema的表。

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