问题描述:

  CodeSmith是现在比较实用的代码生成器,但是我们发现一个问题:

  使用CodeSmith编写MySQL模板的时候,会发现一个问题:MySQL数据表中的列说明获取不到,也就是column.Description。如图:

我们打开CodeSmith编写一个简单的Model实体类的示例模板如下:

 <%--
 Name:           MySQL Model实体模板
 Author:         孤影[QQ:]
 Description:    CodeSmith连接MySQL生成Model实体模板
 --%>

 <%@ Template Language="C#" TargetLanguage="C#" ResponseEncoding="UTF-8" %>

 <%@ Assembly Name="SchemaExplorer" %>
 <%@ Import Namespace="SchemaExplorer" %>

 <%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Description="目标数据表" %>
 <%@ Property Name="ModelNamespace" Type="System.String" Description="Model实体所在的命名空间" %>

 using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
 namespace <%=ModelNamespace %>
 {
     /// <summary>
     /// <%=SourceTable.Description %>
     /// </summary>
     public class <%=SourceTable.Name %>
     {
         <%
             // 循环遍历 获取当前数据表中的所有列
             foreach(ColumnSchema column in SourceTable.Columns){
                 Response.WriteLine(string.Format("// {0}",column.Description));
                 Response.WriteLine(string.Format("public {0} {1} ;",GetCSharpVariableType(column),column.Name));
             }
         %>
     }
 }

 <script runat="template">
 // 获取指定列对应的C#数据类型
 public string GetCSharpVariableType(ColumnSchema column)
 {
     if (column.Name.EndsWith("TypeCode")) return column.Name;

     switch (column.DataType)
     {
         case DbType.AnsiString: return "string";
         case DbType.AnsiStringFixedLength: return "string";
         case DbType.Binary: return "byte[]";
         case DbType.Boolean: return "bool";
         case DbType.Byte: return "byte";
         case DbType.Currency: return "decimal";
         case DbType.Date: return "DateTime";
         case DbType.DateTime: return "DateTime";
         case DbType.Decimal: return "decimal";
         case DbType.Double: return "double";
         case DbType.Guid: return "Guid";
         case DbType.Int16: return "short";
         case DbType.Int32: return "int";
         case DbType.Int64: return "long";
         case DbType.Object: return "object";
         case DbType.SByte: return "sbyte";
         case DbType.Single: return "float";
         case DbType.String: return "string";
         case DbType.StringFixedLength: return "string";
         case DbType.Time: return "TimeSpan";
         case DbType.UInt16: return "ushort";
         case DbType.UInt32: return "uint";
         case DbType.UInt64: return "ulong";
         case DbType.VarNumeric: return "decimal";
         default:
         {
             return "__UNKNOWN__" + column.NativeType;
         }
     }
 }
 </script>

一个简单的CodeSmith生成Model实体的模板

然后我们点击生成,生成的代码如下图:

当然,使用SQL Server及其他数据库都是可以获取到的,这是为什么呢?

逼的没招没招了的时候,果断打开.NET Reflector,看看CodeSmith对SQL Server和MySQL二者,数据表生成操作的时候,有什么不同的地方,或者有什么缺少的地方。

CodeSmith中对MySQL操作的DLL组件位置是:“X:\...\CodeSmith\v7.0\SchemaProviders\SchemaExplorer.MySQLSchemaProvider.dll”

展开后,开始一个个找里面的方法,突然发现一个亮点:“GetTableColumns(string connectionString, TableSchema table);”

这个字面的意思不就是获取列数据么?打开看看。。。可惜,里面只是根据表查询所有列,并没有Description相关操作。

继续找,继续对比。。。最终终于找到问题了:

方法“GetExtendedProperties(string connectionString, SchemaObjectBase schemaObject);”里面的查询语句是:

string str = string.Format("SELECT EXTRA, COLUMN_DEFAULT, COLUMN_TYPE FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = '{0}' AND TABLE_NAME = '{1}' AND COLUMN_NAME = '{2}'", schema.Table.Database.Name, schema.Table.Name, schema.Name);

这个不就是获取Column列中的扩展属性的方法么?!

