// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
//
|
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
//
|
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
|
using Admin.NET.Core.Service;
|
using DbType = SqlSugar.DbType;
|
|
namespace Admin.NET.Core;
|
|
/// <summary>
|
/// 代码生成帮助类
|
/// </summary>
|
public static class CodeGenUtil
|
{
|
/// <summary>
|
/// 转换大驼峰法命名
|
/// </summary>
|
/// <param name="columnName">字段名</param>
|
/// <param name="dbColumnNames">EntityBase 实体属性名称</param>
|
/// <returns></returns>
|
public static string CamelColumnName(string columnName, string[] dbColumnNames)
|
{
|
if (columnName.Contains('_'))
|
{
|
var arrColName = columnName.Split('_');
|
var sb = new StringBuilder();
|
foreach (var col in arrColName)
|
{
|
if (col.Length > 0)
|
sb.Append(col[..1].ToUpper() + col[1..].ToLower());
|
}
|
columnName = sb.ToString();
|
}
|
else
|
{
|
var propertyName = dbColumnNames.FirstOrDefault(c => c.ToLower() == columnName.ToLower());
|
if (!string.IsNullOrEmpty(propertyName))
|
{
|
columnName = propertyName;
|
}
|
else
|
{
|
columnName = columnName[..1].ToUpper() + columnName[1..].ToLower();
|
}
|
}
|
return columnName;
|
}
|
|
|
public static string ConvertDataType(string dataType)
|
{
|
if (string.IsNullOrEmpty(dataType)) return "";
|
if (dataType.StartsWith("System.Nullable"))
|
dataType = new Regex(@"(?i)(?<=\[)(.*)(?=\])").Match(dataType).Value; // 中括号[]里面值
|
|
switch (dataType)
|
{
|
case "System.Guid": return "Guid";
|
case "System.String": return "string";
|
case "System.Int32": return "int";
|
case "System.Int64": return "long";
|
case "System.Single": return "float";
|
case "System.Double": return "double";
|
case "System.Decimal": return "decimal";
|
case "System.Boolean": return "bool";
|
case "System.DateTime": return "DateTime";
|
case "System.DateTimeOffset": return "DateTimeOffset";
|
case "System.Byte": return "byte";
|
case "System.Byte[]": return "byte[]";
|
default:
|
break;
|
}
|
return dataType;
|
}
|
|
// 根据数据库类型来处理对应的数据字段类型
|
public static string ConvertDataType(DbColumnInfo dbColumnInfo, DbType dbType = DbType.Custom)
|
{
|
if (dbType == DbType.Custom)
|
dbType = App.GetOptions<DbConnectionOptions>().ConnectionConfigs[0].DbType;
|
|
var dataType = dbType switch
|
{
|
DbType.Oracle => ConvertDataType_OracleSQL(string.IsNullOrEmpty(dbColumnInfo.OracleDataType) ? dbColumnInfo.DataType : dbColumnInfo.OracleDataType, dbColumnInfo.Length, dbColumnInfo.Scale),
|
DbType.PostgreSQL => ConvertDataType_PostgreSQL(dbColumnInfo.DataType),
|
_ => ConvertDataType_Default(dbColumnInfo.DataType),
|
};
|
return dataType + (dbColumnInfo.IsNullable ? "?" : "");
|
}
|
|
public static string ConvertDataType_OracleSQL(string dataType, int? length, int? scale)
|
{
|
switch (dataType.ToLower())
|
{
|
case "interval year to month":
|
return "int";
|
|
case "interval day to second":
|
return "TimeSpan";
|
|
case "smallint":
|
return "Int16";
|
|
case "int":
|
case "integer":
|
return "int";
|
|
case "long":
|
return "long";
|
|
case "float":
|
return "float";
|
|
case "decimal":
|
return "decimal";
|
|
case "number":
|
if (length != null)
|
{
|
if (scale != null && scale > 0)
|
{
|
return "decimal";
|
}
|
else if ((scale != null && scale == 0) || scale == null)
|
{
|
if (length > 1 && length < 12)
|
{
|
return "int";
|
}
|
else if (length > 11)
|
{
|
return "long";
|
}
|
}
|
if (length == 1)
|
{
|
return "bool";
|
}
|
}
|
return "decimal";
|
|
case "char":
|
case "clob":
|
case "nclob":
|
case "nchar":
|
case "nvarchar":
|
case "varchar":
|
case "nvarchar2":
|
case "varchar2":
|
case "rowid":
|
return "string";
|
|
case "timestamp":
|
case "timestamp with time zone":
|
case "timestamptz":
|
case "timestamp without time zone":
|
case "date":
|
case "time":
|
case "time with time zone":
|
case "timetz":
|
case "time without time zone":
|
return "DateTime";
|
|
case "bfile":
|
case "blob":
|
case "raw":
|
return "byte[]";
|
|
default:
|
return "object";
|
}
|
}
|
|
//PostgreSQL数据类型对应的字段类型
|
public static string ConvertDataType_PostgreSQL(string dataType)
|
{
|
switch (dataType)
|
{
|
case "int2":
|
case "smallint":
|
return "Int16";
|
|
case "int4":
|
case "integer":
|
return "int";
|
|
case "int8":
|
case "bigint":
|
return "long";
|
|
case "float4":
|
case "real":
|
return "float";
|
|
case "float8":
|
case "double precision":
|
return "double";
|
|
case "numeric":
|
case "decimal":
|
case "path":
|
case "point":
|
case "polygon":
|
case "interval":
|
case "lseg":
|
case "macaddr":
|
case "money":
|
return "decimal";
|
|
case "boolean":
|
case "bool":
|
case "box":
|
case "bytea":
|
return "bool";
|
|
case "varchar":
|
case "character varying":
|
case "geometry":
|
case "name":
|
case "text":
|
case "char":
|
case "character":
|
case "cidr":
|
case "circle":
|
case "tsquery":
|
case "tsvector":
|
case "txid_snapshot":
|
case "xml":
|
case "json":
|
return "string";
|
|
case "uuid":
|
return "Guid";
|
|
case "timestamp":
|
case "timestamp with time zone":
|
case "timestamptz":
|
case "timestamp without time zone":
|
case "date":
|
case "time":
|
case "time with time zone":
|
case "timetz":
|
case "time without time zone":
|
return "DateTime";
|
|
case "bit":
|
case "bit varying":
|
return "byte[]";
|
|
case "varbit":
|
return "byte";
|
|
default:
|
return "object";
|
}
|
}
|
|
public static string ConvertDataType_Default(string dataType)
|
{
|
return dataType.ToLower() switch
|
{
|
"tinytext" or "mediumtext" or "longtext" or "mid" or "text" or "varchar" or "char" or "nvarchar" or "nchar" or "timestamp" => "string",
|
"int" => "int",
|
"smallint" => "Int16",
|
//"tinyint" => "byte",
|
"tinyint" => "bool", // MYSQL
|
"bigint" or "integer" => "long",
|
"bit" => "bool",
|
"money" or "smallmoney" or "numeric" or "decimal" => "decimal",
|
"real" => "Single",
|
"datetime" or "smalldatetime" => "DateTime",
|
"float" or "double" => "double",
|
"image" or "binary" or "varbinary" => "byte[]",
|
"uniqueidentifier" => "Guid",
|
_ => "object",
|
};
|
}
|
|
/// <summary>
|
/// 数据类型转显示类型
|
/// </summary>
|
/// <param name="dataType"></param>
|
/// <returns></returns>
|
public static string DataTypeToEff(string dataType)
|
{
|
if (string.IsNullOrEmpty(dataType)) return "";
|
return dataType?.TrimEnd('?') switch
|
{
|
"string" => "Input",
|
"int" => "InputNumber",
|
"long" => "Input",
|
"float" => "Input",
|
"double" => "Input",
|
"decimal" => "Input",
|
"bool" => "Switch",
|
"Guid" => "Input",
|
"DateTime" => "DatePicker",
|
_ => "Input",
|
};
|
}
|
|
// 是否通用字段
|
public static bool IsCommonColumn(string columnName)
|
{
|
var columnList = new List<string>()
|
{
|
nameof(EntityBaseData.CreateOrgId),
|
nameof(EntityTenant.TenantId),
|
//修改，显示出 日期 【Editby shaocx,2024-05-07】
|
//nameof(EntityBase.CreateTime),
|
//nameof(EntityBase.UpdateTime),
|
nameof(EntityBase.CreateUserId),
|
nameof(EntityBase.UpdateUserId),
|
//nameof(EntityBase.CreateUserName),
|
//nameof(EntityBase.UpdateUserName),
|
nameof(EntityBase.IsDelete)
|
};
|
return columnList.Contains(columnName);
|
}
|
|
// 是否不增改查询字段
|
public static bool IsNoNeedCurdColumn(string columnName)
|
{
|
var columnList = new List<string>()
|
{
|
nameof(EntityBase.CreateTime),
|
nameof(EntityBase.UpdateTime),
|
nameof(EntityBase.CreateUserName),
|
nameof(EntityBase.UpdateUserName)
|
};
|
return columnList.Contains(columnName);
|
}
|
|
// 是否是 基础时间字段
|
public static bool IsEntityBaseTimeColumn(string columnName)
|
{
|
var columnList = new List<string>()
|
{
|
nameof(EntityBase.CreateTime),
|
nameof(EntityBase.UpdateTime)
|
};
|
return columnList.Contains(columnName);
|
}
|
|
|
///// <summary>
|
///// 校验 代码配置
|
///// </summary>
|
///// <param name="list"></param>
|
//public static void ValidateCodeGenConfig(List<SysCodeGenConfig> list)
|
//{
|
// //如果是下拉框，那么就必须选择字典！【Editby shaocx,2024-04-16】
|
// foreach (var item in list)
|
// {
|
// if (IsSelect(item.EffectType) && string.IsNullOrEmpty(item.DictTypeCode))
|
// {
|
// throw Oops.Oh($"字段{item.ColumnName}的作用类型是'下拉框',字典必须选择！");
|
// }
|
// if (IsIngoreConfigAddEdit(item.ColumnName.ToLower()) && item.WhetherRequired == YesNoEnum.Y.ToString() && item.WhetherAddUpdate == YesNoEnum.N.ToString())
|
// {
|
// throw Oops.Oh($"字段{item.ColumnName}必填字段必须配置增改！");
|
// }
|
// }
|
//}
|
|
/// <summary>
|
/// 校验 代码配置
|
/// </summary>
|
/// <param name="list"></param>
|
public static void ValidateCodeGenConfig(List<CodeGenConfig> list)
|
{
|
//如果是下拉框，那么就必须选择字典！【Editby shaocx,2024-04-16】
|
foreach (var item in list)
|
{
|
if (IsSelect(item.EffectType) && string.IsNullOrEmpty(item.DictTypeCode))
|
{
|
throw Oops.Oh($"字段{item.ColumnName}的作用类型是'下拉框',字典必须选择！");
|
}
|
if (IsIngoreConfigAddEdit(item.ColumnName.ToLower()) && item.WhetherRequired == YesNoEnum.Y.ToString() && item.WhetherAddUpdate == YesNoEnum.N.ToString())
|
{
|
throw Oops.Oh($"字段{item.ColumnName}必填字段必须配置增改！");
|
}
|
|
if (item.WhetherTable == YesNoEnum.N.ToString() && item.WhetherKeyQuery == YesNoEnum.Y.ToString())
|
{
|
throw Oops.Oh($"字段{item.ColumnName}列设置不显示但是配置了关键字查询！");
|
}
|
}
|
}
|
|
public static bool IsSelect(string effectType)
|
{
|
if (effectType.ToLower() == "select" || effectType == "EnumSelector")
|
{
|
return true;
|
}
|
return false;
|
}
|
|
public static bool IsIngoreConfigAddEdit(string columnName)
|
{
|
if (columnName.ToLower() != "id" && columnName.ToLower() != "isdelete")
|
{
|
return true;
|
}
|
return false;
|
}
|
}
|
