用CodeSmith生成数据库实体类的代码

用CodeSmith生成数据库实体类的代码.

在工作中,经常中与数据库层打交道,所以,我一般用一个数据库实体类来代码来进行相应的操作.
而CodeSmith则是一个功能强大的代码生成模板系统,用它可以非常快的写出你自己的模板
我这个代码只是相让大家对CodeSmith如何从数据库的表来生成代码有一个了解.
模板生成的类支持VB.NET和C#语言.

代码如下:

<%@ CodeTemplate Language="VB" TargetLanguage="Text" Src="" Inherits="" Debug="False" Description="Template description here." %>
<%@ Assembly Name="System.Data" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Assembly Name="System.Design" %>
<%@ Import Namespace="System.Data" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Import NameSpace="System.IO" %>
<%@ Import NameSpace="System.Design" %>

<%@ Property Name="CurDB" Type="SchemaExplorer.DatabaseSchema" Category="Context" Description="当前操作的数据库." %>
<%@ Property Name="CurTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="选择的数据表名称." %>
<%@ Property Name="MakeLanguage" Type="ML" Default="0" Optional="False" Category="Other" Description="设置要生成的操作语言" %>
<%@ Property Name="NameSpaces" Type="String" Default="EagleFly" Category="Other" Description="类文件的名空间" %>

<%--------------------------------模板开始-------------------------------%>
<%If MakeLanguage = ML.VB Then %>
Imports System
Imports System.Data
Imports EagleFlyFCL.Data

Namespace Model

<Serializable()>Public Class <%= CurTable.Name %>Entity

Public Sub New()
'ToDo
End Sub

<%=GetMakeCode(ML.VB)%>
End Class

End Namespace
<% ElseIf MakeLanguage = ML.CSharp Then %>
using System;
using System.Data;
using EagleFlyFCL.Data;

namespace <%=NameSpaces%>.Service.Model
{

[Serializable()]
public class <%= CurTable.Name %>Entity:EntityObjectBase
{
public <%= CurTable.Name %>Entity()
{
// TODO: 在此处添加构造函数逻辑
}

<%=GetMakeCode(ML.CSharp)%>
}
}
<% End If %>
<%--------------------------------模板结束-------------------------------%>

<script runat="template">
#region "Normal"

Public Enum ML
VB = 0
CSharp = 2
End Enum

'得到指定语言生成的代码
Public Function GetMakeCode(ByVal MakeLang As ML) As String
If MakeLang = ML.VB Then
Return GetVBCode()
ElseIf MakeLang = ML.CSharp Then
Return GetCSCode()
End If
End Function

#end region

Private Function FirstToLower(ByVal Str As String) As String
Dim S As String
S = Str.Substring(0, 1).ToLower & Str.Substring(1)
Return S
End Function

Public Function GetVBCode() As String
Dim Field As ColumnSchema : Dim SB As New System.Text.StringBuilder(100)
Dim ColumnName As String : Dim ColumnType As String
Dim FL As String

For Each Field In CurTable.Columns
ColumnName = Field.Name
ColumnType = GetColumnDataType(ML.VB, Field)
FL = FirstToLower(ColumnName)

SB.Append(vbTab & "Private _" & FL & " As " & ColumnType & GetDefaultValue(ColumnType) & vbCrLf)
SB.Append(vbTab & "<ColumnName()> Public Property " & ColumnName & "() As " & ColumnType & vbCrLf)
SB.Append(vbTab & vbTab & "Get" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "Return _" & FL & vbCrLf)
SB.Append(vbTab & vbTab & "End Get" & vbCrLf)
SB.Append(vbTab & vbTab & "Set(ByVal Value As " & ColumnType & ")" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "_" & FL & " = Value" & vbCrLf)
SB.Append(vbTab & vbTab & "End Set" & vbCrLf)
SB.Append(vbTab & "End Property" & vbCrLf & vbCrLf)

Next

Return SB.ToString

End Function

Public Function GetCSCode() As String

Dim Field As ColumnSchema : Dim SB As New System.Text.StringBuilder(100)
Dim ColumnName As String : Dim ColumnType As String

For Each Field In CurTable.Columns
ColumnName = Field.Name
ColumnType = GetColumnDataType(ML.CSharp, Field)