对比发现,SQL Server的dll里这个方法的下面,有返回Description,而MySQL正好没有!

二话不说,找到CodeSmith的源码包解压,翻出MySQL的项目:“X:\...\CodeSmith\v7.0\Samples\Samples\Projects\CSharp\MySQLSchemaProvider”

然后打开Visual Studio载入"MySQLSchemaProvider.csproj",有很多错误,那是因为缺少了引用,添加CodeSmith\bin里面的相关引用即可。

需要引用的组件你可以在下面两个CodeSmith安装目录中找到:

“X:\...\CodeSmith\v7.0\bin\”、“X:\...\CodeSmith\v7.0\AddIns\”

添加引用之后,错误就全部没了:

然后我们果断开始修改代码、首先找到刚刚那个获取列扩展属性的方法:

public ExtendedProperty[] GetExtendedProperties(string connectionString, SchemaObjectBase schemaObject)

然后我们根据观察SQL Server的代码,发现MySQL里面这个方法:

在SQL语句查询的时候少查询了一项数据:“COLUMN_COMMENT”,于是我们首先修改它查询的SQL语句如下:

string commandText = string.Format(@"SELECT EXTRA, COLUMN_DEFAULT, COLUMN_TYPE, COLUMN_COMMENT
                                                      FROM INFORMATION_SCHEMA.COLUMNS
                                                      WHERE TABLE_SCHEMA = '{0}' AND TABLE_NAME = '{1}' AND COLUMN_NAME = '{2}'",
                                                      columnSchema.Table.Database.Name, columnSchema.Table.Name, columnSchema.Name);

既然上面查询了,按照正常的数据查询流程,下面应该遍历读取,然后返回吧?

于是继续看,下面有一个while,正是将上面查询出来的数据返回的,我们对比SQL Server的代码发现:

上面查询出来的每一项,下面都有获取返回,而我们刚刚添加的那个“COLUMN_COMMENT”则没有进行数据获取、没有怎么办?加呗~

获取每个数据后,最后统一将封装在“extendedProperties”中,于是我们也将获取到的Description添加进去,其与步骤省略。最终修改的代码如下:

 public ExtendedProperty[] GetExtendedProperties(string connectionString, SchemaObjectBase schemaObject)
         {
             List<ExtendedProperty> extendedProperties = new List<ExtendedProperty>();

             if (schemaObject is ColumnSchema)
             {
                 ColumnSchema columnSchema = schemaObject as ColumnSchema;

                 string commandText = string.Format(@"SELECT EXTRA, COLUMN_DEFAULT, COLUMN_TYPE, COLUMN_COMMENT
                                                       FROM INFORMATION_SCHEMA.COLUMNS
                                                       WHERE TABLE_SCHEMA = '{0}' AND TABLE_NAME = '{1}' AND COLUMN_NAME = '{2}'",
                                                       columnSchema.Table.Database.Name, columnSchema.Table.Name, columnSchema.Name);

                 using (DbConnection connection = CreateConnection(connectionString))
                 {
                     connection.Open();

                     DbCommand command = connection.CreateCommand();
                     command.CommandText = commandText;
                     command.Connection = connection;

                     using (IDataReader reader = command.ExecuteReader(CommandBehavior.CloseConnection))
                     {
                         while (reader.Read())
                         {
                             ).ToLower();
                             );
                             string columndefault = "";
                             if (!columndefaultisnull)
                             {
                                 columndefault = reader.GetString().ToUpper();
                             }
                             ).ToUpper();
                             );

                             );
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.IsIdentity, isIdentity, columnSchema.DataType));

                             if (isIdentity)
                             {
                                 /*
                                 MySQL auto_increment doesn't work exactly like SQL Server's IDENTITY
                                 I believe that auto_increment is equivalent to IDENTITY(1, 1)
                                 However, auto_increment behaves differently from IDENTITY when used
                                 with multi-column primary keys.  See the MySQL Reference Manual for details.
                                 */
                                 extendedProperties.Add(, columnSchema.DataType));
                                 extendedProperties.Add(, columnSchema.DataType));
                             }

                             extendedProperties.Add(new ExtendedProperty("CS_ColumnDefaultIsNull", columndefaultisnull, DbType.Boolean)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.DefaultValue, columndefault, DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnDefault", columndefault, DbType.String)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.SystemType, columntype, DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnType", columntype, DbType.String)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnExtra", extra.ToUpper(), DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_Description", columncomment, DbType.String));
                         }

                         if (!reader.IsClosed)
                             reader.Close();
                     }

                     if (connection.State != ConnectionState.Closed)
                         connection.Close();
                 }
             }
             if (schemaObject is TableSchema)
             {
                 TableSchema tableSchema = schemaObject as TableSchema;
                 string commandText = string.Format(@"SHOW CREATE TABLE `{0}`.`{1}`", tableSchema.Database.Name, tableSchema.Name);

                 using (DbConnection connection = CreateConnection(connectionString))
                 {
                     connection.Open();

                     DbCommand command = connection.CreateCommand();
                     command.CommandText = commandText;
                     command.Connection = connection;

                     using (IDataReader reader = command.ExecuteReader(CommandBehavior.CloseConnection))
                     {
                         while (reader.Read())
                         {
                             );
                             extendedProperties.Add(new ExtendedProperty("CS_CreateTableScript", createtable, DbType.String));
                         }

                         if (!reader.IsClosed)
                             reader.Close();
                     }

                     if (connection.State != ConnectionState.Closed)
                         connection.Close();
                 }
             }

             return extendedProperties.ToArray();
         }

