完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

时间:2022-08-20 09:33:15

问题描述:

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

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

完美解决CodeSmith无法获取MySQL表及列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实体的模板

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

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

当然,使用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无法获取MySQL表及列Description说明注释的方案

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

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

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

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

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

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

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

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

在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的代码发现:

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

上面查询出来的每一项,下面都有获取返回,而我们刚刚添加的那个“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,再次生成。。。-_-# 我去!这是在逗我么。

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

再次回到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

完美解决CodeSmith无法获取MySQL表及列Description说明注释的方案

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

我这个处理表说明是通过截取已获得的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. 在&period;net core中完美解决多租户分库分表的问题

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

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

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

随机推荐

  1. Vimium 下载 像个 Geek 一样去浏览

    插件地址(被墙):https://chrome.google.com/webstore/detail/dbepggeogbaibhgnhhndojpepiihcmeb 本地下载:http://file ...

  2. mssql 用户只能查看授权的数据库

    问题背景:公司的一台数据库服务器上放在多个数据库,每个数据库都使用不同的登录名称,但在将项目文件发布到Ftp时,有些Ftp的信息是在客户那边的 一旦客户那边使用配置文件中的数据库信息连接到数据库他就能 ...

  3. adodb&period;RecordSet的属性和方法

    为了更精确地跟踪数据,要用RecordSet组件创建包括数据的游标,游标就是储存在内存中的数据: rs = Server.CreateObject("ADODB.RecordSet&quot ...

  4. linux下查看系统属性

    inux下查看系统属性1.查看cpu信息查看所有cpu信息:cat /proc/cpuinfo查看cpu类型: grep "model name" /proc/cpuinfo2.查 ...

  5. PHP http&lowbar;build&lowbar;query&lpar;&rpar;方法

    http_build_query (PHP 5) http_build_query -- 生成 url-encoded 之后的请求字符串描述 string http_build_query ( arr ...

  6. js indexOf within Switch

    https://*.com/questions/22277447/indexof-within-switch switch (true) { case (msgRes.inde ...

  7. JAVASCRIPT校验大全&lbrack;转&rsqb;

    var IsFireFox = document.getElementById &&! document.all;//判断是否为FireFox //页面里回车到下一控件的焦点 func ...

  8. Android之 ListView(1)

    ListView是Android中最常用的控件之一. 当有太多数据需要显示的时候,ListView就派上用场了.它允许用户通过滑动手指的方式,将数据滑入滑出界面. 一.最简单的ListView实现 1 ...

  9. Matlab中struct的用法

    struct在matlab中是用来建立结构体数组的.通常有两种用法: s = struct('field1',{},'field2',{},...)  这是建立一个空的结构体,field1,field ...

  10. iOS:二维码的扫描

    iOS 中二维码的扫描借用#import <AVFoundation/AVFoundation.h> 实现,会用到<AVCaptureMetadataOutputObjectsDel ...