MySql生成.Net中Model类的存储过程

最近用Mysql,花时间写了一个过程用来生成EF的model类，有注释，挺方便，用EF类Entity自动生成的没有字段注释。

要用到的函数：

DELIMITER $$
DROP FUNCTION IF EXISTS `fun_GetCSTypeFromDbType`$$
CREATE FUNCTION `fun_GetCSTypeFromDbType`(#本函数不需要在线上执行
dbtype VARCHAR(50)
)
RETURNS VARCHAR(50)

BEGIN
DECLARE ret VARCHAR(50);
IF dbtype='nvarchar' OR dbtype='varchar' OR dbtype='text'
THEN SET ret='string';
ELSEIF dbtype='datetime' OR dbtype='date'
THEN SET ret='DateTime';
ELSEIF dbtype='bit'
THEN SET ret='bool';
ELSEIF dbtype='bigint'
THEN SET ret='long';
ELSE SET ret=dbtype;
END IF;
RETURN ret;
END$$

DELIMITER ;
/*
select fun_GetCSTypeFromDbType('bit');
*/

DELIMITER $$
DROP PROCEDURE IF EXISTS `P_GenerateModel`$$
CREATE PROCEDURE `P_GenerateModel`(##本存储过程不会在程序中调用,不需在线上执行，调用了函数 fun_GetCSTypeFromDbType 如果在其他库执行要注意,author:pukuimin
tableSchema VARCHAR(50),-- 用到的数据库
tableName VARCHAR(50) #要生成的类名
)

BEGIN
DECLARE count1_1 INT;
DECLARE retString VARCHAR(8000);
DECLARE table_name1,column_Name1,is_nullable1,data_type1,character_maximum_length1,column_comment1,Column_Key1 VARCHAR(100);
DECLARE REFERENCED_TABLE_NAME1,REFERENCED_COLUMN_NAME1 VARCHAR(100);
DECLARE pri_Column_Name,pri_data_type VARCHAR(100);
-- 需要定义接收游标数据的变量
-- 遍历数据结束标志
DECLARE done INT DEFAULT 0;
DECLARE cur CURSOR FOR SELECT table_name,column_Name,is_nullable,data_type,character_maximum_length,column_comment,Column_Key FROM information_schema.COLUMNS WHERE table_schema=tableSchema AND table_name = tableName;

DECLARE cur2 CURSOR FOR SELECT table_name,column_name,REFERENCED_TABLE_NAME,REFERENCED_COLUMN_NAME FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE WHERE table_schema=tableSchema AND table_name = tableName;

-- 将结束标志绑定到游标
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;
SET retString = ''; #初始化为空串
SET retString=CONCAT(retString,'namespace YCF.Stock.Core.Entities
{
using Abp.Domain.Entities;
using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using YCF.Stock.Core.Enum;
');
SET retString=CONCAT(retString,'    [Table("',tableName,'")]',CHAR(10));

SELECT column_Name,data_type INTO pri_Column_Name, pri_data_type FROM information_schema.COLUMNS WHERE table_schema=tableSchema AND table_name = tableName AND Column_Key='PRI' LIMIT 1;
SET retString=CONCAT(retString,'    public partial class ',tableName,' : Entity<',fun_GetCSTypeFromDbType(pri_data_type),'>',CHAR(10));
SET retString=CONCAT(retString,'    {',CHAR(10));
#set retString=CONCAT(retString,'//    priName:',pri_Column_Name,',priType:',pri_data_type,char(10));

-- 打开游标
OPEN cur;

-- 开始循环
read_loop: LOOP
-- 提取游标里的数据，这里只有一个，多个的话也一样；
FETCH cur INTO table_name1,column_Name1,is_nullable1,data_type1,character_maximum_length1,column_comment1,Column_Key1;
-- 声明结束的时候
IF done=1 THEN
LEAVE read_loop;
END IF;
-- 这里做你想做的循环的事件
#set retString=CONCAT(retString,'',CHAR(10));
SET retString=CONCAT(retString,'
/// <summary>
/// ',column_comment1,'
/// </summary>
');
IF data_type1='varchar' OR data_type1='varchar' THEN
SET retString=CONCAT(retString,'        [StringLength(',character_maximum_length1,')]
[MaxLength(',character_maximum_length1,')]',CHAR(10));
ELSEIF data_type1='date' OR data_type1='bit' THEN
SET retString=CONCAT(retString,'        [Column(TypeName = "',data_type1,'")]',CHAR(10));
END IF;
IF Column_Key1='PRI' THEN
SET retString=CONCAT(retString,'        [Column("',column_Name1,'")]',CHAR(10),'        public override ',fun_GetCSTypeFromDbType(data_type1),' Id { get; set; }',CHAR(10));
ELSE
SET retString=CONCAT(retString,'        public ',fun_GetCSTypeFromDbType(data_type1),CASE WHEN is_nullable1 ='YES' THEN '?' ELSE '' END,' ',column_Name1,' { get; set; }',CHAR(10));
END IF;

END LOOP;
-- 关闭游标
CLOSE cur;
SET done=0;#重新把done标识为0，开始第二个游标
SET retString=CONCAT(retString,'/*',CHAR(10));
-- 打开游标2
OPEN cur2;
-- 开始循环
read_loop2: LOOP
-- 提取游标里的数据，这里只有一个，多个的话也一样；
FETCH cur2 INTO table_name1,column_Name1,REFERENCED_TABLE_NAME1,REFERENCED_COLUMN_NAME1;
-- 声明结束的时候
IF done=1 THEN
LEAVE read_loop2;
END IF;
-- 这里做你想做的循环的事件
#set retString=CONCAT(retString,'',CHAR(10));
IF REFERENCED_TABLE_NAME1 IS NOT NULL THEN
SET retString=CONCAT(retString,'        public virtual ',REFERENCED_TABLE_NAME1,' ',REFERENCED_TABLE_NAME1,' { get; set; }',CHAR(10));
END IF;
END LOOP;
-- 关闭游标2
CLOSE cur2;
SET retString=CONCAT(retString,'/*',CHAR(10));
SET retString=CONCAT(retString,'#warning 因数据库表名全部自动小写，请手动调整导航属性 public virtual XXX XXX { get; set; }',CHAR(10));
SET retString=CONCAT(retString,'
}
}',CHAR(10));
SELECT retString;
END$$

DELIMITER ;
/*
call P_GenerateModel('MyDB','ItemContractStock');
*/

调用生成，复制到Model.cs中就行了。方便快捷。

不同的项目，可能要根据不同的类名、命名空间等等把固定的部分修改一下，其他的都一样，就是字段、类型和注释的生成。