71 lines
2.1 KiB
C#
71 lines
2.1 KiB
C#
using System.ComponentModel.DataAnnotations;
|
|
using System.Reflection;
|
|
using System.Text;
|
|
using Yi.Framework.CodeGen.Domain.Shared.Enums;
|
|
|
|
namespace Yi.Framework.CodeGen.Domain.Handlers
|
|
{
|
|
public class FieldTemplateHandler : TemplateHandlerBase, ITemplateHandler
|
|
{
|
|
public HandledTemplate Invoker(string str, string path)
|
|
{
|
|
var output = new HandledTemplate();
|
|
output.TemplateStr = str.Replace("@field", BuildFields());
|
|
output.BuildPath = path;
|
|
return output;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 生成Fields
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public string BuildFields()
|
|
{
|
|
StringBuilder fieldStrs = new StringBuilder();
|
|
|
|
|
|
foreach (var field in Table.Fields)
|
|
{
|
|
var typeStr = typeof(FieldTypeEnum).GetFields().Where(x => x.Name == field.FieldType.ToString())?.FirstOrDefault().GetCustomAttribute<DisplayAttribute>().Name;
|
|
|
|
if (typeStr is null)
|
|
{
|
|
continue;
|
|
}
|
|
var nameStr = field.Name;
|
|
|
|
//添加备注
|
|
if (!string.IsNullOrEmpty(field.Description))
|
|
{
|
|
var desStr = "/// <summary>\n" +
|
|
$"///{field.Description}\n" +
|
|
"/// </summary>\n";
|
|
fieldStrs.AppendLine(desStr);
|
|
}
|
|
|
|
//添加长度
|
|
if (field.Length != 0)
|
|
{
|
|
var lengthStr = $"[SugarColumn(Length ={field.Length})]";
|
|
fieldStrs.AppendLine(lengthStr);
|
|
}
|
|
|
|
//添加可空类型
|
|
string nullStr = "";
|
|
if (field.IsRequired == false)
|
|
{
|
|
nullStr = "?";
|
|
}
|
|
|
|
//添加字段
|
|
var fieldStr = $"public {typeStr}{nullStr} {nameStr} {{ get; set; }}";
|
|
|
|
fieldStrs.AppendLine(fieldStr);
|
|
}
|
|
|
|
return fieldStrs.ToString();
|
|
}
|
|
}
|
|
}
|