tk/FreeSql
1
0
Fork 0
mirror of https://github.com/nsnail/FreeSql.git synced 2025-06-18 03:53:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

- 修复 DbFirst string not null 未生成特性 IsNullable = false 的问题；#678

Browse Source
This commit is contained in:
2881099 2021-02-07 17:23:30 +08:00
parent d7cc7cebe9
commit 6cba1dedaf
1 changed files with 31 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
Extensions/FreeSql.Generator/RazorModel.cs
View File

@ -9,24 +9,27 @@ using System.Linq;
using System.Text; using System.Text;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
public class RazorModel { public class RazorModel
public RazorModel(IFreeSql fsql, string nameSpace, bool[] NameOptions, List<DbTableInfo> tables, DbTableInfo table) { {
public RazorModel(IFreeSql fsql, string nameSpace, bool[] NameOptions, List<DbTableInfo> tables, DbTableInfo table)
{
this.fsql = fsql; this.fsql = fsql;
this.NameSpace = nameSpace; this.NameSpace = nameSpace;
this.NameOptions = NameOptions; this.NameOptions = NameOptions;
this.tables = tables; this.tables = tables;
this.table = table; this.table = table;
} }
public IFreeSql fsql { get; set; } public IFreeSql fsql { get; set; }
public bool[] NameOptions { get; set; } public bool[] NameOptions { get; set; }
public List<DbTableInfo> tables { get; set; } public List<DbTableInfo> tables { get; set; }
public DbTableInfo table { get; set; } public DbTableInfo table { get; set; }
public List<DbColumnInfo> columns => this.table.Columns; public List<DbColumnInfo> columns => this.table.Columns;
public string NameSpace { get; set; } public string NameSpace { get; set; }
public string FullTableName => $"{(new[] { "public", "dbo" }.Contains(table.Schema) ? "" : table.Schema)}.{table.Name}".TrimStart('.'); public string FullTableName => $"{(new[] { "public", "dbo" }.Contains(table.Schema) ? "" : table.Schema)}.{table.Name}".TrimStart('.');
public string GetCsName(string name) { public string GetCsName(string name)
{
name = Regex.Replace(name.TrimStart('@', '.'), @"[^\w]", "_"); name = Regex.Replace(name.TrimStart('@', '.'), @"[^\w]", "_");
name = char.IsLetter(name, 0) ? name : string.Concat("_", name); name = char.IsLetter(name, 0) ? name : string.Concat("_", name);
if (NameOptions[0]) name = UFString(name); if (NameOptions[0]) name = UFString(name);
@ -35,18 +38,21 @@ public class RazorModel {
if (NameOptions[3]) name = string.Join("", name.Split('_').Select(a => UFString(a))); if (NameOptions[3]) name = string.Join("", name.Split('_').Select(a => UFString(a)));
return name; return name;
} }
public string UFString(string text) { public string UFString(string text)
{
text = Regex.Replace(text, @"[^\w]", "_"); text = Regex.Replace(text, @"[^\w]", "_");
if (text.Length <= 1) return text.ToUpper(); if (text.Length <= 1) return text.ToUpper();
else return text.Substring(0, 1).ToUpper() + text.Substring(1, text.Length - 1); else return text.Substring(0, 1).ToUpper() + text.Substring(1, text.Length - 1);
} }
public string LFString(string text) { public string LFString(string text)
{
text = Regex.Replace(text, @"[^\w]", "_"); text = Regex.Replace(text, @"[^\w]", "_");
if (text.Length <= 1) return text.ToLower(); if (text.Length <= 1) return text.ToLower();
else return text.Substring(0, 1).ToLower() + text.Substring(1, text.Length - 1); else return text.Substring(0, 1).ToLower() + text.Substring(1, text.Length - 1);
} }
public string GetCsType(DbColumnInfo col) { public string GetCsType(DbColumnInfo col)
{
if (fsql.Ado.DataType == FreeSql.DataType.MySql) if (fsql.Ado.DataType == FreeSql.DataType.MySql)
if (col.DbType == (int)MySqlDbType.Enum || col.DbType == (int)MySqlDbType.Set) if (col.DbType == (int)MySqlDbType.Enum || col.DbType == (int)MySqlDbType.Set)
return $"{this.GetCsName(this.FullTableName)}{this.GetCsName(col.Name).ToUpper()}{(col.IsNullable ? "?" : "")}"; return $"{this.GetCsName(this.FullTableName)}{this.GetCsName(col.Name).ToUpper()}{(col.IsNullable ? "?" : "")}";
@ -181,9 +187,10 @@ public class RazorModel {
if (dbinfo != null && dbinfo.isnullable != col.IsNullable) if (dbinfo != null && dbinfo.isnullable != col.IsNullable)
{ {
if (col.IsNullable && fsql.DbFirst.GetCsType(col).Contains("?") == false && col.CsType.IsValueType) var cstype = fsql.DbFirst.GetCsType(col);
if (col.IsNullable && cstype.Contains("?") == false && col.CsType.IsValueType)
sb.Add("IsNullable = true"); sb.Add("IsNullable = true");
if (col.IsNullable == false && fsql.DbFirst.GetCsType(col).Contains("?") == true) if (col.IsNullable == false && (cstype.Contains("?") == true || cstype == "string"))
sb.Add("IsNullable = false"); sb.Add("IsNullable = false");
} }
@ -200,7 +207,7 @@ public class RazorModel {
} }
} }
//else //else
//sb.Add("CanInsert = false"); //sb.Add("CanInsert = false");
} }
} }
if (sb.Any() == false) return null; if (sb.Any() == false) return null;
@ -260,8 +267,10 @@ public class RazorModel {
{ {
if (fsql.Ado.DataType != FreeSql.DataType.MySql && fsql.Ado.DataType != FreeSql.DataType.OdbcMySql) return null; if (fsql.Ado.DataType != FreeSql.DataType.MySql && fsql.Ado.DataType != FreeSql.DataType.OdbcMySql) return null;
var sb = new StringBuilder(); var sb = new StringBuilder();
foreach (var col in table.Columns) { foreach (var col in table.Columns)
if (col.DbType == (int)MySqlDbType.Enum || col.DbType == (int)MySqlDbType.Set) { {
if (col.DbType == (int)MySqlDbType.Enum || col.DbType == (int)MySqlDbType.Set)
{
if (col.DbType == (int)MySqlDbType.Set) sb.Append("\r\n\t[Flags]"); if (col.DbType == (int)MySqlDbType.Set) sb.Append("\r\n\t[Flags]");
sb.Append($"\r\n\tpublic enum {this.GetCsName(this.FullTableName)}{this.GetCsName(col.Name).ToUpper()}"); sb.Append($"\r\n\tpublic enum {this.GetCsName(this.FullTableName)}{this.GetCsName(col.Name).ToUpper()}");
if (col.DbType == (int)MySqlDbType.Set) sb.Append(" : long"); if (col.DbType == (int)MySqlDbType.Set) sb.Append(" : long");
@ -272,10 +281,12 @@ public class RazorModel {
int unknow_idx = 0; int unknow_idx = 0;
string exp2 = string.Concat(col.DbTypeTextFull); string exp2 = string.Concat(col.DbTypeTextFull);
int quote_pos = -1; int quote_pos = -1;
while (true) { while (true)
{
int first_pos = quote_pos = exp2.IndexOf('\'', quote_pos + 1); int first_pos = quote_pos = exp2.IndexOf('\'', quote_pos + 1);
if (quote_pos == -1) break; if (quote_pos == -1) break;
while (true) { while (true)
{
quote_pos = exp2.IndexOf('\'', quote_pos + 1); quote_pos = exp2.IndexOf('\'', quote_pos + 1);
if (quote_pos == -1) break; if (quote_pos == -1) break;
int r_cout = 0; int r_cout = 0;
@ -284,7 +295,8 @@ public class RazorModel {
// else break; // else break;
//} //}
while (exp2[++quote_pos] == '\'') r_cout++; while (exp2[++quote_pos] == '\'') r_cout++;
if (r_cout % 2 == 0/* && quote_pos - first_pos > 2*/) { if (r_cout % 2 == 0/* && quote_pos - first_pos > 2*/)
{
string str2 = exp2.Substring(first_pos + 1, quote_pos - first_pos - 2).Replace("''", "'"); string str2 = exp2.Substring(first_pos + 1, quote_pos - first_pos - 2).Replace("''", "'");
if (Regex.IsMatch(str2, @"^[\u0391-\uFFE5a-zA-Z_\$][\u0391-\uFFE5a-zA-Z_\$\d]*$")) if (Regex.IsMatch(str2, @"^[\u0391-\uFFE5a-zA-Z_\$][\u0391-\uFFE5a-zA-Z_\$\d]*$"))
slkdgjlksdjg += ", " + str2; slkdgjlksdjg += ", " + str2;