最新版CodeSmith3.2（.net2.0）教程之六：SQL2005数据模板（下）

我们回顾一下上一节的内容：我们对我们要生成的数据模板进行了分析，并生成了静态的部分，接下来我们就来对动态部分进行设置。

1.建立枚举型属性

有时候我们需要用户提供的属性是特定的几种，比如endpoint 的状态只有STARTED, STOPPED, DISABLED三种，而除了这三种以外的都会引起T-SQL错误。如果这些信息都让用户手动输入的话又有可能输入错误的危险，向用户提供一选择菜单让他直接选择里面已经定义好的值就更为保险。而这就是枚举型属性。

为了建立枚举型属性，你需要定义供选择的值。首先点击CodeSmith Studio上方的菜单，选择Edit > Insert Content > Script Block.创建一种新的类型：
1

<script runat="template">
2

Public Enum StateEnum
3

STARTED
4

STOPPED
5

DISABLED
6

End Enum</script>
现在你就可以使用CodeSmith 直接定义一种属性使用这种新的类型了：


<%@ Property Name="InitialState" Type="StateEnum" Category="Options" Default="STARTED" Description="The initial state of the Web service." %>
另外你还需要添加一段代码。因为在后台，.net会把枚举型变量作为整数型来处理，而你准备插入的是字符串。为了完成这个转换，你需要在代码段中添加辅助函数：
1

Public Function GetState (ByVal State As StateEnum) As String
2

Select Case State
3

Case StateEnum.STARTED
4

GetState = "STARTED"
5

Case StateEnum.STOPPED
6

GetState = "STOPPED"
7

Case StateEnum.DISABLED
8

GetState = "DISABLED"
9

End Select
10

End Function
11


做完这个以后，就可以实现在用户对InitialState 属性进行选择是实现字符串相应，通过在模板中插入 <%= GetState(InitialState) %>实现插入相应选择字符串。接下来你自己完成authentication和port属性的辅助函数，下面是我们的完整模板：
1

<%@ CodeTemplate Language="VB" TargetLanguage="T-SQL" Description="Create an HTTP Endpoint." %>
2

<%@ Property Name="InitialState" Type="StateEnum" Category="Options" Default="STARTED" Description="The initial state of the Web service." %>
3

<%@ Property Name="Authentication" Type="AuthenticationEnum" Category="Options" Default="INTEGRATED" Description="Authentication method." %>
4

<%@ Property Name="Port" Type="PortsEnum" Category="Options" Default="CLEAR" Description="Port to use." %>
5

CREATE PROC dbo.PersonAddressTypeProc
6

AS
7

SELECT
8

AddressTypeID,
9

Name,
10

rowguid,
11

ModifiedDate
12

FROM
13

Person.AddressType
14

GO
15

CREATE ENDPOINT GetAddressType
16

STATE = <%= GetState(InitialState) %>
17

AS HTTP
18

(
19

PATH = '/AddressType',
20

AUTHENTICATION = (<%= GetAuthentication(Authentication) %>),
21

PORTS = (<%= GetPort(Port) %>),
22

SITE = 'localhost'
23

)
24

FOR SOAP
25

(
26

WEBMETHOD 'AddressTypeList'
27

(NAME='AdventureWorks.dbo.PersonAddressTypeProc'),
28

BATCHES = DISABLED,
29

WSDL = DEFAULT,
30

DATABASE = 'AdventureWorks',
31

NAMESPACE = 'http://AdventureWorks/AddressType'
32

)
33

GO
34

<script runat="template">
35

Public Enum StateEnum
36

STARTED
37

STOPPED
38

DISABLED
39

End Enum
40


41

Public Enum AuthenticationEnum
42

BASIC
43

DIGEST
44

NTLM
45

KERBEROS
46

INTEGRATED
47

End Enum
48


49

Public Enum PortsEnum
50

CLEAR
51

SSL
52

End Enum
53


54

Public Function GetState (ByVal State As StateEnum) As String
55

Select Case State
56

Case StateEnum.STARTED
57

GetState = "STARTED"
58

Case StateEnum.STOPPED
59

GetState = "STOPPED"
60

