使用Easy4net編寫代碼生成器
在項目中常常要手動建立和數據庫對應的實體類，若是數據庫表比較多或者表字段比較多，那會是一個工做量很是大的事情，因此我根據本身的需求寫了一個簡單的代碼生成工具，工具使用Easy4net框架開發。mysql
下面是代碼目錄結構：git
項目開始預備步驟：sql
1. 建立項目EntityCodeBuilder數據庫
2. 引入Easy4net項目源碼app
3. 建立Entity類庫框架
4. 建立TableName和TableColumn類工具
5. 在App.config配置文件中配置數據庫鏈接信息sqlserver
TableName源碼：
using System; using System.Collections.Generic; using System.Linq; using System.Text; namespace EntityCodeBuilder.Entity { public class TableName { public string Name { get; set; } } }
TableColumn源碼：ui
using System; using System.Collections.Generic; using System.Linq; using System.Text; namespace EntityCodeBuilder.Entity { public class TableColumn { public string Name { get; set; } public string Type { get; set; } public string IsIdentity { get; set; } public string IsPrimaryKey { get; set; } public string IsNull { get; set; } } }
App.config配置文件代碼：spa
<appSettings> <add key="DbType" value="sqlserver"/> <add key="connectionString" value="Data Source=.;Initial Catalog=test;User ID=test;Password=test123;Trusted_Connection=no;Min Pool Size=10;Max Pool Size=100;"/> <!--<add key="DbType" value="mysql"/> <add key="connectionString" value="Data Source=.;port=8001;User ID=test;Password=123456;DataBase=test;Min Pool Size=10;Max Pool Size=100;"/>--> </appSettings>
UI上的代碼不細說了，主要是有幾個類的代碼：
TableHelper中的代碼，獲取數據庫全部表名：
/// <summary> /// 獲取數據庫全部表名 /// </summary> /// <param name="connection"></param> /// <returns></returns> public static List<TableName> GetTables() { SqlConnection connection = (SqlConnection)DbFactory.CreateDbConnection(AdoHelper.ConnectionString); List<TableName> tablelist = new List<TableName>(); try { if (connection.State == ConnectionState.Closed) { connection.Open(); DataTable objTable = connection.GetSchema("Tables"); foreach (DataRow row in objTable.Rows) { TableName tb = new TableName(); tb.Name = row[2].ToString(); tablelist.Add(tb); } } } catch(Exception e) { throw e; } finally { if (connection != null && connection.State == ConnectionState.Closed) { connection.Dispose(); } } return tablelist; }
TableHelper中的代碼，根據表名獲取全部列：
/// <summary> /// 獲取字段 /// </summary> /// <param name="connection"></param> /// <param name="TableName"></param> /// <returns></returns> public static List<TableColumn> GetColumnField(string TableName) { StringBuilder sb = new StringBuilder(); sb.Append(" SELECT a.name,"); sb.Append(" b.name as type,"); sb.Append(" CASE COLUMNPROPERTY(a.id,a.name,'IsIdentity') WHEN 1 THEN '√' ELSE '' END as IsIdentity, "); sb.Append(" CASE WHEN EXISTS ( SELECT * FROM sysobjects WHERE xtype='PK' AND name IN ( SELECT name FROM sysindexes WHERE id=a.id AND indid IN ( SELECT indid FROM sysindexkeys "); sb.Append(" WHERE id=a.id AND colid IN ( SELECT colid FROM syscolumns WHERE id=a.id AND name=a.name ) ) ) ) THEN '√' ELSE '' END as IsPrimaryKey,"); sb.Append(" CASE a.isnullable WHEN 1 THEN '√' ELSE '' END as IsNull "); sb.Append(" FROM syscolumns a "); sb.Append(" LEFT JOIN systypes b ON a.xtype=b.xusertype "); sb.Append(" INNER JOIN sysobjects c ON a.id=c.id AND c.xtype='U' AND c.name<>'dtproperties' "); sb.Append(" LEFT JOIN syscomments d ON a.cdefault=d.id "); sb.Append(" WHERE c.name = '").Append(TableName).Append("' "); sb.Append(" ORDER BY c.name, a.colorder"); //使用Easy4net框架查詢數據 List<TableColumn> list = db.FindBySql<TableColumn>(sb.ToString()); return list; }
知道表名，字段名稱，字段類型後，要生成實體類，須要根據數據庫中的數據類型轉換爲C#中的類型：
using System; using System.Collections.Generic; using System.Linq; using System.Text; namespace WindowsDemo { public class TypeHelper { public static string GetType(string type) { string newType = "String"; switch (type) { case "varchar": case "varchar2": case "nvarchar": case "char": newType = "String"; break; case "int": case "integer": case "bit": case "smallint": newType = "int"; break; case "long": case "bitint": newType = "long"; break; case "date": case "datetime": case "datetime2": case "datetimeoffset": newType = "DateTime"; break; case "decimal": case "number": case "money": case "numeric": newType = "Decimal"; break; case "double": newType = "double"; break; case "float": newType = "float"; break; } return newType; } } }
下面就是生成代碼的具體步驟了：
using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.IO; using EntityCodeBuilder.Entity; namespace WindowsDemo { public class CreateFileHelper { /// <summary> /// 建立類文件目錄和文件 /// </summary> /// <param name="tables">全部表</param> /// <param name="fileDir">文件目錄</param> public static void Create(List<TableName> tables, string fileDir) { CreateDirectory(fileDir); foreach (TableName table in tables) { //實體類名稱 string entityName = GenVarName(table.Name); //實體類文件名 string filePath = fileDir + entityName + ".cs"; //文件是否存在 bool exists = File.Exists(filePath); //建立文件 FileStream fs = new FileStream(filePath, exists ? FileMode.Open : FileMode.Create, FileAccess.Write); StreamWriter sw = new StreamWriter(fs); //生成代碼 string code = CreateFileHelper.BuilderCode(table.Name); //寫入代碼到文件 sw.WriteLine(code); sw.Close(); fs.Close(); } } /// <summary> /// 建立文件目錄 /// </summary> /// <param name="targetDir"></param> private static void CreateDirectory(string targetDir) { DirectoryInfo dir = new DirectoryInfo(targetDir); if (!dir.Exists) dir.Create(); } /// <summary> /// 根據表名，生成代碼 /// </summary> /// <param name="tableName">表名</param> /// <returns></returns> public static string BuilderCode(string tableName) { string entityName = GenVarName(tableName); StringBuilder sb = new StringBuilder(); sb.Append("using System;").Append("\n"); sb.Append("using System.Collections.Generic; ").Append("\n"); sb.Append("using System.Linq; ").Append("\n"); sb.Append("using System.Text; ").Append("\n"); sb.Append("using System.Text; ").Append("\n"); sb.Append("namespace Easy4net.Entity ").Append("\n"); sb.Append("{ ").Append("\n"); sb.Append("\t [Table(Name = \"").Append(tableName).Append("\")] ").Append("\n"); sb.Append("\t public class ").Append(entityName).Append("\n"); sb.Append("\t { ").Append("\n"); List<TableColumn> columns = TableHelper.GetColumnField(tableName); foreach (TableColumn column in columns) { string type = TypeHelper.GetType(column.Type); if (column.IsPrimaryKey == "√") { //[Id(Name = "UserID", Strategy = GenerationType.INDENTITY)] string strategy = "GUID"; if (column.IsIdentity == "√") { strategy = "INDENTITY"; } sb.Append("\t\t").Append("[Id(Name = \"").Append(column.Name).Append("\", Strategy = GenerationType.").Append(strategy).Append(")]").Append("\n"); } else { sb.Append("\t\t").Append("[Column(Name = \"").Append(column.Name).Append("\")]").Append("\n"); } string fieldName = GenVarName(column.Name); sb.Append("\t\t").Append("public ").Append(type).Append(" ").Append(fieldName).Append("{ get; set; } \n\n"); } sb.Append("\t } ").Append("\n"); sb.Append("} ").Append("\n"); return sb.ToString(); } /// <summary> /// 將數據庫中變量名改成駝峯命名 /// 如 user_name 改成 UserName /// </summary> /// <param name="name">變量名</param> /// <returns></returns> public static string GenVarName(string name) { string first = name.Substring(0, 1); name = name.Substring(1, name.Length - 1); name = first.ToUpper() + name; int index = name.IndexOf("_"); while (index != -1) { if (name.Length >= index + 2) { first = name.Substring(index + 1, 1); string start = name.Substring(0, index); string end = name.Substring(index + 2, name.Length - index - 2); name = start + first.ToUpper() + end; index = name.IndexOf("_"); } } name = name.Replace("_", ""); return name; } } }
完成上面代碼，這個工具的核心功能基本就完成了。
源碼下載地址：點擊下載
相關文章
- 1. 手寫代碼生成器
- 2. jhipster代碼生成器使用
- 3. 代碼生成器的使用
- 4. [NHibernate]代碼生成器的使用
- 5. 初步使用jeesite代碼生成器
- 6. mybatis generator代碼生成器使用
- 7. MyBatis-Plus3.1.0 代碼生成器的使用
- 8. 安卓代碼生成器的使用
- 9. 動軟代碼生成器的使用
- 10. 使用 MyBatis-Plus 代碼生成器
- 更多相關文章...
- • XML 編碼 - XML 教程
- • Markdown 代碼 - Markdown 教程
- • IntelliJ IDEA代碼格式化設置
- • Composer 安裝與使用
相關標籤/搜索
每日一句
-
每一个你不满意的现在，都有一个你没有努力的曾经。
歡迎關注本站公眾號,獲取更多信息
相關文章
- 1. 手寫代碼生成器
- 2. jhipster代碼生成器使用
- 3. 代碼生成器的使用
- 4. [NHibernate]代碼生成器的使用
- 5. 初步使用jeesite代碼生成器
- 6. mybatis generator代碼生成器使用
- 7. MyBatis-Plus3.1.0 代碼生成器的使用
- 8. 安卓代碼生成器的使用
- 9. 動軟代碼生成器的使用
- 10. 使用 MyBatis-Plus 代碼生成器