344 lines
12 KiB
C#
344 lines
12 KiB
C#
using Newtonsoft.Json.Linq;
|
||
using System.Data;
|
||
using System.Data.SQLite;
|
||
using static NPOI.HSSF.Util.HSSFColor;
|
||
|
||
namespace Ramitta
|
||
{
|
||
public class SQLite : IDisposable
|
||
{
|
||
private SQLiteConnection db;
|
||
private bool disposed = false;
|
||
|
||
// 构造函数，初始化数据库连接
|
||
public SQLite(string file,bool ReadOnly=false)
|
||
{
|
||
var sql = $"Data Source={file};Version=3;{(ReadOnly ? "Read Only=True;" : "")}";
|
||
db = new SQLiteConnection(sql);
|
||
db.Open();
|
||
}
|
||
|
||
|
||
// 创建表：根据表名和字段定义创建表
|
||
// 参数: tableName - 表名
|
||
// 参数: columns - 字段定义字典，键为字段名，值为字段类型
|
||
// 例如: CreateTable("Users", new Dictionary<string, string> { {"Id", "INTEGER"}, {"Name", "TEXT"} });
|
||
public void CreateTable(string tableName, Dictionary<string, string> columns)
|
||
{
|
||
// 构建列定义的字符串
|
||
var columnsDefinition = string.Join(", ", columns.Select(c => $"{c.Key} {c.Value}"));
|
||
string createTableQuery = $"CREATE TABLE IF NOT EXISTS {tableName} ({columnsDefinition});";
|
||
|
||
using (var cmd = new SQLiteCommand(createTableQuery, db))
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
}
|
||
// 获取数据库中所有表名
|
||
// 返回: 包含所有表名的字符串列表
|
||
public List<string> GetAllTableNames()
|
||
{
|
||
List<string> tableNames = new List<string>();
|
||
|
||
string query = "SELECT name FROM sqlite_master WHERE type='table';";
|
||
|
||
using (var cmd = new SQLiteCommand(query, db))
|
||
using (var reader = cmd.ExecuteReader())
|
||
{
|
||
while (reader.Read())
|
||
{
|
||
string tableName = reader.GetString(0);
|
||
tableNames.Add(tableName);
|
||
}
|
||
}
|
||
|
||
return tableNames;
|
||
}
|
||
// 向已存在的表中添加新列
|
||
// 参数: tableName - 表名
|
||
// 参数: columnName - 要添加的列名
|
||
// 参数: columnType - 列的数据类型
|
||
// 例如: AddColumn("Users", "Email", "TEXT");
|
||
public void AddColumn(string tableName, string columnName, string columnType)
|
||
{
|
||
// 检查表是否存在
|
||
if (!TableExists(tableName))
|
||
{
|
||
throw new ArgumentException($"表 '{tableName}' 不存在");
|
||
}
|
||
|
||
// 检查列是否已存在
|
||
if (ColumnExists(tableName, columnName))
|
||
{
|
||
Console.WriteLine($"列 '{columnName}' 在表 '{tableName}' 中已存在");
|
||
return;
|
||
}
|
||
|
||
// 构建添加列的SQL语句
|
||
string addColumnQuery = $"ALTER TABLE {tableName} ADD COLUMN {columnName} {columnType};";
|
||
|
||
using (var cmd = new SQLiteCommand(addColumnQuery, db))
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
Console.WriteLine($"已向表 '{tableName}' 添加列 '{columnName}'");
|
||
}
|
||
|
||
// 检查表是否存在
|
||
private bool TableExists(string tableName)
|
||
{
|
||
string query = "SELECT count(*) FROM sqlite_master WHERE type='table' AND name=@tableName;";
|
||
|
||
using (var cmd = new SQLiteCommand(query, db))
|
||
{
|
||
cmd.Parameters.AddWithValue("@tableName", tableName);
|
||
var result = cmd.ExecuteScalar();
|
||
return Convert.ToInt32(result) > 0;
|
||
}
|
||
}
|
||
|
||
// 检查列是否已存在
|
||
private bool ColumnExists(string tableName, string columnName)
|
||
{
|
||
string query = $"PRAGMA table_info({tableName});";
|
||
|
||
using (var cmd = new SQLiteCommand(query, db))
|
||
using (var reader = cmd.ExecuteReader())
|
||
{
|
||
while (reader.Read())
|
||
{
|
||
if (reader["name"].ToString().Equals(columnName, StringComparison.OrdinalIgnoreCase))
|
||
{
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// 批量添加多个列
|
||
// 参数: tableName - 表名
|
||
// 参数: columns - 字段定义字典，键为字段名，值为字段类型
|
||
// 例如: AddColumns("Users", new Dictionary<string, string> { {"Email", "TEXT"}, {"Age", "INTEGER"} });
|
||
public void AddColumns(string tableName, Dictionary<string, string> columns)
|
||
{
|
||
foreach (var column in columns)
|
||
{
|
||
AddColumn(tableName, column.Key, column.Value);
|
||
}
|
||
}
|
||
|
||
// 插入数据：向指定表插入一条记录
|
||
// 参数: tableName - 表名
|
||
// 参数: columnValues - 字段和对应值的字典
|
||
// 例如: InsertData("Users", new Dictionary<string, object> { {"Name", "John"}, {"Age", 30} });
|
||
public void InsertData(string tableName, Dictionary<string, object> columnValues)
|
||
{
|
||
// 构建列和参数的字符串
|
||
var columns = string.Join(", ", columnValues.Keys);
|
||
var parameters = string.Join(", ", columnValues.Keys.Select(k => "@" + k));
|
||
|
||
string insertQuery = $"INSERT INTO {tableName} ({columns}) VALUES ({parameters})";
|
||
|
||
using (var cmd = new SQLiteCommand(insertQuery, db))
|
||
{
|
||
foreach (var kvp in columnValues)
|
||
{
|
||
cmd.Parameters.AddWithValue("@" + kvp.Key, kvp.Value);
|
||
}
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
}
|
||
|
||
// 查询数据：执行任意查询语句并返回结果
|
||
// 参数: query - SQL查询语句
|
||
// 参数: parameters - 可选的查询参数字典
|
||
// 例如: SelectData("SELECT * FROM Users WHERE Age > @age", new Dictionary<string, object> { {"age", 25} });
|
||
public List<Dictionary<string, object>> SelectData(string query, Dictionary<string, object> parameters = null)
|
||
{
|
||
var result = new List<Dictionary<string, object>>();
|
||
|
||
using (var cmd = new SQLiteCommand(query, db))
|
||
{
|
||
// 添加查询参数（如果有的话）
|
||
if (parameters != null)
|
||
{
|
||
foreach (var kvp in parameters)
|
||
{
|
||
cmd.Parameters.AddWithValue("@" + kvp.Key, kvp.Value);
|
||
}
|
||
}
|
||
|
||
using (SQLiteDataReader reader = cmd.ExecuteReader())
|
||
{
|
||
while (reader.Read())
|
||
{
|
||
var row = new Dictionary<string, object>();
|
||
for (int i = 0; i < reader.FieldCount; i++)
|
||
{
|
||
row[reader.GetName(i)] = reader.GetValue(i);
|
||
}
|
||
result.Add(row);
|
||
}
|
||
}
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
// 更新数据：根据条件更新指定表中的记录
|
||
// 参数: tableName - 表名
|
||
// 参数: columnValues - 需要更新的字段和对应值的字典
|
||
// 参数: condition - 更新条件
|
||
// 例如: UpdateData("Users", new Dictionary<string, object> { {"Age", 31} }, "Name = 'John'");
|
||
public void UpdateData(string tableName, Dictionary<string, object> columnValues, string condition)
|
||
{
|
||
// 构建SET子句
|
||
var setClause = string.Join(", ", columnValues.Keys.Select(k => $"{k} = @{k}"));
|
||
string updateQuery = $"UPDATE {tableName} SET {setClause} WHERE {condition}";
|
||
|
||
using (var cmd = new SQLiteCommand(updateQuery, db))
|
||
{
|
||
foreach (var kvp in columnValues)
|
||
{
|
||
cmd.Parameters.AddWithValue("@" + kvp.Key, kvp.Value);
|
||
}
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
}
|
||
|
||
// 删除数据：根据条件删除指定表中的记录
|
||
// 参数: tableName - 表名
|
||
// 参数: condition - 删除条件
|
||
// 例如: DeleteData("Users", "Age < 18");
|
||
public void DeleteData(string tableName, string condition)
|
||
{
|
||
string deleteQuery = $"DELETE FROM {tableName} WHERE {condition}";
|
||
|
||
using (var cmd = new SQLiteCommand(deleteQuery, db))
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
}
|
||
|
||
// 支持事务操作：允许在同一个事务中执行多个操作
|
||
// 参数: transactionActions - 一个包含多个数据库操作的委托
|
||
// 例如: ExecuteTransaction(() =>
|
||
// {
|
||
// InsertData("Users", new Dictionary<string, object> { {"Name", "Alice"}, {"Age", 28} });
|
||
// UpdateData("Users", new Dictionary<string, object> { {"Age", 29} }, "Name = 'Bob'");
|
||
// });
|
||
public void ExecuteTransaction(Action transactionActions)
|
||
{
|
||
using (var transaction = db.BeginTransaction())
|
||
{
|
||
try
|
||
{
|
||
transactionActions.Invoke(); // 执行多个操作
|
||
transaction.Commit(); // 提交事务
|
||
}
|
||
catch (Exception)
|
||
{
|
||
transaction.Rollback(); // 回滚事务
|
||
throw;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 删除所有表
|
||
public void DropAllTables()
|
||
{
|
||
// 获取所有表名
|
||
string getTablesQuery = "SELECT name FROM sqlite_master WHERE type='table';";
|
||
var tables = SelectData(getTablesQuery);
|
||
|
||
foreach (var table in tables)
|
||
{
|
||
string tableName = table["name"].ToString();
|
||
string dropTableQuery = $"DROP TABLE IF EXISTS {tableName};";
|
||
|
||
using (var cmd = new SQLiteCommand(dropTableQuery, db))
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
}
|
||
}
|
||
|
||
public int ExecuteBatchInsert(string tableName, List<string> columns, string columnsStr, List<Dictionary<string, object>> batch, int batchIndex)
|
||
{
|
||
var valueParams = new List<string>();
|
||
var parameters = new Dictionary<string, object>();
|
||
|
||
for (int i = 0; i < batch.Count; i++)
|
||
{
|
||
var paramNames = columns.Select(col => $"@p{batchIndex}_{i}_{col}").ToList();
|
||
valueParams.Add($"({string.Join(", ", paramNames)})");
|
||
|
||
foreach (var col in columns)
|
||
{
|
||
parameters[$"p{batchIndex}_{i}_{col}"] = batch[i][col] ?? DBNull.Value;
|
||
}
|
||
}
|
||
|
||
string insertQuery = $"INSERT INTO {tableName} ({columnsStr}) VALUES {string.Join(", ", valueParams)}";
|
||
|
||
using (var transaction = db.BeginTransaction())
|
||
using (var cmd = new SQLiteCommand(insertQuery, db, transaction))
|
||
{
|
||
foreach (var param in parameters)
|
||
{
|
||
cmd.Parameters.AddWithValue(param.Key, param.Value);
|
||
}
|
||
|
||
int result = cmd.ExecuteNonQuery();
|
||
transaction.Commit();
|
||
return result;
|
||
}
|
||
}
|
||
|
||
// 修改主方法
|
||
public int BulkInsert(string tableName, List<Dictionary<string, object>> dataList)
|
||
{
|
||
if (dataList == null || dataList.Count == 0)
|
||
return 0;
|
||
|
||
int insertedCount = 0;
|
||
var columns = dataList[0].Keys.ToList();
|
||
var columnsStr = string.Join(", ", columns);
|
||
int batchSize = 500;
|
||
int batchIndex = 0; // 添加批次索引
|
||
|
||
for (int i = 0; i < dataList.Count; i += batchSize)
|
||
{
|
||
var batch = dataList.Skip(i).Take(batchSize).ToList();
|
||
insertedCount += ExecuteBatchInsert(tableName, columns, columnsStr, batch, batchIndex);
|
||
batchIndex++; // 递增批次索引
|
||
}
|
||
|
||
return insertedCount;
|
||
}
|
||
|
||
// 释放资源，关闭数据库连接
|
||
// 确保数据库连接在对象销毁时被正确关闭
|
||
public void Dispose()
|
||
{
|
||
if (!disposed)
|
||
{
|
||
if (db != null && db.State == ConnectionState.Open)
|
||
{
|
||
db.Close();
|
||
db.Dispose();
|
||
}
|
||
disposed = true;
|
||
}
|
||
GC.SuppressFinalize(this);
|
||
}
|
||
|
||
// 析构函数，调用Dispose释放资源
|
||
~SQLite()
|
||
{
|
||
Dispose();
|
||
}
|
||
}
|
||
}
|