最终修改完成的“GetExtendedProperties”方法

然后我们F6生成一个修改后的dll组件"SchemaExplorer.MySQLSchemaProvider.dll"。

找到默认的dll:“X:\...\CodeSmith\v7.0\SchemaProviders\SchemaExplorer.MySQLSchemaProvider.dll”,替.....不行,还是先备份一下。。。哈哈

然后替换。打开重启CodeSmith,再次生成。。。-_-# 我去!这是在逗我么。

再次回到Visual Studio中仔细看看整个方法,最后发现。。服了。它的这个方法的判断逻辑是:

 public ExtendedProperty[] GetExtendedProperties(string connectionString, SchemaObjectBase schemaObject)

 {

   List<要返回的东西>......

   if(schemaObject 是一个 ColumnSchema)// 如果是一个列对象

   {

     // 这里面也就是我们刚刚改的,获取列说明部分的代码

   }

   if(schemaObject 是一个 TableSchema)// 完全没有注意下面的这个判断,如果是一个表对象!!!

   {

     // 这里也就是我们下面要动手脚的地方了。

   }

 }

废话不多说。直接上这个方法最终的代码

         public ExtendedProperty[] GetExtendedProperties(string connectionString, SchemaObjectBase schemaObject)
         {
             List<ExtendedProperty> extendedProperties = new List<ExtendedProperty>();

             if (schemaObject is ColumnSchema)
             {
                 ColumnSchema columnSchema = schemaObject as ColumnSchema;

                 string commandText = string.Format(@"SELECT EXTRA, COLUMN_DEFAULT, COLUMN_TYPE, COLUMN_COMMENT
                                                       FROM INFORMATION_SCHEMA.COLUMNS
                                                       WHERE TABLE_SCHEMA = '{0}' AND TABLE_NAME = '{1}' AND COLUMN_NAME = '{2}'",
                                                       columnSchema.Table.Database.Name, columnSchema.Table.Name, columnSchema.Name);

                 using (DbConnection connection = CreateConnection(connectionString))
                 {
                     connection.Open();

                     DbCommand command = connection.CreateCommand();
                     command.CommandText = commandText;
                     command.Connection = connection;

                     using (IDataReader reader = command.ExecuteReader(CommandBehavior.CloseConnection))
                     {
                         while (reader.Read())
                         {
                             ).ToLower();
                             );
                             string columndefault = "";
                             if (!columndefaultisnull)
                             {
                                 columndefault = reader.GetString().ToUpper();
                             }
                             ).ToUpper();
                             );

                             );
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.IsIdentity, isIdentity, columnSchema.DataType));

                             if (isIdentity)
                             {
                                 /*
                                 MySQL auto_increment doesn't work exactly like SQL Server's IDENTITY
                                 I believe that auto_increment is equivalent to IDENTITY(1, 1)
                                 However, auto_increment behaves differently from IDENTITY when used
                                 with multi-column primary keys.  See the MySQL Reference Manual for details.
                                 */
                                 extendedProperties.Add(, columnSchema.DataType));
                                 extendedProperties.Add(, columnSchema.DataType));
                             }

                             extendedProperties.Add(new ExtendedProperty("CS_ColumnDefaultIsNull", columndefaultisnull, DbType.Boolean)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.DefaultValue, columndefault, DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnDefault", columndefault, DbType.String)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty(ExtendedPropertyNames.SystemType, columntype, DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnType", columntype, DbType.String)); // Added for Backwards Compatibility.
                             extendedProperties.Add(new ExtendedProperty("CS_ColumnExtra", extra.ToUpper(), DbType.String));
                             extendedProperties.Add(new ExtendedProperty("CS_Description", columncomment, DbType.String));
                         }

                         if (!reader.IsClosed)
                             reader.Close();
                     }

                     if (connection.State != ConnectionState.Closed)
                         connection.Close();
                 }
             }
             if (schemaObject is TableSchema)
             {
                 TableSchema tableSchema = schemaObject as TableSchema;
                 string commandText = string.Format(@"SHOW CREATE TABLE `{0}`.`{1}`", tableSchema.Database.Name, tableSchema.Name);

                 using (DbConnection connection = CreateConnection(connectionString))
                 {
                     connection.Open();

                     DbCommand command = connection.CreateCommand();
                     command.CommandText = commandText;
                     command.Connection = connection;

                     using (IDataReader reader = command.ExecuteReader(CommandBehavior.CloseConnection))
                     {
                         while (reader.Read())
                         {
                             );
                             extendedProperties.Add(new ExtendedProperty("TS_Description", createtable, DbType.String));
                             int engineIndex = createtable.LastIndexOf("ENGINE");
                             int commentIndex = createtable.LastIndexOf("COMMENT=");
                             );
                             if (commentIndex > engineIndex)
                             {
                                 tableDescription = createtable.Substring(commentIndex + ).Replace("'", "");
                             }
                             extendedProperties.Add(new ExtendedProperty("CS_Description", tableDescription, DbType.String));

                         }

                         if (!reader.IsClosed)
                             reader.Close();
                     }

                     if (connection.State != ConnectionState.Closed)
                         connection.Close();
                 }
             }

             return extendedProperties.ToArray();
         }

