#if MYSQL_6_9 || MYSQL_6_10 /* 2021.11.07 */ using Externals.MySql.Data.MySqlClient; using System; using System.Collections.Generic; using System.Data; using System.Net; using System.Security.Cryptography.X509Certificates; using System.Text; using System.Transactions; using static Apewer.Source.SourceUtility; namespace Apewer.Source { /// public class MySql : DbClient { #region connection private string _connstr = null; /// 创建实例。 public MySql(string connnectionString, Timeout timeout = default) : base(timeout) { _connstr = connnectionString; } /// 构建连接字符串以创建实例。 public MySql(string address, string store, string user, string pass, Timeout timeout = null) : base(timeout) { var a = address ?? ""; var s = store ?? ""; var u = user ?? ""; var p = pass ?? ""; var cs = $"server={a}; database={s}; uid={u}; pwd={p}; "; _connstr = cs; } #endregion #region override /// protected override string NewConnectionString() => _connstr; /// protected override IDataAdapter CreateDataAdapter(IDbCommand command) => new MySqlDataAdapter((MySqlCommand)command); /// protected override IDbConnection CreateConnection() => new MySqlConnection(); /// protected override IDataParameter CreateParameter() => new MySqlParameter(); /// public override string[] StoreNames() { throw new NotImplementedException(); } /// public override string[] TableNames() { var store = StoreName(); var sql = $"select table_name from information_schema.tables where table_schema='{store}' and table_type='base table'"; return TextColumn(sql); } /// public override string[] ColumnNames(string tableName) { var store = StoreName(); var table = TextUtility.AntiInject(tableName); var sql = $"select column_name from information_schema.columns where table_schema='{store}' and table_name='{table}'"; return TextColumn(sql); } /// protected override string Initialize(TableStructure structure) { // 检查现存表。 var exists = false; var tables = TableNames(); if (tables.Length > 0) { var lower = structure.Name.ToLower(); foreach (var table in tables) { if (TextUtility.IsBlank(table)) continue; if (table.ToLower() == lower) { exists = true; break; } } } if (exists) { var columns = ColumnNames(structure.Name); if (columns.Length > 0) { var lower = new List(columns.Length); var added = 0; foreach (var column in columns) { if (TextUtility.IsBlank(column)) continue; lower.Add(column.ToLower()); added++; } lower.Capacity = added; columns = lower.ToArray(); } var sqlsb = new StringBuilder(); foreach (var column in structure.Columns) { // 检查 Independent 特性。 if (structure.Independent && column.Independent) continue; // 去重。 var lower = column.Field.ToLower(); if (columns.Contains(lower)) continue; var type = Declaration(column); if (type.IsEmpty()) return TextUtility.Merge("类型 ", column.Type.ToString(), " 不受支持。"); // alter table `_record` add column `_index` bigint; sqlsb.Append("alter table `", structure.Name, "` add column ", type, "; "); } var sql = sqlsb.ToString(); if (sql.IsEmpty()) return null; var execute = Execute(sql); return execute.Success ? null : execute.Message; } else { // create table _record (`_index` bigint, `_key` varchar(255), `_text` longtext) engine=innodb default charset=utf8mb4 var columns = new List(structure.Columns.Length); var columnsAdded = 0; var primarykeys = new List(); foreach (var column in structure.Columns) { // 检查 Independent 特性。 if (structure.Independent && column.Independent) continue; // 字段。 var type = Declaration(column); if (type.IsEmpty()) return TextUtility.Merge("类型 ", column.Type.ToString(), " 不受支持。"); if (!column.Independent) { if (column.PrimaryKey) primarykeys.Add(column.Field); if (column.Incremental) type += " auto_increment"; } columns.Add(type); columnsAdded++; } columns.Capacity = columnsAdded; var table = structure.Name; var joined = string.Join(", ", columns); // 设置主键。 string sql; var sqlPrimaryKey = primarykeys.Count < 1 ? "" : (", primary key (" + string.Join(", ", primarykeys.ToArray()) + ")"); sql = TextUtility.Merge("create table `", table, "`(", joined, sqlPrimaryKey, ") engine=innodb default charset=utf8mb4; "); var execute = Execute(sql); return execute.Success ? null : execute.Message; } } /// 插入记录。返回错误信息。 public override string Insert(object record, string table = null) { if (record == null) return "参数无效。"; SourceUtility.FixProperties(record); var structure = TableStructure.Parse(record.GetType()); if (structure == null) return "无法解析记录模型。"; if (string.IsNullOrEmpty(table)) table = structure.Name; if (string.IsNullOrEmpty(table)) return "表名称无效。"; // 排除字段。 var excluded = new List(); foreach (var ca in structure.Columns) { // 排除不使用 ORM 的属性。 if (ca.Independent || ca.Incremental) excluded.Add(ca.Field); } var ps = structure.CreateParameters(record, Parameter, excluded); var psc = ps.Length; if (psc < 1) return "数据模型不包含字段。"; var names = new List(psc); var values = new List(psc); foreach (var p in ps) { var pn = p.ParameterName; names.Add("`" + p + "`"); values.Add("@" + p); } var ns = string.Join(", ", names); var vs = string.Join(", ", values); var sql = $"insert into `{table}` ({ns}) values ({vs}); "; var execute = Execute(sql, ps); if (execute.Success) return TextUtility.Empty; return execute.Message; } /// 更新记录,实体中的 Key 属性不被更新。返回错误信息。 /// 无法更新带有 Independent 特性的模型(缺少 Key 属性)。 public override string Update(IRecord record, string table = null) { if (record == null) return "参数无效。"; FixProperties(record); SetUpdated(record); var structure = TableStructure.Parse(record.GetType()); if (structure == null) return "无法解析记录模型。"; if (structure.Independent) return "无法更新带有 Independent 特性的模型。"; if (string.IsNullOrEmpty(table)) table = structure.Name; if (string.IsNullOrEmpty(table)) return "表名称无效。"; // 排除字段。 var excluded = new List(); if (structure.Key != null) excluded.Add(structure.Key.Field); foreach (var ca in structure.Columns) { // 排除不使用 ORM 的属性、自增属性和主键属性。 if (ca.Independent || ca.Incremental || ca.PrimaryKey) excluded.Add(ca.Field); } var ps = structure.CreateParameters(record, Parameter, excluded); var psc = ps.Length; if (psc < 1) return "数据模型不包含字段。"; var items = new List(psc); foreach (var p in ps) { var pn = p.ParameterName; items.Add(TextUtility.Merge("`", pn, "` = @", pn)); } var key = record.Key.SafeKey(); var sql = $"update `{table}` set {string.Join(", ", items)} where `{structure.Key.Field}`='{key}'; "; var execute = Execute(sql, ps); if (execute.Success) return TextUtility.Empty; return execute.Message; } /// protected override string KeysSql(string tableName, string keyField, string flagField, long flagValue) { if (flagValue == 0) return $"select `{keyField}` from `{tableName}`"; else return $"select `{keyField}` from `{tableName}` where `{flagField}` = {flagValue}"; } /// protected override string RecordSql(string tableName, string keyField, string keyValue, string flagField, long flagValue) { if (flagValue == 0) return $"select * from `{tableName}` where `{keyField}` = '{keyValue}' limit 1"; else return $"select * from `{tableName}` where `{keyField}` = '{keyValue}' and `{flagField}` = {flagValue} limit 1"; } /// protected override string RecordsSql(string tableName, string flagField, long flagValue) { if (flagValue == 0) return $"select * from `{tableName}`"; else return $"select * from `{tableName}` where `{flagField}` = {flagValue}"; } #endregion #region special string StoreName() => Internals.TextHelper.ParseConnectionString(_connstr).GetValue("database") ?? ""; /// 获取所有视图的名称。 public string[] ViewNames() { var store = StoreName(); var sql = $"select table_name from information_schema.tables where table_schema='{store}' and table_type='view'"; return TextColumn(sql); } /// 获取记录。 /// 填充的记录模型。 /// 要跳过的记录数,可用最小值为 0。 /// 要获取的记录数,可用最小值为 1。 public Result Range(Type model, int skip, int count) where T : class, new() { if (model == null) return new Result("参数 model 无效。"); if (skip < 0) return new Result("参数 skip 超出了范围。"); if (count < 1) return new Result("参数 count 超出了范围。"); var ts = TableStructure.Parse(model); if (ts.Name.IsEmpty()) return new Result($"无法解析类型 {model.FullName}。"); var sql = $"select * from `{ts.Name}` limit {skip}, {count}"; using (var query = Query(sql)) { if (!query.Success) return new Result(query.Message); return new Result(query.Fill()); } } /// 对表添加列,返回错误信息。 /// 记录类型。 /// 列名称。 /// 字段类型。 /// 字段长度,仅对 VarChar 和 NVarChar 类型有效。 /// public string AddColumn(string column, ColumnType type, int length = 0) where T : class, IRecord { var columnName = column.SafeName(); if (columnName.IsEmpty()) return "列名无效。"; var ta = TableAttribute.Parse(typeof(T)); if (ta == null) return "无法解析记录模型。"; var tableName = ta.Name; var columeType = ""; switch (type) { case ColumnType.Integer: columeType = "bigint"; break; case ColumnType.Float: columeType = "double"; break; case ColumnType.Bytes: columeType = "longblob"; break; case ColumnType.DateTime: columeType = "datetime"; break; case ColumnType.VarChar: case ColumnType.NVarChar: columeType = $"varchar({length})"; break; case ColumnType.VarChar191: case ColumnType.NVarChar191: columeType = "varchar(191)"; break; case ColumnType.VarCharMax: case ColumnType.NVarCharMax: columeType = "varchar(max)"; break; case ColumnType.Text: columeType = "longtext"; break; } if (columeType.IsEmpty()) return "类型不支持。"; var sql = $"alter table `{tableName}` add {columnName} {columeType}; "; var execute = Execute(sql) as Execute; var error = execute.Message; return error; } #endregion #region static /// /// /// internal static MySqlParameter Parameter(Parameter parameter) { if (parameter == null) throw new InvalidOperationException("参数无效。"); return Parameter(parameter.Name, parameter.Type, parameter.Size, parameter.Value); } /// internal static MySqlParameter Parameter(string name, ColumnType type, Int32 size, object value) { if (TextUtility.IsBlank(name)) return null; var dbtype = MySqlDbType.Int64; switch (type) { case ColumnType.Bytes: dbtype = MySqlDbType.LongBlob; break; case ColumnType.Integer: dbtype = MySqlDbType.Int64; break; case ColumnType.Float: dbtype = MySqlDbType.Double; break; case ColumnType.DateTime: dbtype = MySqlDbType.DateTime; break; case ColumnType.VarChar: case ColumnType.VarChar191: case ColumnType.VarCharMax: case ColumnType.NVarChar: case ColumnType.NVarChar191: case ColumnType.NVarCharMax: dbtype = MySqlDbType.VarChar; break; case ColumnType.Text: case ColumnType.NText: dbtype = MySqlDbType.LongText; break; default: throw new InvalidOperationException(TextUtility.Merge("类型 ", type.ToString(), " 不受支持。")); } switch (type) { case ColumnType.VarChar: case ColumnType.NVarChar: size = NumberUtility.Restrict(size, 0, 65535); break; case ColumnType.VarChar191: case ColumnType.NVarChar191: size = NumberUtility.Restrict(size, 0, 191); break; default: size = 0; break; } if (value is string && value != null && size > 0) { value = TextUtility.Left((string)value, size); } var parameter = new MySqlParameter(); parameter.ParameterName = name; parameter.MySqlDbType = dbtype; parameter.Value = value; if (size > 0) parameter.Size = size; return parameter; } /// internal static MySqlParameter Parameter(string name, MySqlDbType type, object value, Int32 size = 0) { var parameter = new MySqlParameter(); parameter.ParameterName = name; parameter.MySqlDbType = type; parameter.Value = value; if (size > 0) parameter.Size = size; return parameter; } private static string Declaration(ColumnAttribute column) { var type = TextUtility.Empty; var length = Math.Max(0, (int)column.Length); switch (column.Type) { case ColumnType.Integer: type = "bigint"; break; case ColumnType.Float: type = "double"; break; case ColumnType.Bytes: type = "longblob"; break; case ColumnType.DateTime: type = "datetime"; break; case ColumnType.VarChar: type = TextUtility.Merge("varchar(", Math.Max(65535, length).ToString(), ")"); break; case ColumnType.VarChar191: type = TextUtility.Merge("varchar(191)"); break; case ColumnType.VarCharMax: type = TextUtility.Merge("varchar(max)"); break; case ColumnType.Text: type = TextUtility.Merge("longtext"); break; case ColumnType.NVarChar: type = TextUtility.Merge("varchar(", Math.Min(65535, length).ToString(), ")"); break; case ColumnType.NVarChar191: type = TextUtility.Merge("varchar(191)"); break; case ColumnType.NVarCharMax: type = TextUtility.Merge("varchar(65535)"); break; case ColumnType.NText: type = TextUtility.Merge("longtext"); break; default: return TextUtility.Empty; } return TextUtility.Merge("`", (string)column.Field, "` ", type); } /// 获取每个数据库中,每个表的容量,单位为字节。 public static Dictionary> GetTablesCapacity(MySql source) { var result = new Dictionary>(); if (source != null && source.Connect().IsEmpty()) { var sql = "select `table_schema`, `table_name`, `engine`, `data_length`, `index_length` from `information_schema`.tables order by `table_schema`, `table_name`"; using (var query = (Query)source.Query(sql)) { for (var r = 0; r < query.Rows; r++) { var store = query.Text(r, "table_schema"); var table = query.Text(r, "table_name"); var engine = query.Text(r, "engine"); var capacity = NumberUtility.Int64(query.Text(r, "data_length")) + NumberUtility.Int64(query.Text(r, "index_length")); if (store == "mysql") continue; if (store == "information_schema") continue; if (store == "performance_schema") continue; if (engine != "MyISAM" && engine != "InnoDB") continue; if (!result.ContainsKey(store)) result.Add(store, new Dictionary()); if (!result[store].ContainsKey(table)) result[store].Add(table, 0L); result[store][table] = capacity; } } } return result; } #endregion } } #endif