SB.Append(vbTab & vbTab & "private " & ColumnType & " _" & ColumnName & GetDefaultValue(ColumnType) & vbCrLf)
SB.Append(vbTab & vbTab & "[ColumnName()]public " & ColumnType & " " & ColumnName & vbCrLf)
SB.Append(vbTab & vbTab & "{" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "get" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "{" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & vbTab & "return _" & ColumnName & ";" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "}" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "set" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "{" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & vbTab & "_" & ColumnName & " = value" & ";" & vbCrLf)
SB.Append(vbTab & vbTab & vbTab & "}" & vbCrLf)
SB.Append(vbTab & vbTab & "}" & vbCrLf & vbCrLf)
Next

Return SB.ToString

End Function

Public Function GetDefaultValue(ByVal ColumnType As String) As String
Select Case ColumnType
Case "Integer", "Double"
Return "=0 "
Case "int", "double"
Return "=0; "
Case "String"
Return "= """""
Case "string"
Return "= """";"
Case "DateTime"
If MakeLanguage=ML.VB Then
Return "=System.DateTime.Now "
Else
Return "=System.DateTime.Now;"
End If
Case "Boolean"
Return "=False "
Case "bool"
Return "=false; "
Case Else
If MakeLanguage=ML.VB Then
Return "=new " & ColumnType
Else
Return "=new " & ColumnType & ";"
End If
End Select
End Function

'返回字段的对应数据类别
Public Function GetColumnDataType(ByVal MakeLang As ML, ByVal ColumnField As SchemaExplorer.ColumnSchema) As String
Select Case ColumnField.NativeType.ToLower
Case "int", "tinyint", "smallint", "bigint"
If MakeLang = ML.VB Then
Return "Integer"
ElseIf MakeLang = ML.CSharp Then
Return "int"
End If
Case "decimal", "float", "money", "numeric", "smallmoney"
If MakeLang = ML.VB Then
Return "Double"
ElseIf MakeLang = ML.CSharp Then
Return "double"
End If
Case "char", "nchar", "ntext", "text", "varchar", "nvarchar"
If MakeLang = ML.VB Then
Return "String"
ElseIf MakeLang = ML.CSharp Then
Return "string"
End If
Case "smalldatetime", "timestamp", "datetime"
Return "DateTime"
Case "bit"
If MakeLang = ML.VB Then
Return "Boolean"
ElseIf MakeLang = ML.CSharp Then
Return "bool"
End If
Case "binary", "image", "varbinary"
If MakeLang = ML.VB Then
Return "Byte()"
ElseIf MakeLang = ML.CSharp Then
Return "byte[]"
End If
End Select
End Function

</script>

生成后的代码以VB.NET示例:

Imports System
Imports System.Data
Imports EagleFlyFCL.Data

Namespace ModelNamespace Model

<Serializable()>Public Class CategoriesEntityClass CategoriesEntity

Public Sub New()Sub New()
'ToDo
End Sub

Private _categoryID As Integer=0
<ColumnName()> Public Property CategoryID()Property CategoryID() As Integer
Get
Return _categoryID
End Get
Set(ByVal Value As Integer)
_categoryID = Value
End Set
End Property

Private _categoryName As String= ""
<ColumnName()> Public Property CategoryName()Property CategoryName() As String
Get
Return _categoryName
End Get
Set(ByVal Value As String)
_categoryName = Value
End Set
End Property

Private _description As String= ""
<ColumnName()> Public Property Description()Property Description() As String
Get
Return _description
End Get
Set(ByVal Value As String)
_description = Value
End Set
End Property

Private _picture As Byte()
<ColumnName()> Public Property Picture()Property Picture() As Byte()
Get
Return _picture
End Get
Set(ByVal Value As Byte())
_picture = Value
End Set
End Property

Private _startDate As DateTime=System.DateTime.Now
<ColumnName()> Public Property StartDate()Property StartDate() As DateTime
Get
Return _startDate
End Get
Set(ByVal Value As DateTime)
_startDate = Value
End Set
End Property

Private _amount As Integer=0
<ColumnName()> Public Property Amount()Property Amount() As Integer
Get
Return _amount
End Get
Set(ByVal Value As Integer)
_amount = Value
End Set
End Property

Private _isBuy As Boolean=False
<ColumnName()> Public Property IsBuy()Property IsBuy() As Boolean
Get
Return _isBuy
End Get
Set(ByVal Value As Boolean)
_isBuy = Value
End Set
End Property

Private _price As Double=0
<ColumnName()> Public Property Price()Property Price() As Double
Get
Return _price
End Get
Set(ByVal Value As Double)
_price = Value
End Set
End Property

End Class

End Namespace