You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

861 lines
32 KiB

  1. #if NET40 || NET461
  2. /* 2020.9.4.0 */
  3. using Apewer;
  4. using Apewer.Source;
  5. using Externals.MySql.Data.MySqlClient;
  6. using System;
  7. using System.Collections.Generic;
  8. using System.Data;
  9. using System.Data.SqlClient;
  10. using System.Data.OleDb;
  11. using System.Data.Common;
  12. using System.Text;
  13. using System.Windows.Forms;
  14. using System.Windows.Media.Animation;
  15. namespace Apewer.Source
  16. {
  17. /// <summary></summary>
  18. public class MySql : IDatabase, IDisposable
  19. {
  20. #region fields & properties
  21. private const string EmptyString = TextUtility.EmptyString;
  22. private MySqlConnection _connection = null;
  23. private Timeout _timeout = new Timeout();
  24. private string _address = EmptyString;
  25. private string _store = EmptyString;
  26. private string _user = "root";
  27. private string _pass = EmptyString;
  28. /// <summary></summary>
  29. public string Address { get { return _address; } set { _address = TextUtility.AntiInject(value); } }
  30. /// <summary></summary>
  31. public string Store { get { return _store; } set { _store = TextUtility.AntiInject(value); } }
  32. /// <summary></summary>
  33. public string User { get { return _user; } set { _user = TextUtility.AntiInject(value); } }
  34. /// <summary></summary>
  35. public string Pass { get { return _pass; } set { _pass = TextUtility.AntiInject(value); } }
  36. /// <summary></summary>
  37. public Timeout Timeout { get { return _timeout; } set { _timeout = value; } }
  38. /// <summary></summary>
  39. public bool Online
  40. {
  41. get
  42. {
  43. if (_connection == null) return false;
  44. return _connection.State == ConnectionState.Open;
  45. }
  46. }
  47. /// <summary></summary>
  48. public MySql() { }
  49. /// <summary></summary>
  50. public MySql(string address, string store, string user, string pass = null)
  51. {
  52. Address = address;
  53. Store = store;
  54. User = user;
  55. Pass = pass;
  56. }
  57. #endregion
  58. #region methods
  59. private string CombineString()
  60. {
  61. return TextUtility.Merge("server=", _address, "; database=", _store, "; uid=", _user, "; pwd=", _pass, ";");
  62. }
  63. /// <summary></summary>
  64. public bool Connect()
  65. {
  66. if (_connection == null)
  67. {
  68. _connection = new MySqlConnection();
  69. _connection.ConnectionString = CombineString();
  70. }
  71. else
  72. {
  73. if (_connection.State == ConnectionState.Open) return true;
  74. }
  75. try
  76. {
  77. _connection.Open();
  78. switch (_connection.State)
  79. {
  80. case ConnectionState.Open: return true;
  81. default: return false;
  82. }
  83. }
  84. catch (Exception)
  85. {
  86. Close();
  87. return false;
  88. }
  89. }
  90. /// <summary></summary>
  91. public void Close()
  92. {
  93. if (_connection != null)
  94. {
  95. _connection.Close();
  96. _connection.Dispose();
  97. _connection = null;
  98. }
  99. }
  100. /// <summary></summary>
  101. public void Dispose() { Close(); }
  102. /// <summary></summary>
  103. public IQuery Query(string tsql, IEnumerable<IDataParameter> parameters)
  104. {
  105. if (tsql.IsBlank()) return Example.InvalidQueryStatement;
  106. const string table = "queryresult";
  107. var connected = Connect();
  108. if (!connected) return Example.InvalidQueryConnection;
  109. var query = new Query();
  110. try
  111. {
  112. var command = new MySqlCommand();
  113. command.Connection = _connection;
  114. command.CommandTimeout = _timeout.Query;
  115. command.CommandText = tsql;
  116. if (parameters != null)
  117. {
  118. foreach (var parameter in parameters)
  119. {
  120. if (parameter == null) continue;
  121. command.Parameters.Add(parameter);
  122. }
  123. }
  124. using (var ds = new DataSet())
  125. {
  126. using (var da = new MySqlDataAdapter(tsql, _connection))
  127. {
  128. da.Fill(ds, table);
  129. query.Table = ds.Tables[table];
  130. }
  131. }
  132. command.Dispose();
  133. query.Success = true;
  134. }
  135. catch (Exception exception)
  136. {
  137. query.Success = false;
  138. query.Exception = exception;
  139. }
  140. return query;
  141. }
  142. /// <summary></summary>
  143. public IExecute Execute(string tsql, IEnumerable<IDataParameter> parameters)
  144. {
  145. if (tsql.IsBlank()) return Example.InvalidExecuteStatement;
  146. var connected = Connect();
  147. if (!connected) return Example.InvalidExecuteConnection;
  148. var transaction = _connection.BeginTransaction();
  149. var execute = new Execute();
  150. try
  151. {
  152. var command = new MySqlCommand();
  153. command.Connection = _connection;
  154. command.Transaction = transaction;
  155. command.CommandTimeout = _timeout.Execute;
  156. command.CommandText = tsql;
  157. if (parameters != null)
  158. {
  159. foreach (var parameter in parameters)
  160. {
  161. if (parameter == null) continue;
  162. command.Parameters.Add(parameter);
  163. }
  164. }
  165. execute.Rows += command.ExecuteNonQuery();
  166. transaction.Commit();
  167. command.Dispose();
  168. execute.Success = true;
  169. }
  170. catch (Exception exception)
  171. {
  172. try { transaction.Rollback(); } catch { }
  173. execute.Success = false;
  174. execute.Exception = exception;
  175. }
  176. try { transaction.Dispose(); } catch { }
  177. return execute;
  178. }
  179. /// <summary></summary>
  180. public IQuery Query(string tsql)
  181. {
  182. return Query(tsql, null);
  183. }
  184. /// <summary></summary>
  185. public IExecute Execute(string tsql, IEnumerable<Parameter> parameters)
  186. {
  187. var dps = null as List<IDataParameter>;
  188. if (parameters != null)
  189. {
  190. var count = ClassUtility.Count(parameters);
  191. dps = new List<IDataParameter>(count);
  192. foreach (var p in parameters)
  193. {
  194. var dp = CreateDataParameter(p);
  195. dps.Add(dp);
  196. }
  197. }
  198. return Execute(tsql, dps);
  199. }
  200. /// <summary></summary>
  201. public IExecute Execute(string tsql)
  202. {
  203. return Execute(tsql, null as IEnumerable<IDataParameter>);
  204. }
  205. #endregion
  206. #region linq
  207. private List<string> QueryFirstColumn(string sql)
  208. {
  209. using (var query = Query(sql) as Query) return query.ReadColumn();
  210. }
  211. /// <summary></summary>
  212. public List<string> QueryAllTableNames()
  213. {
  214. var sql = TextUtility.Merge("select table_name from information_schema.tables where table_schema='", _store, "' and table_type='base table';");
  215. return QueryFirstColumn(sql);
  216. }
  217. /// <summary></summary>
  218. public List<string> QueryAllViewNames()
  219. {
  220. var sql = TextUtility.Merge("select table_name from information_schema.tables where table_schema='", _store, "' and table_type='view';");
  221. return QueryFirstColumn(sql);
  222. }
  223. /// <summary></summary>
  224. public List<string> QueryAllColumnNames(string table)
  225. {
  226. var sql = TextUtility.Merge("select column_name from information_schema.columns where table_schema='", _store, "' and table_name='", TextUtility.AntiInject(table), "';");
  227. return QueryFirstColumn(sql);
  228. }
  229. /// <summary>获取用于创建表的语句。</summary>
  230. private string GetCreateStetement(TableStructure structure)
  231. {
  232. // 检查现存表。
  233. var exists = false;
  234. var tables = QueryAllTableNames();
  235. if (tables.Count > 0)
  236. {
  237. var lower = structure.Table.ToLower();
  238. foreach (var table in tables)
  239. {
  240. if (TextUtility.IsBlank(table)) continue;
  241. if (table.ToLower() == lower)
  242. {
  243. exists = true;
  244. break;
  245. }
  246. }
  247. }
  248. if (exists)
  249. {
  250. var columns = QueryAllColumnNames(structure.Table);
  251. if (columns.Count > 0)
  252. {
  253. var lower = new List<string>(columns.Count);
  254. var added = 0;
  255. foreach (var column in columns)
  256. {
  257. if (TextUtility.IsBlank(column)) continue;
  258. lower.Add(column.ToLower());
  259. added++;
  260. }
  261. lower.Capacity = added;
  262. columns = lower;
  263. }
  264. var sqlsb = new StringBuilder();
  265. foreach (var column in structure.Columns.Values)
  266. {
  267. // 检查 Independent 特性。
  268. if (structure.Independent && column.Independent) continue;
  269. // 去重。
  270. var lower = column.Field.ToLower();
  271. if (columns.Contains(lower)) continue;
  272. var type = GetColumnDeclaration(column);
  273. if (type.IsEmpty()) return TextUtility.Merge("类型 ", column.Type.ToString(), " 不受支持。");
  274. // alter table `_record` add column `_index` bigint;
  275. sqlsb.Append("alter table `", structure.Table, "` add column ", type, "; ");
  276. }
  277. var sql = sqlsb.ToString();
  278. return sql;
  279. }
  280. else
  281. {
  282. // create table _record (`_index` bigint, `_key` varchar(255), `_text` longtext) engine=innodb default charset=utf8mb4
  283. var columns = new List<string>(structure.Columns.Count);
  284. var columnsAdded = 0;
  285. var primarykey = null as string;
  286. foreach (var kvp in structure.Columns)
  287. {
  288. var property = kvp.Key;
  289. var column = kvp.Value;
  290. // 检查 Independent 特性。
  291. if (structure.Independent && column.Independent) continue;
  292. // 字段。
  293. var type = GetColumnDeclaration(column);
  294. if (type.IsEmpty()) return TextUtility.Merge("类型 ", column.Type.ToString(), " 不受支持。");
  295. columns.Add(type);
  296. columnsAdded++;
  297. // 主键。
  298. if (property == "Key") primarykey = column.Field;
  299. }
  300. columns.Capacity = columnsAdded;
  301. var table = structure.Table;
  302. var joined = string.Join(", ", columns);
  303. // 设置主键。
  304. string sql;
  305. if (!structure.Independent && !string.IsNullOrEmpty(primarykey))
  306. {
  307. sql = TextUtility.Merge("create table `", table, "`(", joined, ", primary key (", primarykey, ") ) engine=innodb default charset=utf8mb4; ");
  308. }
  309. else
  310. {
  311. sql = TextUtility.Merge("create table `", table, "`(", joined, ") engine=innodb default charset=utf8mb4; ");
  312. }
  313. return sql;
  314. }
  315. }
  316. /// <summary></summary>
  317. private string CreateTable(Type model, out string sql)
  318. {
  319. if (model == null)
  320. {
  321. sql = "";
  322. return "指定的类型无效。";
  323. }
  324. var structure = null as TableStructure;
  325. try { structure = TableStructure.ParseModel(model); }
  326. catch (Exception exception)
  327. {
  328. sql = "";
  329. return exception.Message;
  330. }
  331. // 连接数据库。
  332. if (!Connect())
  333. {
  334. sql = "";
  335. return "连接数据库失败。";
  336. }
  337. sql = GetCreateStetement(structure);
  338. if (sql.NotEmpty())
  339. {
  340. var execute = Execute(sql);
  341. if (!execute.Success) return execute.Error;
  342. }
  343. return null;
  344. }
  345. /// <summary></summary>
  346. public string CreateTable(Type model)
  347. {
  348. return CreateTable(model, out string sql);
  349. }
  350. /// <summary></summary>
  351. public string CreateTable<T>() where T : Record
  352. {
  353. return CreateTable(typeof(T));
  354. }
  355. /// <summary></summary>
  356. public string CreateTable(Record model)
  357. {
  358. if (model == null) return "参数无效。";
  359. return CreateTable(model.GetType());
  360. }
  361. /// <summary></summary>
  362. /// <summary>插入记录。成功时候返回空字符串,发生异常时返回异常信息。</summary>
  363. public string Insert(Record entity)
  364. {
  365. if (entity == null) return "参数无效。";
  366. entity.FixProperties();
  367. var structure = null as TableStructure;
  368. try { structure = TableStructure.ParseModel(entity); }
  369. catch (Exception exception) { return exception.Message; }
  370. var parameters = structure.CreateDataParameters(entity, CreateDataParameter);
  371. var sql = GenerateInsertStatement(structure.Table, parameters);
  372. var execute = Execute(sql, parameters);
  373. if (execute.Success) return TextUtility.EmptyString;
  374. return execute.Error;
  375. }
  376. /// <summary>
  377. /// <para>更新记录,实体中的 Created 和 Key 属性不被更新。成功时返回空字符串,发生异常时返回异常信息。</para>
  378. /// <para>无法更新拥有 Independent 特性的模型。</para>
  379. /// </summary>
  380. public string Update(Record entity)
  381. {
  382. if (entity == null) return "参数无效。";
  383. entity.FixProperties();
  384. entity.Updated = ClockUtility.Lucid;
  385. var structure = null as TableStructure;
  386. try { structure = TableStructure.ParseModel(entity); }
  387. catch (Exception exception) { return exception.Message; }
  388. // 检查 Independent 特性。
  389. if (structure.Independent) return "无法更新拥有 Independent 特性的模型。";
  390. var parameters = structure.CreateDataParameters(entity, CreateDataParameter, "_created", "_key");
  391. var sql = GenerateUpdateStatement(structure, entity.Key, parameters);
  392. var execute = Execute(sql, parameters);
  393. if (execute.Success) return TextUtility.EmptyString;
  394. return execute.Error;
  395. }
  396. /// <summary></summary>
  397. public Result<List<T>> QueryRecords<T>(string tsql) where T : Record
  398. {
  399. if (tsql.IsEmpty()) return new Result<List<T>>(new ArgumentException());
  400. try
  401. {
  402. // 解析模型,抛出 Exception。
  403. TableStructure.ParseModel(typeof(T));
  404. var query = Query(tsql) as Query;
  405. var list = query.Fill<T>();
  406. query.Dispose();
  407. return new Result<List<T>>(list);
  408. }
  409. catch (Exception exception)
  410. {
  411. return new Result<List<T>>(exception);
  412. }
  413. }
  414. /// <summary></summary>
  415. public Result<T> QueryRecord<T>(string key, long flag = 0) where T : Record
  416. {
  417. var k = TextUtility.RestrictGuid(key);
  418. if (k.IsEmpty()) new Result<T>(new Exception("参数无效。"));
  419. try
  420. {
  421. var structure = TableStructure.ParseModel(typeof(T));
  422. var sqlflag = (flag == 0) ? TextUtility.EmptyString : TextUtility.Merge(" `_flag`=", flag.ToString(), " and");
  423. var sqlkey = TextUtility.Merge(" `_key`='", k, "'");
  424. var sql = TextUtility.Merge("select * from `", structure.Table, "` where ", sqlflag, sqlkey, " limit 1;");
  425. var result = QueryRecords<T>(sql);
  426. var list = result.Entity;
  427. if (list != null && list.Count > 0) return new Result<T>(list[0]);
  428. return new Result<T>(null, "无结果。");
  429. }
  430. catch (Exception exception)
  431. {
  432. return new Result<T>(exception);
  433. }
  434. }
  435. /// <summary>获取所有记录。Flag 为 0 时将忽略 Flag 条件。</summary>
  436. public Result<List<T>> QueryRecords<T>(long flag = 0) where T : Record
  437. {
  438. try
  439. {
  440. var structure = TableStructure.ParseModel(typeof(T));
  441. var sqlflag = (flag == 0) ? TextUtility.EmptyString : TextUtility.Merge(" where `_flag`=", flag.ToString());
  442. var sql = TextUtility.Merge("select * from `", structure.Table, "`", sqlflag, "; ");
  443. return QueryRecords<T>(sql);
  444. }
  445. catch (Exception exception)
  446. {
  447. return new Result<List<T>>(exception);
  448. }
  449. }
  450. /// <summary>查询有效的 Key 值。</summary>
  451. public Result<List<string>> QueryKeys(Type model, long flag = 0)
  452. {
  453. if (model != null)
  454. {
  455. try
  456. {
  457. var type = model;
  458. var ts = TableStructure.ParseModel(type);
  459. var sqlflag = (flag == 0) ? TextUtility.EmptyString : TextUtility.Merge(" where `_flag`=", flag.ToString());
  460. var sql = TextUtility.Merge("select `_key` from `", ts.Table, "`", sqlflag, "; ");
  461. using (var query = Query(sql) as Query)
  462. {
  463. var list = query.ReadColumn((r) => TextUtility.Trim(query.Text(r, 0)));
  464. return new Result<List<string>>(list);
  465. }
  466. }
  467. catch (Exception argException)
  468. {
  469. return new Result<List<string>>(argException);
  470. }
  471. }
  472. return new Result<List<string>>(new Exception("参数无效。"));
  473. }
  474. /// <summary>查询有效的 Key 值。</summary>
  475. public Result<List<string>> QueryKeys<T>(long flag = 0) where T : Record
  476. {
  477. try
  478. {
  479. var type = typeof(T);
  480. var ts = TableStructure.ParseModel(type);
  481. var sqlflag = (flag == 0) ? TextUtility.EmptyString : TextUtility.Merge(" where _flag=", flag.ToString());
  482. var sql = TextUtility.Merge("select `_key` from `", ts.Table, "`", sqlflag, "; ");
  483. using (var query = Query(sql) as Query)
  484. {
  485. var list = query.ReadColumn((r) => TextUtility.Trim(query.Text(r, 0)));
  486. return new Result<List<string>>(list);
  487. }
  488. }
  489. catch (Exception ex)
  490. {
  491. return new Result<List<string>>(ex);
  492. }
  493. }
  494. #endregion
  495. #region static
  496. /// <summary></summary>
  497. /// <exception cref="ArgumentNullException"></exception>
  498. /// <exception cref="InvalidOperationException"></exception>
  499. internal static MySqlParameter CreateDataParameter(Parameter parameter)
  500. {
  501. if (parameter == null) throw new InvalidOperationException("参数无效。");
  502. return CreateDataParameter(parameter.Name, parameter.Type, parameter.Size, parameter.Value);
  503. }
  504. /// <summary></summary>
  505. internal static MySqlParameter CreateDataParameter(string name, ColumnType type, Int32 size, object value)
  506. {
  507. if (TextUtility.IsBlank(name)) return null;
  508. var dbtype = MySqlDbType.Int64;
  509. switch (type)
  510. {
  511. case ColumnType.Binary:
  512. dbtype = MySqlDbType.LongBlob;
  513. break;
  514. case ColumnType.Integer:
  515. dbtype = MySqlDbType.Int64;
  516. break;
  517. case ColumnType.Float:
  518. dbtype = MySqlDbType.Double;
  519. break;
  520. case ColumnType.DateTime:
  521. dbtype = MySqlDbType.DateTime;
  522. break;
  523. case ColumnType.VarChar:
  524. case ColumnType.VarChar255:
  525. case ColumnType.VarCharMax:
  526. case ColumnType.NVarChar:
  527. case ColumnType.NVarChar255:
  528. case ColumnType.NVarCharMax:
  529. dbtype = MySqlDbType.VarChar;
  530. break;
  531. case ColumnType.Text:
  532. case ColumnType.NText:
  533. dbtype = MySqlDbType.LongText;
  534. break;
  535. default:
  536. throw new InvalidOperationException(TextUtility.Merge("类型 ", type.ToString(), " 不受支持。"));
  537. }
  538. switch (type)
  539. {
  540. case ColumnType.VarChar:
  541. case ColumnType.NVarChar:
  542. size = NumberUtility.RestrictValue(size, 0, 65535);
  543. break;
  544. case ColumnType.VarChar255:
  545. case ColumnType.NVarChar255:
  546. size = NumberUtility.RestrictValue(size, 0, 255);
  547. break;
  548. default:
  549. size = 0;
  550. break;
  551. }
  552. if (value is string && value != null && size > 0)
  553. {
  554. value = TextUtility.RestrictLength((string)value, size);
  555. }
  556. var parameter = new MySqlParameter();
  557. parameter.ParameterName = name;
  558. parameter.MySqlDbType = dbtype;
  559. parameter.Value = value;
  560. if (size > 0) parameter.Size = size;
  561. return parameter;
  562. }
  563. /// <summary></summary>
  564. internal static MySqlParameter CreateDataParameter(string name, MySqlDbType type, object value, Int32 size = 0)
  565. {
  566. var parameter = new MySqlParameter();
  567. parameter.ParameterName = name;
  568. parameter.MySqlDbType = type;
  569. parameter.Value = value;
  570. if (size > 0) parameter.Size = size;
  571. return parameter;
  572. }
  573. private static string GetColumnDeclaration(ColumnAttribute column)
  574. {
  575. var type = TextUtility.EmptyString;
  576. var length = Math.Max(0, (int)column.Length);
  577. switch (column.Type)
  578. {
  579. case ColumnType.Integer:
  580. type = "bigint";
  581. break;
  582. case ColumnType.Float:
  583. type = "double";
  584. break;
  585. case ColumnType.Binary:
  586. type = "longblob";
  587. break;
  588. case ColumnType.DateTime:
  589. type = "datetime";
  590. break;
  591. case ColumnType.VarChar:
  592. type = TextUtility.Merge("varchar(", Math.Max(65535, length).ToString(), ")");
  593. break;
  594. case ColumnType.VarChar255:
  595. type = TextUtility.Merge("varchar(255)");
  596. break;
  597. case ColumnType.VarCharMax:
  598. type = TextUtility.Merge("varchar(max)");
  599. break;
  600. case ColumnType.Text:
  601. type = TextUtility.Merge("longtext");
  602. break;
  603. case ColumnType.NVarChar:
  604. type = TextUtility.Merge("varchar(", Math.Min(65535, length).ToString(), ")");
  605. break;
  606. case ColumnType.NVarChar255:
  607. type = TextUtility.Merge("varchar(255)");
  608. break;
  609. case ColumnType.NVarCharMax:
  610. type = TextUtility.Merge("varchar(65535)");
  611. break;
  612. case ColumnType.NText:
  613. type = TextUtility.Merge("longtext");
  614. break;
  615. default:
  616. return TextUtility.EmptyString;
  617. }
  618. return TextUtility.Merge("`", (string)column.Field, "` ", type);
  619. }
  620. private static string GetParameterName(string argParameter)
  621. {
  622. var name = TextUtility.AntiInject(argParameter, 255);
  623. if (name.StartsWith("@") && name.Length > 1)
  624. {
  625. name = name.Substring(1, name.Length - 1);
  626. }
  627. return name;
  628. }
  629. private static string GetParameterName(IDataParameter argParameter)
  630. {
  631. var name = TextUtility.EmptyString;
  632. if (argParameter != null)
  633. {
  634. name = GetParameterName(argParameter.ParameterName);
  635. }
  636. return name;
  637. }
  638. private static List<string> GetParametersNames(IEnumerable<IDataParameter> argParameters)
  639. {
  640. var columns = new List<string>();
  641. if (argParameters != null)
  642. {
  643. columns.Capacity = ClassUtility.Count(argParameters);
  644. var columnsAdded = 0;
  645. foreach (var parameter in argParameters)
  646. {
  647. var name = GetParameterName(parameter);
  648. var isblank = TextUtility.IsBlank(name);
  649. if (isblank) continue;
  650. columns.Add(name);
  651. columnsAdded++;
  652. }
  653. columns.Capacity = columnsAdded;
  654. }
  655. return columns;
  656. }
  657. private static string GenerateInsertStatement(string argTable, List<string> argColumns)
  658. {
  659. var result = TextUtility.EmptyString;
  660. var table = TextUtility.AntiInject(argTable, 255);
  661. if (argColumns != null && !TextUtility.IsBlank(table))
  662. {
  663. var count = argColumns.Count;
  664. var names = new List<string>(count);
  665. var values = new List<string>(count);
  666. foreach (var column in argColumns)
  667. {
  668. //names.Add(TextGenerator.Merge("[", column, "]"));
  669. names.Add(TextUtility.Merge(column));
  670. values.Add("@" + column);
  671. }
  672. var text = new StringBuilder();
  673. if (count > 0)
  674. {
  675. text.Append("insert into `", table, "`(", string.Join(", ", names), ") ");
  676. text.Append("values(", string.Join(", ", values), "); ");
  677. }
  678. result = text.ToString();
  679. }
  680. return result;
  681. }
  682. private static string GenerateUpdateStatement(TableStructure structure, string key, List<string> columns)
  683. {
  684. var result = TextUtility.EmptyString;
  685. var table = TextUtility.AntiInject(structure.Table, 255);
  686. if (TextUtility.IsEmpty(table)) return result;
  687. var safekey = TextUtility.AntiInject(key, 255);
  688. if (TextUtility.IsEmpty(safekey)) return result;
  689. var count = columns == null ? -1 : columns.Count;
  690. if (count < 1) return result;
  691. var items = new List<string>(count);
  692. foreach (var column in columns)
  693. {
  694. items.Add(TextUtility.Merge("`", column, "`=@", column));
  695. }
  696. result = TextUtility.Merge("update `", table, "` set ", string.Join(", ", items), " where `_key`='", safekey, "'; ");
  697. return result;
  698. }
  699. /// <summary>生成 INSERT INTO 语句。表名必须有效,无有效参数时将获取空结果。</summary>
  700. /// <exception cref="System.ArgumentException"></exception>
  701. /// <exception cref="System.ArgumentNullException"></exception>
  702. public static string GenerateInsertStatement(string argTable, IEnumerable<IDataParameter> parameters)
  703. {
  704. if (argTable == null) throw new ArgumentNullException("argTable");
  705. var table = TextUtility.AntiInject(argTable, 255);
  706. if (TextUtility.IsBlank(table)) throw new ArgumentException("表名无效。", "argTable");
  707. var columns = GetParametersNames(parameters);
  708. if (columns.Count < 1) return TextUtility.EmptyString;
  709. return GenerateInsertStatement(table, columns);
  710. }
  711. /// <summary>生成 UPDATE 语句,键字段名为“_key”。表名必须有效,键值必须有效,无有效参数时将获取空结果。</summary>
  712. /// <exception cref="System.ArgumentException"></exception>
  713. /// <exception cref="System.ArgumentNullException"></exception>
  714. public static string GenerateUpdateStatement(TableStructure structure, string key, IEnumerable<IDataParameter> parameters)
  715. {
  716. if (structure == null) throw new ArgumentNullException("structure");
  717. if (key == null) throw new ArgumentNullException("key");
  718. var table = TextUtility.AntiInject(structure.Table, 255);
  719. if (TextUtility.IsBlank(table)) throw new ArgumentException("表名无效。", "structure");
  720. var safekey = TextUtility.AntiInject(key, 255);
  721. if (TextUtility.IsBlank(safekey)) throw new ArgumentException("键值无效。", "key");
  722. var columns = GetParametersNames(parameters);
  723. if (columns.Count < 1) return TextUtility.EmptyString;
  724. return GenerateUpdateStatement(structure, safekey, columns);
  725. }
  726. /// <summary>获取每个数据库中,每个表的容量,单位为字节。</summary>
  727. public static Dictionary<string, Dictionary<string, long>> GetTablesCapacity(MySql source)
  728. {
  729. var result = new Dictionary<string, Dictionary<string, long>>();
  730. if (source != null && source.Connect())
  731. {
  732. var sql = "select `table_schema`, `table_name`, `engine`, `data_length`, `index_length` from `information_schema`.tables order by `table_schema`, `table_name`";
  733. using (var query = (Query)source.Query(sql))
  734. {
  735. for (var r = 0; r < query.Rows; r++)
  736. {
  737. var store = query.Text(r, "table_schema");
  738. var table = query.Text(r, "table_name");
  739. var engine = query.Text(r, "engine");
  740. var capacity = TextUtility.GetInt64(query.Text(r, "data_length")) + TextUtility.GetInt64(query.Text(r, "index_length"));
  741. if (store == "mysql") continue;
  742. if (store == "information_schema") continue;
  743. if (store == "performance_schema") continue;
  744. if (engine != "MyISAM" && engine != "InnoDB") continue;
  745. if (!result.ContainsKey(store)) result.Add(store, new Dictionary<string, long>());
  746. if (!result[store].ContainsKey(table)) result[store].Add(table, 0L);
  747. result[store][table] = capacity;
  748. }
  749. }
  750. }
  751. return result;
  752. }
  753. #endregion
  754. }
  755. }
  756. #endif