最终的“GetExtendedProperties”方法

重新生成,替换。。。重启CodeSmith,链接MySQL生成。。。。必然果断Ok

网上当然也有很多例子,不过都是只处理了列的说明,没有处理表的说明

我这个处理表说明是通过截取已获得的CreateTableScript里面的数据,获取的表说明。

码字不容易,感觉不错的话,请不要忘了点赞哦~(*^_^ *)

【本章来自 孤影'Blog:http://www.cnblogs.com/LonelyShadow,码字不容易,转载请注明出处。】

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案的更多相关文章

  1. php实例源码之获取mysql表中所有行和列

    本文章向大家介绍php获取mysql表中所有行和列的源码,主要使用到mysql_num_rows和mysql_fetch_row等php的数据库操作函数,该实例有助于大家熟悉PHP mysql数据库编 ...

  2. 获取MySql每一列的数据类型和长度默认值等信息

    如何获取MySql表中各个列的数据类型? show columns from tablename 返回结果如下: id    int(11)     NO  PRI         auto_incr ...

  3. 完美解决phpstudy安装后mysql无法启动(无需删除原数据库,无需更改任何配置,无需更改端口)直接共存

    PHPstudy与原Mysql兼容解决 一.前言 今天学习php,当然是要先安装好运行环境了,phpstyudy是一个运行php的集成环境, 一键安装对新手很友好,与时作为一个新手,便跟着教程安装了p ...

  4. oracle数据库获取指定表的列的相关信息

    1.很多时候我们需要从数据库中获取指定表的所有列的相关属性,如 name,commens,datatype,datalength,pk等.下面就是制定的语句. select c.TABLE_NAME ...

  5. MySQL 表和列的注释

    像代码一样,可以为表以及表中的列添加注释,方便其他人知晓其功能.对于一些字段,在经过一定时间后,创建者未必也能想起其具体的含意,所以注释显得尤为重要. 注释的添加 注释的添加是通过在定义表或列的时候在 ...

  6. 【Java】自动获取某表某列的最大ID数

    使用场景: 当需要往数据库插入数据时,表的主键需要接着已经有的数据后面进行自增.比如已经wq_customer表里,主键为TBL_ID,如果是空表,那么插入的数据TBL_ID设置为1,如果已经有n条数 ...

  7. CodeSmith无法获取Oracle表注释

    如题:安装CodeSmith5.2版本,SQLServer没有任何问题,而Oracle就只能获取列的注释而不能获取表的注释,经过多方面查找资料后找到了一个最重要的解决方案,Sql语句,如下:selec ...

  8. 在.net core中完美解决多租户分库分表的问题

    前几天有人想做一个多租户的平台,每个租户一个库,可以进行水平扩展,应用端根据登录信息,切换到不同的租户库 计划用ef core实现,他们说做不出来,需要动态创建dbContext,不好实现 然而这个使 ...

  9. mysql 表及其列字符集设置

    --修改表的字符集 alter table rtb_media_daily_report character set gbk; --查询表列字符集 show full columns from rtb ...

