摆脱CodeSmith,使用页面生成.CS文件
大家好,首先我要声明的是:本人不曾对CodeSmith有过任何偏见,相反,在以后我做的几个项目中,
曾大量使用它来生成一些代码。但是由于本人最近在研究IOC类型框架Castle时,为了生成符合我要求的
数据表类型CS文件,发现去定义自己的CodeSmith模版太麻烦了,因此动了使用网页ASPX生成CS文件的念
头[可能有点BT],但是在完成这个程序后发现,这个东西非常好用。虽然目前只能进行SQLSERVER2000数
据库文件生成,但相信扩展到其它数据库类型并不难,但由于本人精力有限,目前只能做到此步。相信
有使用ORACLE,MYSQL等数据库的朋友完善它。
闲话少叙,马上开说。
如何找出当前数据库中的所有表字段和类型,有许多的方法,本人只用最常用的SQL语句来完成,如
下:
SELECT 表名=case when a.colorder=1 then d.name else '' end,
表说明=case when a.colorder=1 then isnull(f.value,'') else '' end,
字段序号=a.colorder,
字段名=a.name,
标识=case when COLUMNPROPERTY( a.id,a.name,'IsIdentity')=1 then '√'else '' end,
主键=case when exists(SELECT 1 FROM sysobjects where xtype='PK' and name in ( SELECT name FROM sysindexes WHERE indid in( SELECT indid FROM sysindexkeys WHERE id = a.id AND colid=a.colid ))) then '√' else '' end,
类型=b.name,
占用字节数=a.length,
长度=COLUMNPROPERTY(a.id,a.name,'PRECISION'),
小数位数=isnull(COLUMNPROPERTY(a.id,a.name,'Scale'),0),
允许空=case when a.isnullable=1 then '√'else '' end,
默认值=isnull(e.text,''),
字段说明=isnull(g.[value],'')FROM syscolumns a left join systypes b on a.xtype=b.xusertype inner join sysobjects d on a.id=d.id and d.xtype='U' and d.name<>'dtproperties' left join syscomments e on a.cdefault=e.id left join sysproperties g on a.id=g.id and a.colid=g.smallid left join sysproperties f on d.id=f.id and f.smallid=0 order by a.id,a.colorder
得到了相应的信息之后,开始生成CS文件,相关函数代码如下:
//写入接口文件内容2
public string InterFaceFileContent(string TableName,string TableField)3
{4
string talbefield=null;5
foreach(string m_tablefield in TableField.Split(';'))6
{7
if (m_tablefield.Trim()!="") talbefield+= m_tablefield+"{get;set;}\r\n ";8
}9

10
StringBuilder content=new StringBuilder();11
content.Append("using System;\r\n");12
content.Append("using System.Data;\r\n");13
content.Append("\r\n");14
content.Append("namespace Entity\r\n");15
content.Append("{\r\n ");16
content.Append("/// <summary>\r\n ");17
content.Append("/// Entity 的摘要说明。\r\n ");18
content.Append("/// </summary>\r\n ");19
content.Append("public interface I"+TableName+"Entity\r\n ");20
content.Append("{\r\n ");21
content.Append(talbefield+"\r\n ");22
content.Append("}\r\n\r\n ");23
24
content.Append("public interface IDao_"+TableName+"Entity\r\n ");25
content.Append("{\r\n ");26
content.Append(string.Format("I{0}Entity Insert(I{0}Entity dse);\r\n ",TableName));27
content.Append(string.Format("I{0}Entity Update(I{0}Entity dse);\r\n ",TableName));28
content.Append("void Delete(string Condition);\r\n ");29
content.Append("DataTable Find(string SqlString);\r\n ");30
content.Append(string.Format("I{0}Entity Find(int id);\r\n ",TableName));31
content.Append("}\r\n");32
content.Append("}\r\n");33
return content.ToString();34
}35

36

37
//写入实体文件内容38
public string EntityFileContent(string TableName,string TableField)39
{40
string tablefield=null;41
string tablefieldname=null; //字段名,例:ID,NAME42
string insertvaluetablefield=null; //例如:ie.ID,ie.Name43
string insertvaluetableindex=null; //例如:'{1}','{2}';44
string updatevaluetablefield=null; //例如:ID='{0}',Name='{1}'45
string createfunction=null; //重载构造函数体46
int i=0;47

48
//拼接出符合上面变量格式的字符串49
foreach(string m_tablefield in TableField.Split(';'))50
{51
if (m_tablefield.Trim()!="")52
{53
StringBuilder field=new StringBuilder();54
field.Append("private "+m_tablefield.Replace(" "," m_")+";\r\n ");55
field.Append(string.Format("public {0}\r\n ",m_tablefield));56
field.Append("{\r\n ");57
field.Append("get { return m_"+m_tablefield.Split(' ')[1]+"; }\r\n ");58
field.Append("set { m_"+m_tablefield.Split(' ')[1]+" = value; }\r\n ");59
field.Append("}\r\n ");60
61
tablefield+= field.ToString()+"\r\n ";62

63
tablefieldname+=m_tablefield.Split(' ')[1]+",";64
insertvaluetablefield+="ie."+m_tablefield.Split(' ')[1]+",";65
insertvaluetableindex+="'{"+i.ToString()+"}',";66
updatevaluetablefield+=m_tablefield.Split(' ')[1]+"='{"+i.ToString()+"}',";67

68
createfunction+="m_"+m_tablefield.Split(' ')[1]+"="+m_tablefield.Split(' ')[1]+";\r\n ";69

70
i++;71
}72
}73
//去掉尾部的‘,’号74
tablefieldname=tablefieldname.Substring(0,tablefieldname.Length-1);75
insertvaluetablefield=insertvaluetablefield.Substring(0,insertvaluetablefield.Length-1);76
insertvaluetableindex=insertvaluetableindex.Substring(0,insertvaluetableindex.Length-1);77
updatevaluetablefield=updatevaluetablefield.Substring(0,updatevaluetablefield.Length-1);78

79
StringBuilder content=new StringBuilder();80
//实体类写入81
118
119
//实体操作类写入120

167

168
return content.ToString();169
}170

171

172
然后用如下代码调用上面的两个函数173

174
public void WriteCS_File(string TableName,string TableField)175
{176
//写入接口文件177
using(FileStream fs = File.Create(Server.MapPath("Test/"+TableName+"_Interface.cs"), 1024))178
{179
Byte[] info = new UTF8Encoding(true).GetBytes(InterFaceFileContent(TableName,TableField));180
// 向文件写信息181
fs.Write(info, 0, info.Length);182
fs.Close();183
}184

185
//写入实体文件186
using(FileStream fs = File.Create(Server.MapPath("Test/"+TableName+"_Entity.cs"), 1024))187
{188
Byte[] info = new UTF8Encoding(true).GetBytes(EntityFileContent(TableName,TableField));189
// 向文件写信息190
fs.Write(info, 0, info.Length);191
fs.Close();192
}193
}194

195

源代码

浙公网安备 33010602011771号