/* 2021.11.07 */
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Data.SQLite;
using System.Text;
//using Mono.Data.Sqlite;
using static Apewer.Source.OrmHelper;
namespace Apewer.Source
{
/// 用于快速连接 SQLite 数据库的辅助。
public class Sqlite : IDbClient
{
#region 基础
private Timeout _timeout = null;
private string _connstring = "";
private string _path = "";
private string _pass = "";
private object _locker = new object();
/// 获取或设置日志记录。
public Logger Logger { get; set; }
/// 超时设定。
public Timeout Timeout { get => _timeout; }
/// 创建连接实例。
/// 注意:
- 构造函数不会创建不存在的文件;
- 参数 path 为文件路径,指定为空时将使用 :memory: 作为路径连接内存。
public Sqlite(string path = null, string pass = null, Timeout timeout = null)
{
_timeout = timeout ?? Timeout.Default;
_path = path.IsEmpty() ? Memory : path;
_pass = pass;
if (pass.IsEmpty()) _connstring = $"data source='{_path}'; version=3; ";
else _connstring = $"data source='{_path}'; password={_pass}; version=3; ";
}
#endregion
#region 连接
private SQLiteConnection _db = null;
/// 数据库已经连接。
public bool Online { get => _db != null && _db.State == ConnectionState.Open; }
/// 连接字符串。
public string ConnectionString { get => _connstring; }
/// 获取当前的 SQLiteConnection 对象。
public IDbConnection Connection { get => _db; }
/// 连接数据库,若未连接则尝试连接。
/// 是否已连接。
public string Connect()
{
if (_db == null)
{
_db = new SQLiteConnection();
_db.ConnectionString = ConnectionString;
}
else
{
if (_db.State == ConnectionState.Open) return null;
}
try
{
_db.Open();
switch (_db.State)
{
case ConnectionState.Open: return null;
default: return $"连接失败,当前处于 {_db.State} 状态。";
}
}
catch (Exception ex)
{
Logger.Error(nameof(Sqlite), "Connection", ex, _db.ConnectionString);
Close();
return ex.Message;
}
}
/// 关闭连接,并释放对象所占用的系统资源。
public void Close()
{
if (_db != null)
{
if (_transaction != null)
{
if (_autocommit) Commit();
else Rollback();
}
lock (_locker)
{
_db.Dispose();
}
_db = null;
}
}
/// 关闭连接,释放对象所占用的系统资源,并清除连接信息。
public void Dispose() { Close(); }
#endregion
#region Transaction
private IDbTransaction _transaction = null;
private bool _autocommit = false;
/// 启动事务。
public string Begin(bool commit = true) => Begin(commit, null);
/// 启动事务。
public string Begin(bool commit, Class isolation)
{
var connect = Connect();
if (connect.NotEmpty()) return connect;
if (_transaction != null) return "存在已启动的事务,无法再次启动。";
try
{
_transaction = isolation ? _db.BeginTransaction(isolation.Value) : _db.BeginTransaction();
_autocommit = commit;
return null;
}
catch (Exception ex)
{
Logger.Error(nameof(Sqlite), "Begin", ex.Message());
return ex.Message();
}
}
/// 提交事务。
public string Commit()
{
if (_transaction == null) return "事务不存在。";
try
{
_transaction.Commit();
RuntimeUtility.Dispose(_transaction);
_transaction = null;
return null;
}
catch (Exception ex)
{
RuntimeUtility.Dispose(_transaction);
_transaction = null;
Logger.Error(nameof(Sqlite), "Commit", ex.Message());
return ex.Message();
}
}
/// 从挂起状态回滚事务。
public string Rollback()
{
if (_transaction == null) return "事务不存在。";
try
{
_transaction.Rollback();
RuntimeUtility.Dispose(_transaction);
_transaction = null;
return null;
}
catch (Exception ex)
{
RuntimeUtility.Dispose(_transaction);
_transaction = null;
Logger.Error(nameof(Sqlite), "Rollback", ex.Message());
return ex.Message();
}
}
#endregion
#region SQL
/// 查询。
public IQuery Query(string sql) => Query(sql, null);
/// 查询。
public IQuery Query(string sql, IEnumerable parameters)
{
if (string.IsNullOrEmpty(sql)) return Example.InvalidQueryStatement;
var connected = Connect();
if (connected.NotEmpty()) return new Query(false, connected);
var query = new Query();
try
{
using (var command = new SQLiteCommand())
{
command.Connection = _db;
command.CommandTimeout = _timeout.Query;
command.CommandText = sql;
if (parameters != null)
{
foreach (var p in parameters)
{
if (p != null) command.Parameters.Add(p);
}
}
using (var dataset = new DataSet())
{
using (var da = new SQLiteDataAdapter(command))
{
var name = "table_" + Guid.NewGuid().ToString("n");
da.Fill(dataset, name);
var table = dataset.Tables[name];
return new Query(table);
}
}
}
}
catch (Exception ex)
{
Logger.Error(nameof(Sqlite), "Query", ex, sql);
return new Query(ex);
}
}
/// 执行单条 Transact-SQL 语句。
public IExecute Execute(string sql) => Execute(sql, null);
/// 执行单条 Transact-SQL 语句,并加入参数。
public IExecute Execute(string sql, IEnumerable parameters)
{
if (string.IsNullOrEmpty(sql)) return Example.InvalidExecuteStatement;
var connected = Connect();
if (connected.NotEmpty()) return new Execute(false, connected);
lock (_locker)
{
var inTransaction = _transaction != null;
if (!inTransaction) Begin();
try
{
using (var command = new SQLiteCommand())
{
command.Connection = _db;
command.Transaction = (SQLiteTransaction)_transaction;
command.CommandTimeout = _timeout.Execute;
command.CommandText = sql;
if (parameters != null)
{
foreach (var p in parameters)
{
if (p != null) command.Parameters.Add(p);
}
}
var rows = command.ExecuteNonQuery();
if (!inTransaction) Commit(); // todo 此处应该检查事务提交产生的错误。
return new Execute(true, rows);
}
}
catch (Exception ex)
{
Logger.Error(nameof(Sqlite), "Execute", ex, sql);
if (!inTransaction) Rollback();
return new Execute(ex);
}
}
}
#endregion
#region 属性。
/// 保存当前数据库到文件,若文件已存在则将重写文件。
public bool Save(string path, string pass = null)
{
if (!StorageUtility.CreateFile(path, 0, true))
{
Logger.Error(nameof(Sqlite), "Save", TextUtility.Merge("创建文件 ", path, " 失败。"));
return false;
}
using (var destination = new Sqlite(path, pass)) return Save(destination);
}
/// 保存当前数据库到目标数据库。
public bool Save(Sqlite destination) => string.IsNullOrEmpty(Backup(this, destination));
/// 加载文件到当前数据库。
public bool Load(string path, string pass = null)
{
using (var source = new Sqlite(path, pass)) return Load(source);
}
/// 加载源数据库到当前数据库。
public bool Load(Sqlite source) => string.IsNullOrEmpty(Backup(source, this));
#endregion
#region ORM
/// 查询数据库中的所有表名。
public string[] TableNames()
{
var list = new List();
if (Connect().IsEmpty())
{
var sql = "select name from sqlite_master where type='table' order by name; ";
var query = (Query)Query(sql);
for (int r = 0; r < query.Rows; r++)
{
var cell = query.Text(r, 0);
if (TextUtility.IsBlank(cell)) continue;
list.Add(cell);
}
query.Dispose();
}
return list.ToArray();
}
/// 查询数据库中的所有视图名。
public string[] ViewNames()
{
var list = new List();
if (Connect().IsEmpty())
{
var sql = "select name from sqlite_master where type='view' order by name; ";
var query = (Query)Query(sql);
for (int r = 0; r < query.Rows; r++)
{
var cell = query.Text(r, 0);
if (TextUtility.IsBlank(cell)) continue;
list.Add(cell);
}
query.Dispose();
}
return list.ToArray();
}
/// 查询表中的所有列名。
public string[] ColumnNames(string table)
{
var list = new List();
if (Connect().IsEmpty())
{
var t = TextUtility.AntiInject(table);
var sql = TextUtility.Merge("pragma table_info('", TextUtility.AntiInject(t), "'); ");
using (var query = Query(sql) as Query)
{
for (int r = 0; r < query.Rows; r++)
{
var cell = query.Text(r, "name");
if (TextUtility.IsBlank(cell)) continue;
list.Add(cell);
}
}
}
return list.ToArray();
}
/// 创建表,不修改已存在表。成功时返回空字符串,发生异常时返回异常信息。
public string Initialize(Record model) => model == null ? "参数无效。" : Initialize(model.GetType());
/// 创建表,不修改已存在表。成功时返回空字符串,发生异常时返回异常信息。
public string Initialize() where T : class, new() => Initialize(typeof(T));
/// 创建表,不修改已存在表。当现存表中缺少模型中属性对应的列时增加列。成功时返回空字符串,发生异常时返回异常信息。
public string Initialize(Type model)
{
var structure = TableStructure.Parse(model);
if (structure == null) return "无法解析记录模型。";
// 连接数据库。
var connect = Connect();
if (connect.NotEmpty()) return connect;
// 检查现存表。
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)
{
return TextUtility.Merge("指定的表已经存在。");
}
else
{
var sqlcolumns = new List();
foreach (var column in structure.Columns)
{
var type = Declaration(column);
if (string.IsNullOrEmpty(type))
{
return TextUtility.Merge("类型 ", column.Type.ToString(), " 不受支持。");
}
sqlcolumns.Add(type);
}
var sql = TextUtility.Merge("create table [", structure.Name, "](", TextUtility.Join(", ", sqlcolumns), "); ");
var execute = Execute(sql);
if (execute.Success) return TextUtility.Empty;
return execute.Message;
}
}
/// 插入记录。返回错误信息。
public string Insert(object record, string table = null)
{
if (record == null) return "参数无效。";
OrmHelper.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 ps = structure.CreateParameters(record, Parameter);
var psc = ps.Length;
if (psc < 1) return "数据模型不包含字段。";
// 合成 SQL 语句。
var sb = new StringBuilder();
sb.Append("insert into [");
sb.Append(table);
sb.Append("](");
for (var i = 0; i < psc; i++)
{
if (i > 0) sb.Append(", ");
sb.Append(ps[i].ParameterName);
}
sb.Append(") values(");
for (var i = 0; i < psc; i++)
{
if (i > 0) sb.Append(", ");
sb.Append("@");
sb.Append(ps[i].ParameterName);
}
sb.Append("); ");
var sql = sb.ToString();
// 执行。
var execute = Execute(sql, ps);
if (execute.Success && execute.Rows > 0) return TextUtility.Empty;
return execute.Message;
}
/// 更新记录,实体中的 Key 属性不被更新。返回错误信息。
/// 无法更新带有 Independent 特性的模型(缺少 Key 属性)。
public 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 ps = structure.CreateParameters(record, Parameter, "_key");
var psc = ps.Length;
if (psc < 1) return "数据模型不包含字段。";
// 合成 SQL 语句。
var sb = new StringBuilder();
sb.Append("update [");
sb.Append(table);
sb.Append("] set ");
for (var i = 0; i < psc; i++)
{
if (i > 0) sb.Append(", ");
sb.Append("[");
sb.Append(ps[i].ParameterName);
sb.Append("] = @");
sb.Append(ps[i].ParameterName);
}
sb.Append(" where [_key] = '");
sb.Append(record.Key.SafeKey());
sb.Append("'; ");
var sql = sb.ToString();
// 执行。
var execute = Execute(sql, ps);
if (execute.Success && execute.Rows > 0) return TextUtility.Empty;
return execute.Message;
}
/// 获取按指定语句查询到的所有记录。
public Result