using Apewer.Internals; using Newtonsoft.Json.Linq; using System; using System.Collections.Generic; using System.Data; using static Apewer.TextUtility; namespace Apewer.Source { /// System.Data.DataTable 装箱查询。 public sealed class Query : IQuery, IDisposable, IToJson { private bool _disposed = false; private bool _success = false; private string _message = null; private DataTable _table = null; private DataTable[] _tables = null; /// 创建实例,默认状态为失败。 public Query(bool success = false, string message = null) { _success = false; _message = message; } /// 创建实例,Exception 为 NULL 时成功,非 NULL 时失败。 public Query(Exception exception) { _success = exception == null; _message = RuntimeUtility.Message(exception); } /// 创建实例,包装一个 DataTable 对象,数据表为 NULL 时失败,非 NULL 时成功。 public Query(DataTable table) { _table = table; _success = table != null; _message = table == null ? "未获取有效的数据表。" : null; } /// 创建实例,包装一个 DataTable 对象。 public Query(DataTable table, bool success, string message = null) { _table = table; _success = success; _message = message; } /// 创建实例,包装多个 DataTable 对象。 public Query(DataTable[] tables, bool success = true, string message = null) { _tables = tables; _success = success; _message = message; if (tables != null && tables.Length > 0) _table = tables[0]; } /// 的隐式转换,判断 执行成功且包含表。 public static implicit operator bool(Query query) => query != null && query.Success && query.Table != null; /// 的隐式转换。 public static implicit operator Query(DataTable table) => new Query(table, table != null, null); /// 的隐式转换。 public static implicit operator DataTable(Query query) => query?.Table; #region Property /// 语句执行成功。 public bool Success { get => _success; } /// 消息。 public string Message { get => _message; } /// 所有结果表。 public DataTable[] Tables { get => _tables; } /// 获取默认结果表。如果设置默认结果表,会丢失设置前的所有结果表。 public DataTable Table { get => _table; } /// 默认表中的数据总行数。 public int Rows { get => _table == null ? 0 : _table.Rows.Count; } /// 默认表中的数据总列数。 public int Columns { get => _table == null ? 0 : _table.Columns.Count; } #endregion #region Value /// 获取默认表中指定单元格的内容。 /// 行索引,从 0 开始。 /// 列索引,从 0 开始。 public object Value(int rowIndex, int columnIndex) => Table.Value(rowIndex, columnIndex); /// 获取默认表中指定单元的内容。 /// 行索引,从 0 开始。 /// 列名称/字段名称,此名称不区分大小写。 public object Value(int rowIndex, string columnName) => Table.Value(rowIndex, columnName); #endregion #region Method /// 释放系统资源。 public void Dispose() { if (_disposed) return; if (_tables != null) { foreach (var table in _tables) RuntimeUtility.Dispose(table); _tables = null; } RuntimeUtility.Dispose(_table); _table = null; _tables = null; _disposed = true; // GC.SuppressFinalize(this); } #endregion #region 模型化、IToJson /// 转换为 Json 对象。 public Json ToJson() { var jsonObject = Json.NewObject(); jsonObject.SetProperty("success", _success); jsonObject.SetProperty("message", _message); if (Table != null) { var columns = SourceUtility.ToJson(Table.Columns); jsonObject.SetProperty("columns", columns); var rows = SourceUtility.ToJson(Table.Rows); jsonObject.SetProperty("rows", rows); } return jsonObject; } /// 转换为模型数组。 public ObjectSet[] ToArray() { var oss = null as ObjectSet[]; var table = _table; if (!_disposed && table != null) { var width = table.Columns.Count; var names = new string[width]; for (var c = 0; c < width; c++) { var name = table.Columns[c].ColumnName; if (name.IsEmpty()) continue; if (names.Contains(name)) continue; names[c] = name; } var unnamed = 1; for (var c = 0; c < width; c++) { if (names[c] != null) continue; while (true) { var name = "unnamed_" + unnamed.ToString(); unnamed++; if (names.Contains(name)) continue; names[c] = name; break; } } var height = table.Rows.Count; oss = new ObjectSet[height]; for (var r = 0; r < height; r++) { var os = new ObjectSet(); var dict = os.Origin; for (var c = 0; c < width; c++) { var v = Value(r, c); dict.Add(names[c], v.IsNull() ? null : v); } oss[r] = os; } } return oss; } #endregion } }