解决CodeSmith不能读取MySql数据库字段说明的问题

打开项目:

C:\Program Files\CodeSmith\v5.2\Samples\Samples\Projects\CSharp\MySQLSchemaProvider

将MySQLSchemaProvider.cs中的GetExtendedProperties修改为:

 

  1           ///   <summary>
  2           ///  Gets the extended properties for a given schema object.
  3           ///   </summary>
  4           ///   <param name="connectionString"> The connection string used to connect to the target database. </param>
  5           ///   <param name="schemaObject"></param>
  6           ///   <returns></returns>
  7           public  ExtendedProperty[] GetExtendedProperties( string  connectionString, SchemaObjectBase schemaObject)
  8          {
  9              List < ExtendedProperty >  extendedProperties  =   new  List < ExtendedProperty > ();
 10 
 11               if  (schemaObject  is  ColumnSchema)
 12              {
 13                  ColumnSchema columnSchema  =  schemaObject  as  ColumnSchema;
 14 
 15                   string  commandText  =   string .Format( @" SELECT EXTRA, COLUMN_DEFAULT, COLUMN_TYPE, COLUMN_COMMENT
 16  FROM INFORMATION_SCHEMA.COLUMNS
 17  WHERE TABLE_SCHEMA = '{0}'
 18    AND TABLE_NAME = '{1}'
 19    AND COLUMN_NAME = '{2}' " , columnSchema.Table.Database.Name, columnSchema.Table.Name, columnSchema.Name);
 20 
 21                   using  (DbConnection connection  =  CreateConnection(connectionString))
 22                  {
 23                      connection.Open();
 24 
 25                      DbCommand command  =  connection.CreateCommand();
 26                      command.CommandText  =  commandText;
 27                      command.Connection  =  connection;
 28 
 29                       using  (IDataReader reader  =  command.ExecuteReader(CommandBehavior.CloseConnection))
 30                      {
 31                           while  (reader.Read())
 32                          {
 33                               string  extra  =  reader.GetString( 0 ).ToLower();
 34                               bool  columndefaultisnull  =  reader.IsDBNull( 1 );
 35                               string  columndefault  =   "" ;
 36                               if  ( ! columndefaultisnull)
 37                              {
 38                                  columndefault  =  reader.GetString( 1 ).ToUpper();
 39                              }
 40                               string  columntype  =  reader.GetString( 2 ).ToUpper();
 41                               string  columncomment  =  reader.GetString( 3 );
 42 
 43                               bool  isIdentity  =  (extra.IndexOf( " auto_increment " )  >   - 1 );
 44                              extendedProperties.Add( new  ExtendedProperty( " CS_IsIdentity " , isIdentity, columnSchema.DataType));
 45 
 46                               if  (isIdentity)
 47                              {
 48                                   /*
 49                                  MySQL auto_increment doesn't work exactly like SQL Server's IDENTITY
 50                                  I believe that auto_increment is equivalent to IDENTITY(1, 1)
 51                                  However, auto_increment behaves differently from IDENTITY when used
 52                                  with multi-column primary keys.  See the MySQL Reference Manual for details.
 53                                   */
 54                                  extendedProperties.Add( new  ExtendedProperty( " CS_IdentitySeed " ,  1 , columnSchema.DataType));
 55                                  extendedProperties.Add( new  ExtendedProperty( " CS_IdentityIncrement " ,  1 , columnSchema.DataType));
 56                              }
 57 
 58                              extendedProperties.Add( new  ExtendedProperty( " CS_ColumnDefaultIsNull " , columndefaultisnull, DbType.Boolean));
 59                              extendedProperties.Add( new  ExtendedProperty( " CS_ColumnDefault " , columndefault, DbType.String));
 60                              extendedProperties.Add( new  ExtendedProperty( " CS_ColumnType " , columntype, DbType.String));
 61                              extendedProperties.Add( new  ExtendedProperty( " CS_ColumnExtra " , extra.ToUpper(), DbType.String));
 62                              extendedProperties.Add( new  ExtendedProperty( " CS_Description " , columncomment, DbType.String));
 63                          }
 64 
 65                           if  ( ! reader.IsClosed)
 66                              reader.Close();
 67                      }
 68 
 69                       if  (connection.State  !=  ConnectionState.Closed)
 70                          connection.Close();
 71                  }
 72              }
 73               if  (schemaObject  is  TableSchema)
 74              {
 75                  TableSchema tableSchema  =  schemaObject  as  TableSchema;
 76                   string  commandText  =   string .Format( @" SHOW CREATE TABLE `{0}`.`{1}` " , tableSchema.Database.Name, tableSchema.Name);
 77 
 78                   using  (DbConnection connection  =  CreateConnection(connectionString))
 79                  {
 80                      connection.Open();
 81 
 82                      DbCommand command  =  connection.CreateCommand();
 83                      command.CommandText  =  commandText;
 84                      command.Connection  =  connection;
 85 
 86                       using  (IDataReader reader  =  command.ExecuteReader(CommandBehavior.CloseConnection))
 87                      {
 88                           while  (reader.Read())
 89                          {
 90                               string  createtable  =  reader.GetString( 1 );
 91                              extendedProperties.Add( new  ExtendedProperty( " CS_CreateTableScript " , createtable, DbType.String));
 92                          }
 93 
 94                           if  ( ! reader.IsClosed)
 95                              reader.Close();
 96                      }
 97 
 98                       if  (connection.State  !=  ConnectionState.Closed)
 99                          connection.Close();
100                  }
101              }
102 
103               return  extendedProperties.ToArray();
104          }

 

 

然后将生成的dSchemaExplorer.MySQLSchemaProvider.dll文件放入C:\Program Files\CodeSmith\v5.2\SchemaProviders

 

示例模板:

 1  <%--  
 2  Name: CSharp Model
 3  Author: ahui
 4  Description: C# Model Class 
 5  Date:  2010 - 12 - 16
 6  --%>
 7  <% @ CodeTemplate Language = " C# "  ResponseEncoding = " UTF-8 "  TargetLanguage = " Text "  Src = ""  Inherits = ""  Debug = " False "  CompilerVersion = " v3.5 "  Description = " Template description here. "   %>
 8  <% @ Assembly Name = " SchemaExplorer "   %>
 9  <% @ Import Namespace = " SchemaExplorer "   %>
10  <% @ Property Name = " SourceTable "  Type = " SchemaExplorer.TableSchema "  Category = " DataTable "  Description = " 数据表 "   %>
11  <% @ Property Name = " NameSpacce "  Type = " System.String "  Category = " Context "  Default = " PubDS.Model "  Description = " 命名空间 "   %>
12  <% @ Property Name = " Author "  Type = " System.String "  Category = " Context "  Default = " ahui "  Description = " 作者 "   %>
13  <% @ Property Name = " Description "  Type = " String "  Category = " Context "  Description = " 类说明 "  Optional = " true " %>
14 
15  < script runat = " template " >
16 
17  public   string  GetSystemType(ColumnSchema col)
18  {
19       string  typeName  =  col.SystemType.Name;
20       if  (col.AllowDBNull  &&  col.SystemType.IsValueType)
21      {
22          typeName  +=   " ? " ;
23      }
24       return  typeName;
25  }
26 
27  </ script >
28  using  System;
29  using  System.Collections.Generic;
30  using  System.Text;
31 
32  namespace   <%=  NameSpacce  %>
33  {
34       ///   <summary>
35       ///   <%= Description %>
36       ///   </summary>
37       public   class   <%=  StringUtil.ToPascalCase(SourceTable.Name)  %>
38      {
39           #region  构造函数
40           public   <%=  StringUtil.ToPascalCase(SourceTable.Name)  %> ()
41          {
42          }
43           #endregion
44          
45           #region  表字段
46  <%   foreach  (ColumnSchema col  in  SourceTable.Columns) {  %>
47           ///   <summary>
48           ///   <%= col.Description %>
49           ///   </summary>
50           public   <%=  GetSystemType(col)  %>   <%=  StringUtil.ToPascalCase(col.Name)  %>
51          {
52               get ;
53               set ;
54          }
55          
56  <%  }  %>
57           #endregion
58      }
59  }