Case StateEnum.DISABLED
61

GetState = "DISABLED"
62

End Select
63

End Function
64


65

Public Function GetAuthentication (ByVal Authentication As AuthenticationEnum) As String
66

Select Case Authentication
67

Case AuthenticationEnum.BASIC
68

GetAuthentication = "BASIC"
69

Case AuthenticationEnum.DIGEST
70

GetAuthentication = "DIGEST"
71

Case AuthenticationEnum.NTLM
72

GetAuthentication = "NTLM"
73

Case AuthenticationEnum.KERBEROS
74

GetAuthentication = "KERBEROS"
75

Case AuthenticationEnum.INTEGRATED
76

GetAuthentication = "INTEGRATED"
77

End Select
78

End Function
79


80

Public Function GetPort (ByVal Port as PortsEnum) As String
81

Select Case Port
82

Case PortsEnum.CLEAR
83

GetPort = "CLEAR"
84

Case PortsEnum.SSL
85

GetPort = "SSL"
86

End Select
87

End Function
88

</script>
做好后还有一个问题又出现了：数据库的连接问题。接着往下看：

2.设置SQL属性

为了生成基于数据库表的模板，你必须让模板知道你要读取那张表。这就意味着需要通过属性提供表的元数据。而CodeSmith包含了SchemaExplorer 库，包含了一系列与数据库相关的接口。其中一个方式就是：TableSchema，它允许用户从数据库中获取一张表的信息。你可以利用SchemaExplorer库的对象模型获取你需要的数据库和表的信息。下面这个属性就是我们所需要的：


<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="Table that the Web service will access." %>



CodeSmith 本身并不知道你要使用SchemaExplorer 库的具体类型，所以我们需要告诉它需要加载包含在库里的什么组件。同时导入SchemaExplorer 命名空间也是非常有用的，它可以让我们输入更少的代码（意思是不用每次都输入该命名空间）：


<%@ Assembly Name="SchemaExplorer" %>


<%@ Import Namespace="SchemaExplorer" %>



当用户通过SchemaExplorer选择一张数据库表的时候，TableSchema 会将其组装并返回给CodeSmith。所有这些通过SchemaExplorer 对象模型都变得如此容易。


<%= SourceTable.Name %>


<%= SourceTable.Database.Name %>
你可以把上面读取到的表名称，表所属数据库名称放在你所需要的地方。然后还剩下需要为存储过程中建立一个字段名称列表。

3.编写数据库相关代码

写这个存储过程模板最为灵活的一部分就是编写一个字段名称列表了。同样要利用到SchemaExplorer。TableSchema 对象包含了字段的相关属性。你可以通过将其放在<% %>中，下面就是一个我们建立的一个字段名称列表代码：


<% For i As Integer = 0 To SourceTable.Columns.Count -1 %>


<%= SourceTable.Columns(i).Name %><% If i < SourceTable.Columns.Count - 1 Then %>,<% End If %>


<% Next %>
在这儿要注意<%%><=%%>的不同，<%%>中间放置执行代码，<=%%>放置计算表达式。你可能会把这两个弄混淆，不过没关系，你可以使用Edit > Insert Content 弹出的子菜单，选择需要的标记，就会自动帮你插入合适的标记。



下面就是我们做的完整的模板代码了：

