/// <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);
}
}
