SelectTable.cs 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972
  1. // 大名科技(天津)有限公司版权所有 电话:18020030720 QQ:515096995
  2. //
  3. // 此源代码遵循位于源代码树根目录中的 LICENSE 文件的许可证
  4. using AspectCore.Extensions.Reflection;
  5. using System.Dynamic;
  6. namespace Admin.NET.Core.Service;
  7. /// <summary>
  8. ///
  9. /// </summary>
  10. public class SelectTable : ISingleton
  11. {
  12. private readonly IdentityService _identitySvc;
  13. private readonly TableMapper _tableMapper;
  14. private readonly ISqlSugarClient _db;
  15. public SelectTable(IdentityService identityService, TableMapper tableMapper, ISqlSugarClient dbClient)
  16. {
  17. _identitySvc = identityService;
  18. _tableMapper = tableMapper;
  19. _db = dbClient;
  20. }
  21. /// <summary>
  22. /// 判断表名是否正确,若不正确则抛异常
  23. /// </summary>
  24. /// <param name="table"></param>
  25. /// <returns></returns>
  26. public virtual bool IsTable(string table)
  27. {
  28. return _db.DbMaintenance.GetTableInfoList().Any(it => it.Name.Equals(table, StringComparison.CurrentCultureIgnoreCase))
  29. ? true
  30. : throw new Exception($"表名【{table}】不正确!");
  31. }
  32. /// <summary>
  33. /// 判断表的列名是否正确,如果不正确则抛异常,更早地暴露给调用方
  34. /// </summary>
  35. /// <param name="table"></param>
  36. /// <param name="col"></param>
  37. /// <returns></returns>
  38. public virtual bool IsCol(string table, string col)
  39. {
  40. return _db.DbMaintenance.GetColumnInfosByTableName(table).Any(it => it.DbColumnName.Equals(col, StringComparison.CurrentCultureIgnoreCase))
  41. ? true
  42. : throw new Exception($"表【{table}】不存在列【{col}】!请检查输入参数");
  43. }
  44. /// <summary>
  45. /// 查询列表数据
  46. /// </summary>
  47. /// <param name="subtable"></param>
  48. /// <param name="page"></param>
  49. /// <param name="count"></param>
  50. /// <param name="json"></param>
  51. /// <param name="dd"></param>
  52. /// <returns></returns>
  53. public virtual Tuple<dynamic, int> GetTableData(string subtable, int page, int count, int query, string json, JObject dd)
  54. {
  55. var role = _identitySvc.GetSelectRole(subtable);
  56. if (!role.Item1)
  57. throw new Exception(role.Item2);
  58. var selectrole = role.Item2;
  59. subtable = _tableMapper.GetTableName(subtable);
  60. var values = JObject.Parse(json);
  61. page = values["page"] == null ? page : int.Parse(values["page"].ToString());
  62. count = values["count"] == null ? count : int.Parse(values["count"].ToString());
  63. query = values["query"] == null ? query : int.Parse(values["query"].ToString());
  64. values.Remove("page");
  65. values.Remove("count");
  66. // 构造查询过程
  67. var tb = SugarQueryable(subtable, selectrole, values, dd);
  68. // 实际会在这里执行
  69. if (query == 1) // 1-总数
  70. {
  71. return new Tuple<dynamic, int>(null, tb.MergeTable().Count());
  72. }
  73. else
  74. {
  75. if (page > 0) // 分页
  76. {
  77. int total = 0;
  78. if (query == 0)
  79. return new Tuple<dynamic, int>(tb.ToPageList(page, count), total); // 0-对象
  80. else
  81. return new Tuple<dynamic, int>(tb.ToPageList(page, count, ref total), total); // 2-以上全部
  82. }
  83. else // 列表
  84. {
  85. IList l = tb.ToList();
  86. return query == 0 ? new Tuple<dynamic, int>(l, 0) : new Tuple<dynamic, int>(l, l.Count);
  87. }
  88. }
  89. }
  90. /// <summary>
  91. /// 解析并查询
  92. /// </summary>
  93. /// <param name="queryJson"></param>
  94. /// <returns></returns>
  95. public virtual JObject Query(string queryJson)
  96. {
  97. var queryJobj = JObject.Parse(queryJson);
  98. return Query(queryJobj);
  99. }
  100. /// <summary>
  101. /// 单表查询
  102. /// </summary>
  103. /// <param name="queryObj"></param>
  104. /// <param name="nodeName">返回数据的节点名称 默认为 infos</param>
  105. /// <returns></returns>
  106. public virtual JObject QuerySingle(JObject queryObj, string nodeName = "infos")
  107. {
  108. var resultObj = new JObject();
  109. var total = 0;
  110. foreach (var item in queryObj)
  111. {
  112. var key = item.Key.Trim();
  113. if (key.EndsWith("[]"))
  114. {
  115. total = QuerySingleList(resultObj, item, nodeName);
  116. }
  117. else if (key.Equals("func"))
  118. {
  119. ExecFunc(resultObj, item);
  120. }
  121. else if (key.Equals("total@") || key.Equals("total"))
  122. {
  123. resultObj.Add("total", total);
  124. }
  125. }
  126. return resultObj;
  127. }
  128. /// <summary>
  129. /// 获取查询语句
  130. /// </summary>
  131. /// <param name="queryObj"></param>
  132. /// <returns></returns>
  133. public virtual string ToSql(JObject queryObj)
  134. {
  135. foreach (var item in queryObj)
  136. {
  137. if (item.Key.Trim().EndsWith("[]"))
  138. return ToSql(item);
  139. }
  140. return string.Empty;
  141. }
  142. /// <summary>
  143. /// 解析并查询
  144. /// </summary>
  145. /// <param name="queryObj"></param>
  146. /// <returns></returns>
  147. public virtual JObject Query(JObject queryObj)
  148. {
  149. var resultObj = new JObject();
  150. int total;
  151. foreach (var item in queryObj)
  152. {
  153. var key = item.Key.Trim();
  154. if (key.Equals("[]")) // 列表
  155. {
  156. total = QueryMoreList(resultObj, item);
  157. resultObj.Add("total", total); // 只要是列表查询都自动返回总数
  158. }
  159. else if (key.EndsWith("[]"))
  160. {
  161. total = QuerySingleList(resultObj, item);
  162. }
  163. else if (key.Equals("func"))
  164. {
  165. ExecFunc(resultObj, item);
  166. }
  167. else if (key.Equals("total@") || key.Equals("total"))
  168. {
  169. // resultObj.Add("total", total);
  170. continue;
  171. }
  172. else // 单条
  173. {
  174. var template = GetFirstData(key, item.Value.ToString(), resultObj);
  175. if (template != null)
  176. resultObj.Add(key, JToken.FromObject(template));
  177. }
  178. }
  179. return resultObj;
  180. }
  181. // 动态调用方法
  182. private object ExecFunc(string funcname, object[] param, Type[] types)
  183. {
  184. var method = typeof(FuncList).GetMethod(funcname);
  185. var reflector = method.GetReflector();
  186. var result = reflector.Invoke(new FuncList(), param);
  187. return result;
  188. }
  189. // 生成sql
  190. private string ToSql(string subtable, int page, int count, int query, string json)
  191. {
  192. var values = JObject.Parse(json);
  193. page = values["page"] == null ? page : int.Parse(values["page"].ToString());
  194. count = values["count"] == null ? count : int.Parse(values["count"].ToString());
  195. query = values["query"] == null ? query : int.Parse(values["query"].ToString());
  196. values.Remove("page");
  197. values.Remove("count");
  198. subtable = _tableMapper.GetTableName(subtable);
  199. var tb = SugarQueryable(subtable, "*", values, null);
  200. var sqlObj = tb.Skip((page - 1) * count).Take(10).ToSql();
  201. return sqlObj.Key;
  202. }
  203. /// <summary>
  204. /// 查询第一条数据
  205. /// </summary>
  206. /// <param name="subtable"></param>
  207. /// <param name="json"></param>
  208. /// <param name="job"></param>
  209. /// <returns></returns>
  210. /// <exception cref="Exception"></exception>
  211. private dynamic GetFirstData(string subtable, string json, JObject job)
  212. {
  213. var role = _identitySvc.GetSelectRole(subtable);
  214. if (!role.Item1)
  215. throw new Exception(role.Item2);
  216. var selectrole = role.Item2;
  217. subtable = _tableMapper.GetTableName(subtable);
  218. var values = JObject.Parse(json);
  219. values.Remove("page");
  220. values.Remove("count");
  221. var tb = SugarQueryable(subtable, selectrole, values, job).First();
  222. var dic = (IDictionary<string, object>)tb;
  223. foreach (var item in values.Properties().Where(it => it.Name.EndsWith("()")))
  224. {
  225. if (item.Value.IsNullOrEmpty())
  226. {
  227. var func = item.Value.ToString().Substring(0, item.Value.ToString().IndexOf("("));
  228. var param = item.Value.ToString().Substring(item.Value.ToString().IndexOf("(") + 1).TrimEnd(')');
  229. var types = new List<Type>();
  230. var paramss = new List<object>();
  231. foreach (var va in param.Split(','))
  232. {
  233. types.Add(typeof(object));
  234. paramss.Add(tb.Where(it => it.Key.Equals(va)).Select(i => i.Value));
  235. }
  236. dic[item.Name] = ExecFunc(func, paramss.ToArray(), types.ToArray());
  237. }
  238. }
  239. return tb;
  240. }
  241. // 单表查询,返回的数据在指定的NodeName节点
  242. private int QuerySingleList(JObject resultObj, KeyValuePair<string, JToken> item, string nodeName)
  243. {
  244. var key = item.Key.Trim();
  245. var jb = JObject.Parse(item.Value.ToString());
  246. int page = jb["page"] == null ? 0 : int.Parse(jb["page"].ToString());
  247. int count = jb["count"] == null ? 10 : int.Parse(jb["count"].ToString());
  248. int query = jb["query"] == null ? 2 : int.Parse(jb["query"].ToString()); // 默认输出数据和数量
  249. int total = 0;
  250. jb.Remove("page"); jb.Remove("count"); jb.Remove("query");
  251. var htt = new JArray();
  252. foreach (var t in jb)
  253. {
  254. var datas = GetTableData(t.Key, page, count, query, t.Value.ToString(), null);
  255. if (query > 0)
  256. total = datas.Item2;
  257. foreach (var data in datas.Item1)
  258. {
  259. htt.Add(JToken.FromObject(data));
  260. }
  261. }
  262. if (!string.IsNullOrEmpty(nodeName))
  263. resultObj.Add(nodeName, htt);
  264. else
  265. resultObj.Add(key, htt);
  266. return total;
  267. }
  268. // 生成sql
  269. private string ToSql(KeyValuePair<string, JToken> item)
  270. {
  271. var jb = JObject.Parse(item.Value.ToString());
  272. int page = jb["page"] == null ? 0 : int.Parse(jb["page"].ToString());
  273. int count = jb["count"] == null ? 10 : int.Parse(jb["count"].ToString());
  274. int query = jb["query"] == null ? 2 : int.Parse(jb["query"].ToString()); // 默认输出数据和数量
  275. jb.Remove("page"); jb.Remove("count"); jb.Remove("query");
  276. foreach (var t in jb)
  277. {
  278. return ToSql(t.Key, page, count, query, t.Value.ToString());
  279. }
  280. return string.Empty;
  281. }
  282. // 单表查询
  283. private int QuerySingleList(JObject resultObj, KeyValuePair<string, JToken> item)
  284. {
  285. var key = item.Key.TrimEnd("[]");
  286. return QuerySingleList(resultObj, item, key);
  287. }
  288. /// <summary>
  289. /// 多列表查询
  290. /// </summary>
  291. /// <param name="resultObj"></param>
  292. /// <param name="item"></param>
  293. /// <returns></returns>
  294. private int QueryMoreList(JObject resultObj, KeyValuePair<string, JToken> item)
  295. {
  296. int total = 0;
  297. var jb = JObject.Parse(item.Value.ToString());
  298. var page = jb["page"] == null ? 0 : int.Parse(jb["page"].ToString());
  299. var count = jb["count"] == null ? 10 : int.Parse(jb["count"].ToString());
  300. var query = jb["query"] == null ? 2 : int.Parse(jb["query"].ToString()); // 默认输出数据和数量
  301. jb.Remove("page"); jb.Remove("count"); jb.Remove("query");
  302. var htt = new JArray();
  303. List<string> tables = new List<string>(), where = new List<string>();
  304. foreach (var t in jb)
  305. {
  306. tables.Add(t.Key); where.Add(t.Value.ToString());
  307. }
  308. if (tables.Count > 0)
  309. {
  310. string table = tables[0].TrimEnd("[]");
  311. var temp = GetTableData(table, page, count, query, where[0], null);
  312. if (query > 0)
  313. total = temp.Item2;
  314. // 关联查询,先查子表数据,再根据外键循环查询主表
  315. foreach (var dd in temp.Item1)
  316. {
  317. var zht = new JObject
  318. {
  319. { table, JToken.FromObject(dd) }
  320. };
  321. for (int i = 1; i < tables.Count; i++) // 从第二个表开始循环
  322. {
  323. string subtable = tables[i];
  324. // 有bug,暂不支持[]分支
  325. //if (subtable.EndsWith("[]"))
  326. //{
  327. // string tableName = subtable.TrimEnd("[]".ToCharArray());
  328. // var jbb = JObject.Parse(where[i]);
  329. // page = jbb["page"] == null ? 0 : int.Parse(jbb["page"].ToString());
  330. // count = jbb["count"] == null ? 0 : int.Parse(jbb["count"].ToString());
  331. // var lt = new JArray();
  332. // foreach (var d in GetTableData(tableName, page, count, query, item.Value[subtable].ToString(), zht).Item1)
  333. // {
  334. // lt.Add(JToken.FromObject(d));
  335. // }
  336. // zht.Add(tables[i], lt);
  337. //}
  338. //else
  339. //{
  340. var ddf = GetFirstData(subtable, where[i].ToString(), zht);
  341. if (ddf != null)
  342. zht.Add(subtable, JToken.FromObject(ddf));
  343. }
  344. htt.Add(zht);
  345. }
  346. }
  347. if (query != 1)
  348. resultObj.Add("[]", htt);
  349. // 分页自动添加当前页数和数量
  350. if (page > 0 && count > 0)
  351. {
  352. resultObj.Add("page", page);
  353. resultObj.Add("count", count);
  354. resultObj.Add("max", (int)Math.Ceiling((decimal)total / count));
  355. }
  356. return total;
  357. }
  358. // 执行方法
  359. private void ExecFunc(JObject resultObj, KeyValuePair<string, JToken> item)
  360. {
  361. var jb = JObject.Parse(item.Value.ToString());
  362. var dataJObj = new JObject();
  363. foreach (var f in jb)
  364. {
  365. var types = new List<Type>();
  366. var param = new List<object>();
  367. foreach (var va in JArray.Parse(f.Value.ToString()))
  368. {
  369. types.Add(typeof(object));
  370. param.Add(va);
  371. }
  372. dataJObj.Add(f.Key, JToken.FromObject(ExecFunc(f.Key, param.ToArray(), types.ToArray())));
  373. }
  374. resultObj.Add("func", dataJObj);
  375. }
  376. /// <summary>
  377. /// 构造查询过程
  378. /// </summary>
  379. /// <param name="subtable"></param>
  380. /// <param name="selectrole"></param>
  381. /// <param name="values"></param>
  382. /// <param name="dd"></param>
  383. /// <returns></returns>
  384. private ISugarQueryable<ExpandoObject> SugarQueryable(string subtable, string selectrole, JObject values, JObject dd)
  385. {
  386. IsTable(subtable);
  387. var tb = _db.Queryable(subtable, "tb");
  388. // select
  389. if (values["@column"].IsNullOrEmpty())
  390. {
  391. ProcessColumn(subtable, selectrole, values, tb);
  392. }
  393. else
  394. {
  395. tb.Select(selectrole);
  396. }
  397. // 前几行
  398. ProcessLimit(values, tb);
  399. // where
  400. ProcessWhere(subtable, values, tb, dd);
  401. // 排序
  402. ProcessOrder(subtable, values, tb);
  403. // 分组
  404. PrccessGroup(subtable, values, tb);
  405. // Having
  406. ProcessHaving(values, tb);
  407. return tb;
  408. }
  409. // 处理字段重命名 "@column":"toId:parentId",对应SQL是toId AS parentId,将查询的字段toId变为parentId返回
  410. private void ProcessColumn(string subtable, string selectrole, JObject values, ISugarQueryable<ExpandoObject> tb)
  411. {
  412. var str = new System.Text.StringBuilder(100);
  413. foreach (var item in values["@column"].ToString().Split(','))
  414. {
  415. var ziduan = item.Split(':');
  416. var colName = ziduan[0];
  417. var ma = new Regex(@"\((\w+)\)").Match(colName);
  418. // 处理max、min这样的函数
  419. if (ma.Success && ma.Groups.Count > 1)
  420. colName = ma.Groups[1].Value;
  421. // 判断列表是否有权限 sum(1)、sum(*)、Count(1)这样的值直接有效
  422. if (colName == "*" || int.TryParse(colName, out int colNumber) || (IsCol(subtable, colName) && _identitySvc.ColIsRole(colName, selectrole.Split(','))))
  423. {
  424. if (ziduan.Length > 1)
  425. {
  426. if (ziduan[1].Length > 20)
  427. throw new Exception("别名不能超过20个字符");
  428. str.Append(ziduan[0] + " as `" + ReplaceSQLChar(ziduan[1]) + "`,");
  429. }
  430. // 不对函数加``,解决sum(*)、Count(1)等不能使用的问题
  431. else if (ziduan[0].Contains('('))
  432. {
  433. str.Append(ziduan[0] + ",");
  434. }
  435. else
  436. str.Append("`" + ziduan[0] + "`" + ",");
  437. }
  438. }
  439. if (string.IsNullOrEmpty(str.ToString()))
  440. throw new Exception($"表名{subtable}没有可查询的字段!");
  441. tb.Select(str.ToString().TrimEnd(','));
  442. }
  443. /// <summary>
  444. /// 构造查询条件 where
  445. /// </summary>
  446. /// <param name="subtable"></param>
  447. /// <param name="values"></param>
  448. /// <param name="tb"></param>
  449. /// <param name="dd"></param>
  450. private void ProcessWhere(string subtable, JObject values, ISugarQueryable<ExpandoObject> tb, JObject dd)
  451. {
  452. var conModels = new List<IConditionalModel>();
  453. if (values["identity"].IsNullOrEmpty())
  454. conModels.Add(new ConditionalModel() { FieldName = values["identity"].ToString(), ConditionalType = ConditionalType.Equal, FieldValue = _identitySvc.GetUserIdentity() });
  455. foreach (var va in values)
  456. {
  457. string key = va.Key.Trim();
  458. string fieldValue = va.Value.ToString();
  459. if (key.StartsWith("@"))
  460. {
  461. continue;
  462. }
  463. if (key.EndsWith("$")) // 模糊查询
  464. {
  465. FuzzyQuery(subtable, conModels, va);
  466. }
  467. else if (key.EndsWith("{}")) // 逻辑运算
  468. {
  469. ConditionQuery(subtable, conModels, va);
  470. }
  471. else if (key.EndsWith("%")) // bwtween查询
  472. {
  473. ConditionBetween(subtable, conModels, va, tb);
  474. }
  475. else if (key.EndsWith("@")) // 关联上一个table
  476. {
  477. if (dd == null)
  478. continue;
  479. var str = fieldValue.Split('/');
  480. var lastTableRecord = ((JObject)dd[str[^2]]);
  481. if (!lastTableRecord.ContainsKey(str[^1]))
  482. throw new Exception($"找不到关联列:{str},请在{str[^2]}@column中设置");
  483. var value = lastTableRecord[str[^1]].ToString();
  484. conModels.Add(new ConditionalModel() { FieldName = key.TrimEnd('@'), ConditionalType = ConditionalType.Equal, FieldValue = value });
  485. }
  486. else if (key.EndsWith("~")) // 不等于(应该是正则匹配)
  487. {
  488. //conModels.Add(new ConditionalModel() { FieldName = key.TrimEnd('~'), ConditionalType = ConditionalType.NoEqual, FieldValue = fieldValue });
  489. }
  490. else if (IsCol(subtable, key.TrimEnd('!'))) // 其他where条件
  491. {
  492. ConditionEqual(subtable, conModels, va);
  493. }
  494. }
  495. if (conModels.Any())
  496. tb.Where(conModels);
  497. }
  498. // "@having":"function0(...)?value0;function1(...)?value1;function2(...)?value2...",
  499. // SQL函数条件,一般和 @group一起用,函数一般在 @column里声明
  500. private void ProcessHaving(JObject values, ISugarQueryable<ExpandoObject> tb)
  501. {
  502. if (values["@having"].IsNullOrEmpty())
  503. {
  504. var hw = new List<IConditionalModel>();
  505. var havingItems = new List<string>();
  506. if (values["@having"].HasValues)
  507. {
  508. havingItems = values["@having"].Select(p => p.ToString()).ToList();
  509. }
  510. else
  511. {
  512. havingItems.Add(values["@having"].ToString());
  513. }
  514. foreach (var item in havingItems)
  515. {
  516. var and = item.ToString();
  517. var model = new ConditionalModel();
  518. if (and.Contains(">="))
  519. {
  520. model.FieldName = and.Split(new string[] { ">=" }, StringSplitOptions.RemoveEmptyEntries)[0];
  521. model.ConditionalType = ConditionalType.GreaterThanOrEqual;
  522. model.FieldValue = and.Split(new string[] { ">=" }, StringSplitOptions.RemoveEmptyEntries)[1];
  523. }
  524. else if (and.Contains("<="))
  525. {
  526. model.FieldName = and.Split(new string[] { "<=" }, StringSplitOptions.RemoveEmptyEntries)[0];
  527. model.ConditionalType = ConditionalType.LessThanOrEqual;
  528. model.FieldValue = and.Split(new string[] { "<=" }, StringSplitOptions.RemoveEmptyEntries)[1];
  529. }
  530. else if (and.Contains(">"))
  531. {
  532. model.FieldName = and.Split(new string[] { ">" }, StringSplitOptions.RemoveEmptyEntries)[0];
  533. model.ConditionalType = ConditionalType.GreaterThan;
  534. model.FieldValue = and.Split(new string[] { ">" }, StringSplitOptions.RemoveEmptyEntries)[1];
  535. }
  536. else if (and.Contains("<"))
  537. {
  538. model.FieldName = and.Split(new string[] { "<" }, StringSplitOptions.RemoveEmptyEntries)[0];
  539. model.ConditionalType = ConditionalType.LessThan;
  540. model.FieldValue = and.Split(new string[] { "<" }, StringSplitOptions.RemoveEmptyEntries)[1];
  541. }
  542. else if (and.Contains("!="))
  543. {
  544. model.FieldName = and.Split(new string[] { "!=" }, StringSplitOptions.RemoveEmptyEntries)[0];
  545. model.ConditionalType = ConditionalType.NoEqual;
  546. model.FieldValue = and.Split(new string[] { "!=" }, StringSplitOptions.RemoveEmptyEntries)[1];
  547. }
  548. else if (and.Contains("="))
  549. {
  550. model.FieldName = and.Split(new string[] { "=" }, StringSplitOptions.RemoveEmptyEntries)[0];
  551. model.ConditionalType = ConditionalType.Equal;
  552. model.FieldValue = and.Split(new string[] { "=" }, StringSplitOptions.RemoveEmptyEntries)[1];
  553. }
  554. hw.Add(model);
  555. }
  556. //var d = db.Context.Utilities.ConditionalModelToSql(hw);
  557. //tb.Having(d.Key, d.Value);
  558. tb.Having(string.Join(",", havingItems));
  559. }
  560. }
  561. // "@group":"column0,column1...",分组方式。如果 @column里声明了Table的id,则id也必须在 @group中声明;其它情况下必须满足至少一个条件:
  562. // 1.分组的key在 @column里声明
  563. // 2.Table主键在 @group中声明
  564. private void PrccessGroup(string subtable, JObject values, ISugarQueryable<ExpandoObject> tb)
  565. {
  566. if (values["@group"].IsNullOrEmpty())
  567. {
  568. var groupList = new List<GroupByModel>(); // 多库兼容写法
  569. foreach (var col in values["@group"].ToString().Split(','))
  570. {
  571. if (IsCol(subtable, col))
  572. {
  573. // str.Append(and + ",");
  574. groupList.Add(new GroupByModel() { FieldName = col });
  575. }
  576. }
  577. if (groupList.Any())
  578. tb.GroupBy(groupList);
  579. }
  580. }
  581. // 处理排序 "@order":"name-,id"查询按 name降序、id默认顺序 排序的User数组
  582. private void ProcessOrder(string subtable, JObject values, ISugarQueryable<ExpandoObject> tb)
  583. {
  584. if (values["@order"].IsNullOrEmpty())
  585. {
  586. var orderList = new List<OrderByModel>(); // 多库兼容写法
  587. foreach (var item in values["@order"].ToString().Split(','))
  588. {
  589. string col = item.Replace("-", "").Replace("+", "").Replace(" desc", "").Replace(" asc", ""); // 增加对原生排序的支持
  590. if (IsCol(subtable, col))
  591. {
  592. orderList.Add(new OrderByModel()
  593. {
  594. FieldName = col,
  595. OrderByType = item.EndsWith("-") || item.EndsWith(" desc") ? OrderByType.Desc : OrderByType.Asc
  596. });
  597. }
  598. }
  599. if (orderList.Any())
  600. tb.OrderBy(orderList);
  601. }
  602. }
  603. /// <summary>
  604. /// 表内参数"@count"(int):查询前几行,不能同时使用count和@count函数
  605. /// </summary>
  606. /// <param name="values"></param>
  607. /// <param name="tb"></param>
  608. private void ProcessLimit(JObject values, ISugarQueryable<ExpandoObject> tb)
  609. {
  610. if (values["@count"].IsNullOrEmpty())
  611. {
  612. int c = values["@count"].ToObject<int>();
  613. tb.Take(c);
  614. }
  615. }
  616. // 条件查询 "key{}":"条件0,条件1...",条件为任意SQL比较表达式字符串,非Number类型必须用''包含条件的值,如'a'
  617. // &, |, ! 逻辑运算符,对应数据库 SQL 中的 AND, OR, NOT。
  618. // 横或纵与:同一字段的值内条件默认 | 或连接,不同字段的条件默认 & 与连接。
  619. // ① & 可用于"key&{}":"条件"等
  620. // ② | 可用于"key|{}":"条件", "key|{}":[] 等,一般可省略
  621. // ③ ! 可单独使用,如"key!":Object,也可像&,|一样配合其他功能符使用
  622. private void ConditionQuery(string subtable, List<IConditionalModel> conModels, KeyValuePair<string, JToken> va)
  623. {
  624. var vakey = va.Key.Trim();
  625. var field = vakey.TrimEnd("{}".ToCharArray());
  626. var columnName = field.TrimEnd(new char[] { '&', '|' });
  627. IsCol(subtable, columnName);
  628. var ddt = new List<KeyValuePair<WhereType, ConditionalModel>>();
  629. foreach (var and in va.Value.ToString().Split(','))
  630. {
  631. var model = new ConditionalModel
  632. {
  633. FieldName = columnName
  634. };
  635. if (and.StartsWith(">="))
  636. {
  637. model.ConditionalType = ConditionalType.GreaterThanOrEqual;
  638. model.FieldValue = and.TrimStart(">=".ToCharArray());
  639. }
  640. else if (and.StartsWith("<="))
  641. {
  642. model.ConditionalType = ConditionalType.LessThanOrEqual;
  643. model.FieldValue = and.TrimStart("<=".ToCharArray());
  644. }
  645. else if (and.StartsWith(">"))
  646. {
  647. model.ConditionalType = ConditionalType.GreaterThan;
  648. model.FieldValue = and.TrimStart('>');
  649. }
  650. else if (and.StartsWith("<"))
  651. {
  652. model.ConditionalType = ConditionalType.LessThan;
  653. model.FieldValue = and.TrimStart('<');
  654. }
  655. model.CSharpTypeName = FuncList.GetValueCSharpType(model.FieldValue);
  656. ddt.Add(new KeyValuePair<WhereType, ConditionalModel>(field.EndsWith("!") ? WhereType.Or : WhereType.And, model));
  657. }
  658. conModels.Add(new ConditionalCollections() { ConditionalList = ddt });
  659. }
  660. /// <summary>
  661. /// "key%":"start,end" => "key%":["start,end"],其中 start 和 end 都只能为 Boolean, Number, String 中的一种,如 "2017-01-01,2019-01-01" ,["1,90000", "82001,100000"] ,可用于连续范围内的筛选
  662. /// 目前不支持数组形式
  663. /// </summary>
  664. /// <param name="subtable"></param>
  665. /// <param name="conModels"></param>
  666. /// <param name="va"></param>
  667. /// <param name="tb"></param>
  668. private void ConditionBetween(string subtable, List<IConditionalModel> conModels, KeyValuePair<string, JToken> va, ISugarQueryable<ExpandoObject> tb)
  669. {
  670. var vakey = va.Key.Trim();
  671. var field = vakey.TrimEnd("%".ToCharArray());
  672. var inValues = new List<string>();
  673. if (va.Value.HasValues)
  674. {
  675. foreach (var cm in va.Value)
  676. {
  677. inValues.Add(cm.ToString());
  678. }
  679. }
  680. else
  681. {
  682. inValues.Add(va.Value.ToString());
  683. }
  684. for (var i = 0; i < inValues.Count; i++)
  685. {
  686. var fileds = inValues[i].Split(',');
  687. if (fileds.Length == 2)
  688. {
  689. var type = FuncList.GetValueCSharpType(fileds[0]);
  690. ObjectFuncModel f = ObjectFuncModel.Create("between", field, $"{{{type}}}:{fileds[0]}", $"{{{type}}}:{fileds[1]}");
  691. tb.Where(f);
  692. }
  693. }
  694. }
  695. /// <summary>
  696. /// 等于、不等于、in 、not in
  697. /// </summary>
  698. /// <param name="subtable"></param>
  699. /// <param name="conModels"></param>
  700. /// <param name="va"></param>
  701. /// <param name="key"></param>
  702. private void ConditionEqual(string subtable, List<IConditionalModel> conModels, KeyValuePair<string, JToken> va)
  703. {
  704. var key = va.Key;
  705. var fieldValue = va.Value.ToString();
  706. // in / not in
  707. if (va.Value is JArray)
  708. {
  709. conModels.Add(new ConditionalModel()
  710. {
  711. FieldName = key.TrimEnd('!'),
  712. ConditionalType = key.EndsWith("!") ? ConditionalType.NotIn : ConditionalType.In,
  713. FieldValue = va.Value.ToObject<string[]>().Aggregate((a, b) => a + "," + b)
  714. });
  715. }
  716. else
  717. {
  718. if (string.IsNullOrEmpty(fieldValue))
  719. {
  720. // is not null or ''
  721. if (key.EndsWith("!"))
  722. {
  723. conModels.Add(new ConditionalModel() { FieldName = key.TrimEnd('!'), ConditionalType = ConditionalType.IsNot, FieldValue = null });
  724. conModels.Add(new ConditionalModel() { FieldName = key.TrimEnd('!'), ConditionalType = ConditionalType.IsNot, FieldValue = "" });
  725. }
  726. //is null or ''
  727. else
  728. {
  729. conModels.Add(new ConditionalModel() { FieldName = key.TrimEnd('!'), FieldValue = null });
  730. }
  731. }
  732. // = / !=
  733. else
  734. {
  735. conModels.Add(new ConditionalModel()
  736. {
  737. FieldName = key.TrimEnd('!'),
  738. ConditionalType = key.EndsWith("!") ? ConditionalType.NoEqual : ConditionalType.Equal,
  739. FieldValue = fieldValue
  740. });
  741. }
  742. }
  743. }
  744. // 模糊搜索 "key$":"SQL搜索表达式" => "key$":["SQL搜索表达式"],任意SQL搜索表达式字符串,如 %key%(包含key), key%(以key开始), %k%e%y%(包含字母k,e,y) 等,%表示任意字符
  745. private void FuzzyQuery(string subtable, List<IConditionalModel> conModels, KeyValuePair<string, JToken> va)
  746. {
  747. var vakey = va.Key.Trim();
  748. var fieldValue = va.Value.ToString();
  749. var conditionalType = ConditionalType.Like;
  750. if (IsCol(subtable, vakey.TrimEnd('$')))
  751. {
  752. // 支持三种like查询
  753. if (fieldValue.StartsWith("%") && fieldValue.EndsWith("%"))
  754. {
  755. conditionalType = ConditionalType.Like;
  756. }
  757. else if (fieldValue.StartsWith("%"))
  758. {
  759. conditionalType = ConditionalType.LikeRight;
  760. }
  761. else if (fieldValue.EndsWith("%"))
  762. {
  763. conditionalType = ConditionalType.LikeLeft;
  764. }
  765. conModels.Add(new ConditionalModel() { FieldName = vakey.TrimEnd('$'), ConditionalType = conditionalType, FieldValue = fieldValue.TrimEnd("%".ToArray()).TrimStart("%".ToArray()) });
  766. }
  767. }
  768. // 处理sql注入
  769. private string ReplaceSQLChar(string str)
  770. {
  771. if (string.IsNullOrWhiteSpace(str))
  772. return string.Empty;
  773. str = str.Replace("'", "");
  774. str = str.Replace(";", "");
  775. str = str.Replace(",", "");
  776. str = str.Replace("?", "");
  777. str = str.Replace("<", "");
  778. str = str.Replace(">", "");
  779. str = str.Replace("(", "");
  780. str = str.Replace(")", "");
  781. str = str.Replace("@", "");
  782. str = str.Replace("=", "");
  783. str = str.Replace("+", "");
  784. str = str.Replace("*", "");
  785. str = str.Replace("&", "");
  786. str = str.Replace("#", "");
  787. str = str.Replace("%", "");
  788. str = str.Replace("$", "");
  789. str = str.Replace("\"", "");
  790. // 删除与数据库相关的词
  791. str = Regex.Replace(str, "delete from", "", RegexOptions.IgnoreCase);
  792. str = Regex.Replace(str, "drop table", "", RegexOptions.IgnoreCase);
  793. str = Regex.Replace(str, "truncate", "", RegexOptions.IgnoreCase);
  794. str = Regex.Replace(str, "xp_cmdshell", "", RegexOptions.IgnoreCase);
  795. str = Regex.Replace(str, "exec master", "", RegexOptions.IgnoreCase);
  796. str = Regex.Replace(str, "net localgroup administrators", "", RegexOptions.IgnoreCase);
  797. str = Regex.Replace(str, "net user", "", RegexOptions.IgnoreCase);
  798. str = Regex.Replace(str, "-", "", RegexOptions.IgnoreCase);
  799. str = Regex.Replace(str, "truncate", "", RegexOptions.IgnoreCase);
  800. return str;
  801. }
  802. /// <summary>
  803. /// 单条插入
  804. /// </summary>
  805. /// <param name="tableName"></param>
  806. /// <param name="cols"></param>
  807. /// <param name="role"></param>
  808. /// <returns>(各种类型的)id</returns>
  809. public object InsertSingle(string tableName, JObject cols, APIJSON_Role role = null)
  810. {
  811. role ??= _identitySvc.GetRole();
  812. var dt = new Dictionary<string, object>();
  813. foreach (var f in cols) // 遍历字段
  814. {
  815. if (//f.Key.ToLower() != "id" && //是否一定要传id
  816. IsCol(tableName, f.Key) &&
  817. (role.Insert.Column.Contains("*") || role.Insert.Column.Contains(f.Key, StringComparer.CurrentCultureIgnoreCase)))
  818. dt.Add(f.Key, FuncList.TransJObjectToSugarPara(f.Value));
  819. }
  820. // 如果外部没传Id,就后端生成或使用数据库默认值,如果都没有会出错
  821. object id;
  822. if (!dt.ContainsKey("id"))
  823. {
  824. id = YitIdHelper.NextId();//自己生成id的方法,可以由外部传入
  825. dt.Add("id", id);
  826. }
  827. else
  828. {
  829. id = dt["id"];
  830. }
  831. _db.Insertable(dt).AS(tableName).ExecuteCommand();//根据主键类型设置返回雪花或自增,目前返回条数
  832. return id;
  833. }
  834. /// <summary>
  835. /// 为每天记录创建udpate sql
  836. /// </summary>
  837. /// <param name="tableName"></param>
  838. /// <param name="record"></param>
  839. /// <param name="role"></param>
  840. /// <returns></returns>
  841. public int UpdateSingleRecord(string tableName, JObject record, APIJSON_Role role = null)
  842. {
  843. role ??= _identitySvc.GetRole();
  844. if (!record.ContainsKey("id"))
  845. throw Oops.Bah("未传主键id");
  846. var dt = new Dictionary<string, object>();
  847. var sb = new StringBuilder(100);
  848. object id = null;
  849. foreach (var f in record)//遍历每个字段
  850. {
  851. if (f.Key.Equals("id", StringComparison.OrdinalIgnoreCase))
  852. {
  853. if (f.Value is JArray)
  854. {
  855. sb.Append($"{f.Key} in (@{f.Key})");
  856. id = FuncList.TransJArrayToSugarPara(f.Value);
  857. }
  858. else
  859. {
  860. sb.Append($"{f.Key}=@{f.Key}");
  861. id = FuncList.TransJObjectToSugarPara(f.Value);
  862. }
  863. }
  864. else if (IsCol(tableName, f.Key) && (role.Update.Column.Contains("*") || role.Update.Column.Contains(f.Key, StringComparer.CurrentCultureIgnoreCase)))
  865. {
  866. dt.Add(f.Key, FuncList.TransJObjectToSugarPara(f.Value));
  867. }
  868. }
  869. string whereSql = sb.ToString();
  870. int count = _db.Updateable(dt).AS(tableName).Where(whereSql, new { id }).ExecuteCommand();
  871. return count;
  872. }
  873. /// <summary>
  874. /// 更新单表,支持同表多条记录
  875. /// </summary>
  876. /// <param name="tableName"></param>
  877. /// <param name="records"></param>
  878. /// <param name="role"></param>
  879. /// <returns></returns>
  880. public int UpdateSingleTable(string tableName, JToken records, APIJSON_Role role = null)
  881. {
  882. role ??= _identitySvc.GetRole();
  883. int count = 0;
  884. if (records is JArray)
  885. {
  886. foreach (var record in records.ToObject<JObject[]>())
  887. {
  888. count += UpdateSingleRecord(tableName, record, role);
  889. }
  890. }
  891. else
  892. {
  893. count = UpdateSingleRecord(tableName, records.ToObject<JObject>(), role);
  894. }
  895. return count;
  896. }
  897. }