1 <%@ CodeTemplate Language="VB" TargetLanguage="T-SQL" Debug="True" Description="Create an HTTP Endpoint." %>
2 <%@ Property Name="InitialState" Type="StateEnum" Category="Options" Default="STARTED" Description="The initial state of the Web service." %>
3 <%@ Property Name="Authentication" Type="AuthenticationEnum" Category="Options" Default="INTEGRATED" Description="Authentication method." %>
4 <%@ Property Name="Port" Type="PortsEnum" Category="Options" Default="CLEAR" Description="Port to use." %>
5 <%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="Table that the Web service will access." %>
6 <%@ Assembly Name="SchemaExplorer" %>
7 <%@ Import Namespace="SchemaExplorer" %>
8 CREATE PROC dbo.<%= SourceTable.Owner %><%= SourceTable.Name %>Proc
9 AS
10 SELECT
11 <% For i As Integer = 0 To SourceTable.Columns.Count -1 %>
12 <%= SourceTable.Columns(i).Name %><% If i < SourceTable.Columns.Count - 1 Then %>,<% End If %>
13 <% Next %>
14 FROM
15 <%= SourceTable.Name %>
16 GO
17 CREATE ENDPOINT Get<%= SourceTable.Name %>
18 STATE = <%= GetState(InitialState) %>
19 AS HTTP
20 (
21 PATH = '/<%= SourceTable.Name %>',
22 AUTHENTICATION = (<%= GetAuthentication(Authentication) %>),
23 PORTS = (<%= GetPort(Port) %>),
24 SITE = 'localhost'
25 )
26 FOR SOAP
27 (
28 WEBMETHOD '<%= SourceTable.Name %>List'
29 (NAME='<%= SourceTable.Database.Name %>.dbo.<%= SourceTable.Owner %><%= SourceTable.Name %>Proc'),
30 BATCHES = DISABLED,
31 WSDL = DEFAULT,
32 DATABASE = '<%= SourceTable.Database.Name %>',
33 NAMESPACE = 'http://<%= SourceTable.Database.Name %>/<%= SourceTable.Name %>'
34 )
35 GO
36 <script runat="template">
37 Public Enum StateEnum
38 STARTED
39 STOPPED
40 DISABLED
41 End Enum
42 Public Enum AuthenticationEnum
43 BASIC
44 DIGEST
45 NTLM
46 KERBEROS
47 INTEGRATED
48 End Enum
49 Public Enum PortsEnum
50 CLEAR
51 SSL
52 End Enum
53
54 Public Function GetState (ByVal State As StateEnum) As String
55 Select Case State
56 Case StateEnum.STARTED
57 GetState = "STARTED"
58 Case StateEnum.STOPPED
59 GetState = "STOPPED"
60 Case StateEnum.DISABLED
61 GetState = "DISABLED"
62 End Select
63 End Function
64
65 Public Function GetAuthentication (ByVal Authentication As AuthenticationEnum) As String
66 Select Case Authentication
67 Case AuthenticationEnum.BASIC
68 GetAuthentication = "BASIC"
69 Case AuthenticationEnum.DIGEST
70 GetAuthentication = "DIGEST"
71 Case AuthenticationEnum.NTLM
72 GetAuthentication = "NTLM"
73 Case AuthenticationEnum.KERBEROS
74 GetAuthentication = "KERBEROS"
75 Case AuthenticationEnum.INTEGRATED
76 GetAuthentication = "INTEGRATED"
77 End Select
78 End Function
79
80 Public Function GetPort (ByVal Port as PortsEnum) As String
81 Select Case Port
82 Case PortsEnum.CLEAR
83 GetPort = "CLEAR"
84 Case PortsEnum.SSL
85 GetPort = "SSL"
86 End Select
87 End Function
88 </script>

4.测试最终结果

现在你已经完成了模板，你可以在CodeSmith Studio测试结果。首先你需要编译模板，CodeSmith 会弹出属性窗口，让你选择相应属性，填好后，点击生成按钮或者按Ctrl+Shift+B快捷键就可以生成了。假如没有错误的话，你就可以在输出窗口看到输出成功的消息了：

---------------- Compile started -----------------
Build complete -- 0 errors, 0 warnings
---------------------- Done ----------------------
Build succeeded

如果模板有任何错误的话，会在输出窗口显示出来。你可以双击错误信息直接撞到有问题的代码行

现在你就可以在属性窗口输入模板需要的属性值了。三个枚举型属性你会发现CodeSmith 提供了下拉菜单供你选择。



通过点击数据源属性窗口旁边的三个小点按钮选择数据源属性



你可以选择数据源里已有的数据源，你也可以选择创建一个新的数据源来选择新数据源里的数据表，选好后按选择按钮回到CodeSmith Studio。
当你设置好属性后，点运行按钮或者按F5快捷键运行模板。CodeSmith 会生成相应的代码，并在输出窗口显示，你可以保存或者复制使用了。