/// <summary> /// 根据表名获取实体类的字符串 /// </summary> /// <param name="db"></param> /// <param name="sql"></param> /// <param name="className"></param> /// <returns></returns> public string TableNameToClass(SqlSugarClient db, string tableName) { var dt = db.GetDataTable(string.Format("select * from {0} where 1<0", tableName)); var reval = DataTableToClass(dt, tableName, null, null); return(reval); }
/// <summary> /// 根据表名获取实体类的字符串 /// </summary> /// <param name="db"></param> /// <param name="tableName">表名</param> /// <returns></returns> public string TableNameToClass(SqlSugarClient db, string tableName) { var dt = db.GetDataTable(string.Format(SqlSugarTool.GetSelectTopSql(), tableName.GetTranslationSqlName())); var tableColumns = GetTableColumns(db, tableName); var reval = DataTableToClass(dt, tableName, null, tableColumns); return(reval); }
/// <summary> /// 创建SQL实体文件,指定表名 /// </summary> public void CreateClassFilesByTableNames(SqlSugarClient db, string fileDirectory, string nameSpace, params string[] tableNames) { var tables = db.GetDataTable("select name from sysobjects where xtype in ('U','V') "); if (tables != null && tables.Rows.Count > 0) { foreach (DataRow dr in tables.Rows) { string tableName = dr["name"].ToString().ToLower(); if (tableNames.Any(it => it.ToLower() == tableName)) { var currentTable = db.GetDataTable(string.Format("select top 1 * from {0}", tableName)); var classCode = DataTableToClass(currentTable, tableName, nameSpace); FileSugar.WriteText(fileDirectory.TrimEnd('\\') + "\\" + tableName + ".cs", classCode); } } } }
/// <summary> /// 创建SQL实体文件 /// </summary> /// <param name="db"></param> /// <param name="tableOrView">是生成视图文件还是表文件,null生成表和视图,true生成表,false生成视图(默认为:null)</param> /// <param name="callBack">回调函数</param> public void CreateClassFilesInterface(SqlSugarClient db, bool?tableOrView, Action <DataTable, string, string> callBack) { var isLog = db.IsEnableLogEvent; db.IsEnableLogEvent = false; string sql = SqlSugarTool.GetCreateClassSql(tableOrView); var tables = db.GetDataTable(sql); if (tables != null && tables.Rows.Count > 0) { foreach (DataRow dr in tables.Rows) { string tableName = dr["name"].ToString(); var currentTable = db.GetDataTable(string.Format(SqlSugarTool.GetSelectTopSql(), GetTableNameWithSchema(db, tableName).GetTranslationSqlName())); string className = db.GetClassTypeByTableName(tableName); callBack(tables, className, tableName); } } db.IsEnableLogEvent = isLog; }
/// <summary> /// 创建实体文件 /// </summary> /// <param name="db"></param> /// <param name="fileDirectory"></param> /// <param name="nameSpace">命名空间(默认:system)</param> /// <param name="tableOrView">是生成视图文件还是表文件,null生成表和视图,true生成表,false生成视图(默认为:null)</param> /// <param name="callBack">生成文件后的处理,参数string为实体名</param> /// <param name="preAction">生成文件前的处理,参数string为表名</param> public void CreateClassFiles(SqlSugarClient db, string fileDirectory, string nameSpace = null, bool?tableOrView = null, Action <string> callBack = null, Action <string> preAction = null) { var isLog = db.IsEnableLogEvent; db.IsEnableLogEvent = false; string sql = SqlSugarTool.GetCreateClassSql(tableOrView); var tables = db.GetDataTable(sql); if (tables != null && tables.Rows.Count > 0) { foreach (DataRow dr in tables.Rows) { string tableName = dr["name"].ToString(); if (preAction != null) { preAction(tableName); } var currentTable = db.GetDataTable(string.Format(SqlSugarTool.GetSelectTopSql(), GetTableNameWithSchema(db, tableName).GetTranslationSqlName())); if (callBack != null) { var tableColumns = GetTableColumns(db, tableName); var classCode = DataTableToClass(currentTable, tableName, nameSpace, tableColumns); string className = db.GetClassTypeByTableName(tableName); classCode = classCode.Replace("class " + tableName, "class " + className); FileSugar.CreateFile(fileDirectory.TrimEnd('\\') + "\\" + className + ".cs", classCode, Encoding.UTF8); callBack(className); } else { var tableColumns = GetTableColumns(db, tableName); string className = db.GetClassTypeByTableName(tableName); var classCode = DataTableToClass(currentTable, className, nameSpace, tableColumns); FileSugar.CreateFile(fileDirectory.TrimEnd('\\') + "\\" + className + ".cs", classCode, Encoding.UTF8); } } } db.IsEnableLogEvent = isLog; }
/// <summary> /// 根据表获取主键 /// </summary> /// <param name="db"></param> /// <param name="tableName"></param> /// <returns></returns> internal static List <string> GetPrimaryKeyByTableNames(SqlSugarClient db, string tableName) { string key = "GetPrimaryKeyByTableName" + tableName; tableName = tableName.ToLower(); var cm = CacheManager <List <KeyValue> > .GetInstance(); List <KeyValue> primaryInfo = null; //获取主键信息 if (cm.ContainsKey(key)) { primaryInfo = cm[key]; } else { string sql = @"select DISTINCT TABLE_NAME as tableName,COLUMN_NAME as keyName from INFORMATION_SCHEMA.COLUMNS where table_name='" + tableName + "' AND COLUMN_KEY='PRI';"; var isLog = db.IsEnableLogEvent; db.IsEnableLogEvent = false; var dt = db.GetDataTable(sql); db.IsEnableLogEvent = isLog; primaryInfo = new List <KeyValue>(); if (dt != null && dt.Rows.Count > 0) { foreach (DataRow dr in dt.Rows) { primaryInfo.Add(new KeyValue() { Key = dr["tableName"].ToString().ToLower(), Value = dr["keyName"].ToString() }); } } cm.Add(key, primaryInfo, cm.Day); } //反回主键 if (!primaryInfo.Any(it => it.Key == tableName)) { return(null); } return(primaryInfo.Where(it => it.Key == tableName).Select(it => it.Value).ToList()); }
/// <summary> ///根据表名获取自添列 keyTableName Value columnName /// </summary> /// <param name="db"></param> /// <param name="tableName"></param> /// <returns></returns> internal static List <KeyValue> GetIdentitiesKeyByTableName(SqlSugarClient db, string tableName) { string key = "GetIdentityKeyByTableName" + tableName; var cm = CacheManager <List <KeyValue> > .GetInstance(); List <KeyValue> identityInfo = null; if (cm.ContainsKey(key)) { identityInfo = cm[key]; return(identityInfo); } else { string sql = string.Format(@" select TABLE_NAME as tableName,COLUMN_NAME as keyName from INFORMATION_SCHEMA.COLUMNS where table_name='" + tableName + @"' AND EXTRA='auto_increment'; ", tableName); var isLog = db.IsEnableLogEvent; db.IsEnableLogEvent = false; var dt = db.GetDataTable(sql); db.IsEnableLogEvent = isLog; identityInfo = new List <KeyValue>(); if (dt != null && dt.Rows.Count > 0) { foreach (DataRow dr in dt.Rows) { identityInfo.Add(new KeyValue() { Key = dr["tableName"].ToString().ToLower(), Value = dr["keyName"].ToString() }); } } cm.Add(key, identityInfo, cm.Day); return(identityInfo); } }