嘗試著寫自己的代碼生成器

  1using System;
  2using System.Collections.Generic;
  3using System.ComponentModel;
  4using System.Data;
  5using System.Drawing;
  6using System.Text;
  7using System.Windows.Forms;
  8using System.Data.SqlClient;
  9 10namespace 代碼生成器
 11 {
 12publicpartialclass Form1 : Form
 13     {
 14public Form1()
 15         {
 16             InitializeComponent();
 17         }
 18 19public DataTable ExecuteDataTable(string sql, params SqlParameter[] paramers)
 20         {
 21using (SqlConnection con = new SqlConnection(tb_Constr.Text))
 22             {
 23                 con.Open();
 24using (SqlCommand cmd = con.CreateCommand())
 25                 {
 26                     cmd.CommandText = sql;
 27                     cmd.Parameters.AddRange(paramers);
 28                     DataTable dt = new DataTable();
 29                     SqlDataAdapter adapter = new SqlDataAdapter(cmd);
 30                     adapter.Fill(dt);
 31return dt;
 32                 }
 33             }
 34         }
 35///<summary> 36/// 數(shù)據(jù)庫字段類型轉(zhuǎn)換為C#字符類型 37///</summary> 38///<param name="dataType">數(shù)據(jù)庫中的字段類型</param> 39///<returns>C#中的字段類型</returns> 40publicstaticstring StrToCsharp(string dataType)
 41         {
 42switch (dataType)
 43             { 
 44case"int":
 45return"int";
 46case"nvarchar":
 47case"varchar":
 48case"char":
 49case"nchar":
 50return"string";
 51case"bit":
 52return"bool";
 53case"datetime":
 54return"DateTime";
 55default :
 56return"object";
 57             }
 58         }
 59///<summary> 60/// 獲取連接數(shù)據(jù)庫的表名 61///</summary> 62///<param name="sender"></param> 63///<param name="e"></param> 64privatevoid btn_Bulider_Click(object sender, EventArgs e)
 65         {
 66             DataTable Table =ExecuteDataTable("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES");
 67foreach (DataRow row in Table.Rows)
 68             { 
 69string TableName=row[0].ToString();
 70               clb_TableName.Items.Add(TableName);
 71             }
 72         }
 73///<summary> 74/// 代碼生成按鈕 75///</summary> 76privatevoid btn_Bulider_Click_1(object sender, EventArgs e)
 77         {
 78//被選中的表 79foreach (string TableName in clb_TableName.CheckedItems)
 80             {
 81                     CreateModel(TableName);
 82                     CreateDAL(TableName);
 83             }
 84 85         }
 86 87privatevoid CreateDAL(string TableName)
 88         {
 89//暫時假設(shè)主鍵名就是id 90              DataTable dt_column = ExecuteDataTable("select * from INFORMATION_SCHEMA.COLUMNS where TABLE_NAME=@tablename and Column_Name<>'id'", new SqlParameter("tablename", TableName));
 91string[] ColumnNames=newstring[dt_column.Rows.Count];//保存所屬表中的字段名 92for(int i=0;i<dt_column.Rows.Count;i++)
 93            {
 94string colname=(string)dt_column.Rows[i]["COLUMN_NAME"];//獲得列名 95                ColumnNames[i]=colname;//將循環(huán)獲取的列名加入列名數(shù)組 96            }
 97string[] paramers = newstring[dt_column.Rows.Count];
 98for(int i=0;i<paramers.Length;i++)
 99             {
100                paramers[i]="@"+ColumnNames[i];
101             }
102103             StringBuilder sb = new StringBuilder();
104             sb.AppendLine("//添加操作");
105             sb.AppendLine(" partial Class " + TableName + "DAL");
106             sb.AppendLine("{");
107             sb.AppendLine("public static int AddNew(" + TableName + "model");//返回新添加字段的主鍵");108             sb.AppendLine("{");
109             sb.AppendLine("object obj= SQLHelper.ExecuteScalar(");
110             sb.AppendLine("\"insert into "+TableName+"("+string.Join(",",ColumnNames)+") output inserted.id values("+string.Join(",",paramers)+")\"");
111foreach (string colname in ColumnNames)
112             { 
113               sb.AppendLine(",new SqlParameter(\""+colname+"\",model."+colname+")");
114             }
115             sb.AppendLine(");");
116             sb.AppendLine("return Convert.ToInt32(obj);");
117             MessageBox.Show(sb.ToString());
118         }
119120///<summary>121/// 生成Model122///</summary>123///<param name="TableName">被選中的表名</param>124///<returns>model模型</returns>125privatevoid CreateModel(string TableName)
126         {
127             DataTable dt_column = ExecuteDataTable("select * from INFORMATION_SCHEMA.COLUMNS where TABLE_NAME=@tablename and Column_Name<>'id'", new SqlParameter("tablename", TableName));
128             StringBuilder sb = new StringBuilder();
129             sb.AppendLine("Class" + TableName);
130             sb.AppendLine("{");
131foreach (DataRow row in dt_column.Rows)
132             {
133string dt_Col = (string)row["COLUMN_NAME"];
134string col_Type = (string)row["DATA_TYPE"];
135string CsType = StrToCsharp(col_Type);
136                 sb.AppendLine("Public " + CsType + "" + dt_Col + "{get;set;}");
137             }
138             sb.AppendLine("}");
139//MessageBox.Show(sb.ToString());140         }
141     }
142 }