CodeSmith基础（二）

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

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

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

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

< %@Assembly Name ="SchemaExplorer" % >
< %@Import Namespace ="SchemaExplorer" % >

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

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

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

CREATE PROCEDURE dbo. Update <%= SourceTable.Name %>

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

<% for ( int i = 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 " >
2 publicstringGetSqlParameterStatement(ColumnSchemacolumn)
3 {
4 stringparam = " @ " + column.Name + " " + column.NativeType;
5
6 switch (column.DataType)
7 {
8 case DbType.Decimal:
9 {
10 param += " ( " + column.Precision + " , " + column.Scale + " ) " ;
11 break ;
12 }
13 default :
14 {
15 if (column.Size > 0 )
16 {
17 param += " ( " + column.Size + " ) " ;
18 }
19 break ;
20 }
21 }
22
23 return param;
24 }
25 </ script >

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

UPDATE [ <%=SourceTable.Name%> ] SET
2

<% for ( int i = 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得到数据表中的主键集合，生成更新条件

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

CodeSmith基础（二）

你可能感兴趣的:(数据结构,sql,sql,server)