281 lines
8.9 KiB
C#
281 lines
8.9 KiB
C#
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
||
//
|
||
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
||
//
|
||
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
||
|
||
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(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.Dm => ConvertDataType_Dm(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 ? "?" : "");
|
||
}
|
||
|
||
// 达梦(DM)数据类型对应的字段类型
|
||
public static string ConvertDataType_Dm(string dataType, int? length, int? scale)
|
||
{
|
||
return ConvertDataType_OracleSQL(dataType, length, scale); //达梦兼容Oracle，目前先这样实现
|
||
}
|
||
|
||
// OracleSQL数据类型对应的字段类型
|
||
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) return "decimal";
|
||
return scale switch
|
||
{
|
||
> 0 => "decimal",
|
||
0 or null when length is > 1 and < 12 => "int",
|
||
0 or null when length > 11 => "long",
|
||
_ => length == 1 ? "bool" : "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" or "integer" => "int",
|
||
"smallint" => "Int16",
|
||
//"tinyint" => "byte",
|
||
"tinyint" => "bool", // MYSQL
|
||
"bigint" => "long",
|
||
"bit" => "bool",
|
||
"money" or "smallmoney" or "numeric" or "decimal" => "decimal",
|
||
"real" => "Single",
|
||
"datetime" or "datetime2" 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" => "InputNumber",
|
||
"float" => "InputNumber",
|
||
"double" => "InputNumber",
|
||
"decimal" => "InputNumber",
|
||
"bool" => "Switch",
|
||
"Guid" => "Input",
|
||
"DateTime" => "DatePicker",
|
||
_ => "Input",
|
||
};
|
||
}
|
||
|
||
// 是否通用字段
|
||
public static bool IsCommonColumn(string columnName)
|
||
{
|
||
var columnList = new List<string>()
|
||
{
|
||
nameof(EntityBaseData.CreateOrgId),
|
||
nameof(EntityBaseData.CreateOrgName),
|
||
nameof(EntityTenant.TenantId),
|
||
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);
|
||
}
|
||
} |