CodeSmith基础（二）

CodeSmith基础（二）

<!-- <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#" xmlns:dc="http://purl.org/dc/elements/1.1/" xmlns:trackback="http://madskills.com/public/xml/rss/module/trackback/"> <rdf:Description rdf:about="http://www.cnblogs.com/Bear-Study-Hard/archive/2005/12/19/300142.html" dc:identifier="http://www.cnblogs.com/Bear-Study-Hard/archive/2005/12/19/300142.html" dc:title="CodeSmith基础（二）" trackback:ping="http://www.cnblogs.com/Bear-Study-Hard/services/trackbacks/300142.aspx" /> </rdf:RDF> -->

本文将介绍CodeSmith与数据库进行交互生成相应的存储过程，本例使用的数据库为SQL Server 2000。
在与数据库进行交互时，我们使用到了一个CodeSmith自带的组件SchemaExplorer，利用这个组件我们可以访问数据库的数据表、存储过程、视图等，并可以得到相应的数据类型、标识列、列的（字段）名称等信息。
下面这个例子是教我们如何生成一个存储过程，虽然网上有很多一样的例子，但是我是从CodeSmith中的英文帮助中自己翻译出来的：）
使用的是SQL Server 2000自带的Northwind数据库，生成一个关于Orders订单表的更新存储过程。

第一步还是指明模板使用的语言和生成的目标语言。

<%@CodeTemplateLanguage="C#"TargetLanguage="T-SQL"Description="Generatesaupdatestoredprocedure."%>

第二步就是我们要加载使用访问数据库的组件SchemaExplorer，并声明其使用的命名空间。

<%@AssemblyName="SchemaExplorer"%>
<%@ImportNamespace="SchemaExplorer"%>

因为是针对表去生成存储过程，则首先要定义一个存储表名称使用的变量，然后指明这个变量类型为数据库中的表，这样我们可以通过这个数据表类型的变量得到相应的表的信息。

<%@PropertyName="SourceTable"Type="SchemaExplorer.TableSchema"Category="Context"Description="Tablethatthestoredproceduresshouldbebasedon."%>

如果想访问视图的话，则将变量类型Type中的SchemaExplorer.TableSchema修改为SchemaExplorer.ViewSchema即可。
得到表名的方法

CREATEPROCEDUREdbo.Update<%=SourceTable.Name%>

下面利用循环语句遍历表的各个列，拼出存储过程需要传递的参数。

<%for(inti=0;i<SourceTable.Columns.Count;i++){%>
<%=GetSqlParameterStatement(SourceTable.Columns[i])%><%if(i<SourceTable.Columns.Count-1){%>,<%}%>
<%}%>

调用的GetSqlParameterStatement方法是用来生成参数的字符串，例如生成“@CustomerID nchar(5)”，后边紧跟的if判断是用来生成参数之间相隔使用的逗号的。
生成参数字符串的方法，参数为SchemaExplorer.ColumnSchema列类型

1<scriptrunat="template">
2publicstringGetSqlParameterStatement(ColumnSchemacolumn)
3{
4stringparam="@"+column.Name+""+column.NativeType;
5
6switch(column.DataType)
7{
8caseDbType.Decimal:
9{
10param+="("+column.Precision+","+column.Scale+")";
11break;
12}
13default:
14{
15if(column.Size>0)
16{
17param+="("+column.Size+")";
18}
19break;
20}
21}
22
23returnparam;
24}
25</script>

下面来生成需要更新的字段，更新时仅能更新非主键字段的值，在SchemaExplorer中支持这种区别，使用SourceTable.NonPrimaryKeyColumns即可得到非主键字段的集合。

1UPDATE[<%=SourceTable.Name%>]SET
2<%for(inti=0;i<SourceTable.NonPrimaryKeyColumns.Count;i++){%>
3[<%=SourceTable.NonPrimaryKeyColumns[i].Name%>]=@<%=SourceTable.NonPrimaryKeyColumns[i].Name%><%if(i<SourceTable.NonPrimaryKeyColumns.Count-1){%>,<%}%>
4<%}%>

然后再使用SourceTable.PrimaryKey.MemberColumns得到数据表中的主键集合，生成更新条件

1WHERE
2<%for(inti=0;i<SourceTable.PrimaryKey.MemberColumns.Count;i++){%>
3<%if(i>0){%>AND<%}%>
4[<%=SourceTable.PrimaryKey.MemberColumns[i].Name%>]=@<%=SourceTable.PrimaryKey.MemberColumns[i].Name%>
5<%}%>

以下为整体的代码结构

1<%@CodeTemplateLanguage="C#"TargetLanguage="T-SQL"
2Description="Generatesaupdatestoredprocedure."%>
3
4<%@PropertyName="SourceTable"Type="SchemaExplorer.TableSchema"
5Category="Context"
6Description="Tablethatthestoredproceduresshouldbebasedon."%>
7
8<%@AssemblyName="SchemaExplorer"%>
9
10<%@ImportNamespace="SchemaExplorer"%>
11
12<scriptrunat="template">
13publicstringGetSqlParameterStatement(ColumnSchemacolumn)
14{
15stringparam="@"+column.Name+""+column.NativeType;
16
17switch(column.DataType)
18{
19caseDbType.Decimal:
20{
21param+="("+column.Precision+","+column.Scale+")";
22break;
23}
24default:
25{
26if(column.Size>0)
27{
28param+="("+column.Size+")";
29}
30break;
31}
32}
33
34returnparam;
35}
36</script>
37
38-----------------------------------------------------------------
39--DateCreated:<%=DateTime.Now.ToLongDateString()%>
40--CreatedBy:GeneratedbyCodeSmith
41-----------------------------------------------------------------
42
43CREATEPROCEDUREdbo.Update<%=SourceTable.Name%>
44<%for(inti=0;i<SourceTable.Columns.Count;i++){%>
45<%=GetSqlParameterStatement(SourceTable.Columns[i])%><%if(i<SourceTable.Columns.Count-1){%>,<%}%>
46<%}%>
47AS
48
49UPDATE[<%=SourceTable.Name%>]SET
50<%for(inti=0;i<SourceTable.NonPrimaryKeyColumns.Count;i++){%>
51[<%=SourceTable.NonPrimaryKeyColumns[i].Name%>]=@<%=SourceTable.NonPrimaryKeyColumns[i].Name%><%if(i<SourceTable.NonPrimaryKeyColumns.Count-1){%>,<%}%>
52<%}%>
53WHERE
54<%for(inti=0;i<SourceTable.PrimaryKey.MemberColumns.Count;i++){%>
55<%if(i>0){%>AND<%}%>
56[<%=SourceTable.PrimaryKey.MemberColumns[i].Name%>]=@<%=SourceTable.PrimaryKey.MemberColumns[i].Name%>
57<%}%>
58