随机推荐

  1. Online Judge(OJ)搭建(第一版)

    搭建 OJ 需要的知识(重要性排序): Java SE(Basic Knowledge, String, FileWriter, JavaCompiler, URLClassLoader, Secur ...

  2. 【原创分享·支付宝支付】HBuilder打包APP调用支付宝客户端支付

    前言 最近有点空余时间,所以,就研究了一下APP支付.前面很早就搞完APP的微信支付了,但是由于时间上和应用上的情况,支付宝一直没空去研究.然后等我空了的时候,发现支付宝居然升级了支付逻辑,虽然目前还 ...

  3. Ajax及跨域

    概念 Ajax Ajax,Asynchronous JavaScript and XML,字面意思:异步的 JavaScript 和 XML,是指一种创建交互式网页应用的网页开发技术. 用于异步地去获 ...

  4. 在Openfire上弄一个简单的推送系统

    推送系统 说是推送系统有点大,其实就是一个消息广播功能吧.作用其实也就是由服务端接收到消息然后推送到订阅的客户端. 思路 对于推送最关键的是服务端向客户端发送数据,客户端向服务端订阅自己想要的消息.这 ...

  5. 多线程的通信和同步(Java并发编程的艺术--笔记)

    1. 线程间的通信机制 线程之间通信机制有两种: 共享内存.消息传递.   2. Java并发 Java的并发采用的是共享内存模型,Java线程之间的通信总是隐式执行,通信的过程对于程序员来说是完全透 ...

  6. WPF做12306验证码点击效果

    一.效果 和12306是一样的,运行一张图上点击多个位置,横线以上和左边框还有有边框位置不允许点击,点击按钮输出坐标集合,也就是12306登陆的时候,需要向后台传递的参数. 二.实现思路 1.获取验证 ...

  7. Xamarin+Prism开发详解五:页面布局基础知识

    说实在的研究Xamarin到现在,自己就没设计出一款好的UI,基本都在研究后台逻辑之类的!作为Xamarin爱好者,一些简单的页面布局知识还是必备的. 布局常见标签: StackLayout Abso ...

  8. ubuntu如何安装nodejs最新版 本

    如何正确的安装nodejs? 我们可以先安装nvm, git clone https://github.com/creationix/nvm.git ~/.nvm 然后打开 ~/.bashrc ,   ...

  9. 简约之美Jodd-http--深入源码理解http协议

    Jodd 是一个开源的 Java 工具集, 包含一些实用的工具类和小型框架.简单,却很强大! jodd-http是一个轻巧的HTTP客户端.现在我们以一个简单的示例从源码层看看是如何实现的? Http ...

  10. GIT笔记命令行(1)

    Git简单易用,只要输入git就可以列出他的所有参数 C:\Users\spu>git usage: git [--version] [--help] [-C <path>] [-c ...