lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-04-05 17:37:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update DbFirst

Browse Source
This commit is contained in:
sunkaixuan 2023-11-06 09:22:40 +08:00
parent 56c7ca6ead
commit 9bda0a2b9c
2 changed files with 41 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
Src/Asp.Net/SqlSugar/Abstract/DbFirstProvider/DbFirstProvider.cs
View File

@ -21,6 +21,8 @@ namespace SqlSugar
private bool IsDefaultValue { get; set; }
private Func<string, bool> WhereColumnsfunc;
private Func<string, string> FormatFileNameFunc { get; set; }
private Func<string, string> FormatClassNameFunc { get; set; }
private Func<string, string> FormatPropertyNameFunc { get; set; }
private bool IsStringNullable {get;set; }
private Func<DbColumnInfo,string,string,string> PropertyTextTemplateFunc { get; set; }
private Func<string, string> ReplaceClassStringFunc { get; set; }
@ -167,6 +169,16 @@ namespace SqlSugar
this.FormatFileNameFunc = formatFileNameFunc;
return this;
}
public IDbFirst FormatClassName(Func<string, string> formatClassNameFunc)
{
this.FormatClassNameFunc = formatClassNameFunc;
return this;
}
public IDbFirst FormatPropertyName(Func<string, string> formatPropertyNameFunc)
{
this.FormatPropertyNameFunc = formatPropertyNameFunc;
return this;
}
public IDbFirst CreatedReplaceClassString(Func<string, string> replaceClassStringFunc)
{
this.ReplaceClassStringFunc = replaceClassStringFunc;
@ -223,12 +235,17 @@ namespace SqlSugar
{
string classText = null;
string className = tableInfo.Name;
var oldClasName = className;
classText = GetClassString(tableInfo, ref className);
result.Remove(className);
if (this.ReplaceClassStringFunc != null)
{
classText=this.ReplaceClassStringFunc(classText);
}
if (FormatClassNameFunc != null&&FormatFileNameFunc != null)
{
className = oldClasName;
}
result.Add(className, classText);
}
catch (Exception ex)
@ -265,6 +282,10 @@ namespace SqlSugar
classText = classText.Replace(DbFirstTemplate.KeyClassName, mappingInfo.EntityName);
}
}
if (FormatClassNameFunc != null)
{
className=FormatClassNameFunc(className);
}
classText = classText.Replace(DbFirstTemplate.KeyClassName, className);
classText = classText.Replace(DbFirstTemplate.KeyNamespace, this.Namespace);
classText = classText.Replace(DbFirstTemplate.KeyUsing, IsAttribute ? (this.UsingTemplate + "using " + UtilConstants.AssemblyName + ";\r\n") : this.UsingTemplate);
@ -278,7 +299,12 @@ namespace SqlSugar
var index = columns.IndexOf(item);
string PropertyText = this.PropertyTemplate;
string PropertyDescriptionText = this.PropertyDescriptionTemplate;
string propertyName = GetPropertyName(item);
string propertyName = GetPropertyName(item);
var oldPropertyName = propertyName;
if (FormatPropertyNameFunc != null)
{
item.DbColumnName=propertyName = FormatPropertyNameFunc(propertyName);
}
string propertyTypeName = GetPropertyTypeName(item);
PropertyText =this.PropertyTextTemplateFunc == null? GetPropertyText(item, PropertyText):this.PropertyTextTemplateFunc(item,this.PropertyTemplate, propertyTypeName);
PropertyDescriptionText = GetPropertyDescriptionText(item, PropertyDescriptionText);
@ -286,10 +312,21 @@ namespace SqlSugar
{
PropertyDescriptionText += "\r\n [SugarColumn(IsArray=true)]";
}
else if (item?.DataType?.StartsWith("json")==true)
else if (item?.DataType?.StartsWith("json") == true)
{
PropertyDescriptionText += "\r\n [SugarColumn(IsJson=true)]";
}
else if (FormatPropertyNameFunc != null)
{
if (PropertyText.Contains("SugarColumn"))
{
PropertyText = PropertyText.Replace(")]", ",ColumnName=\"" + oldPropertyName + "\")]");
}
else
{
PropertyDescriptionText += "\r\n [SugarColumn(ColumnName=\"" + oldPropertyName + "\")]";
}
}
PropertyText = PropertyDescriptionText + PropertyText;
classText = classText.Replace(DbFirstTemplate.KeyPropertyName, PropertyText + (isLast ? "" : ("\r\n" + DbFirstTemplate.KeyPropertyName)));
if (ConstructorText.HasValue() && item.DefaultValue != null)

2
Src/Asp.Net/SqlSugar/Interface/IDbFirst.cs
View File

@ -25,6 +25,8 @@ namespace SqlSugar
Dictionary<string, string> ToClassStringList(string nameSpace = "Models");
void Init();
IDbFirst FormatFileName(Func<string,string> formatFileNameFunc);
IDbFirst FormatClassName(Func<string, string> formatClassNameFunc);
IDbFirst FormatPropertyName(Func<string, string> formatPropertyNameFunc);
IDbFirst StringNullable();
IDbFirst CreatedReplaceClassString(Func<string,string> replaceClassStringFunc);
}