7 changed files with 1847 additions and 1038 deletions
@ -0,0 +1,360 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Data; |
|||
using System.Linq; |
|||
using System.Text; |
|||
using System.Threading; |
|||
using System.Threading.Tasks; |
|||
|
|||
namespace SunlightCentralizedControlManagement_SCCM_.UserClass |
|||
{ |
|||
/// <summary>
|
|||
/// 线程安全的 DataTable 管理器
|
|||
/// 提供对 DataTable 的安全并发访问
|
|||
/// </summary>
|
|||
public class DataTableManager : IDisposable |
|||
{ |
|||
private DataTable _dataTable; |
|||
private readonly ReaderWriterLockSlim _lock = new ReaderWriterLockSlim(); |
|||
private bool _disposed = false; |
|||
|
|||
#region 构造函数
|
|||
public DataTableManager() |
|||
{ |
|||
_dataTable = new DataTable(); |
|||
} |
|||
|
|||
public DataTableManager(DataTable dataTable) |
|||
{ |
|||
_dataTable = dataTable ?? throw new ArgumentNullException(nameof(dataTable)); |
|||
} |
|||
|
|||
public DataTableManager(string tableName) |
|||
{ |
|||
_dataTable = new DataTable(tableName); |
|||
} |
|||
#endregion
|
|||
|
|||
#region 基本操作
|
|||
/// <summary>
|
|||
/// 安全地添加列
|
|||
/// </summary>
|
|||
public void AddColumn(string columnName, Type dataType) |
|||
{ |
|||
ExecuteWriteOperation(table => |
|||
{ |
|||
table.Columns.Add(columnName, dataType); |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地添加行
|
|||
/// </summary>
|
|||
public void AddRow(params object[] values) |
|||
{ |
|||
ExecuteWriteOperation(table => |
|||
{ |
|||
DataRow newRow = table.NewRow(); |
|||
for (int i = 0; i < Math.Min(values.Length, table.Columns.Count); i++) |
|||
{ |
|||
newRow[i] = values[i] ?? DBNull.Value; |
|||
} |
|||
table.Rows.Add(newRow); |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地更新单元格值
|
|||
/// </summary>
|
|||
public bool UpdateCell(int rowIndex, string columnName, object value) |
|||
{ |
|||
return ExecuteWriteOperation(table => |
|||
{ |
|||
if (IsValidRowIndex(table, rowIndex) && table.Columns.Contains(columnName)) |
|||
{ |
|||
table.Rows[rowIndex][columnName] = value ?? DBNull.Value; |
|||
return true; |
|||
} |
|||
return false; |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地删除行
|
|||
/// </summary>
|
|||
public bool DeleteRow(int rowIndex) |
|||
{ |
|||
return ExecuteWriteOperation(table => |
|||
{ |
|||
if (IsValidRowIndex(table, rowIndex)) |
|||
{ |
|||
table.Rows[rowIndex].Delete(); |
|||
return true; |
|||
} |
|||
return false; |
|||
}); |
|||
} |
|||
#endregion
|
|||
|
|||
#region 查询操作
|
|||
/// <summary>
|
|||
/// 安全地获取行
|
|||
/// </summary>
|
|||
public DataRow GetRow(int rowIndex) |
|||
{ |
|||
return ExecuteReadOperation(table => |
|||
{ |
|||
return IsValidRowIndex(table, rowIndex) ? table.Rows[rowIndex] : null; |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地查找行（使用筛选条件）
|
|||
/// </summary>
|
|||
public DataRow[] FindRows(string filterExpression) |
|||
{ |
|||
return ExecuteReadOperation(table => |
|||
{ |
|||
try |
|||
{ |
|||
return string.IsNullOrEmpty(filterExpression) |
|||
? table.Select() |
|||
: table.Select(filterExpression); |
|||
} |
|||
catch (EvaluateException) |
|||
{ |
|||
return new DataRow[0]; |
|||
} |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地获取行数
|
|||
/// </summary>
|
|||
public int GetRowCount() |
|||
{ |
|||
return ExecuteReadOperation(table => table.Rows.Count); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地检查行是否存在
|
|||
/// </summary>
|
|||
public bool RowExists(int rowIndex) |
|||
{ |
|||
return ExecuteReadOperation(table => IsValidRowIndex(table, rowIndex)); |
|||
} |
|||
#endregion
|
|||
|
|||
#region 批量操作
|
|||
/// <summary>
|
|||
/// 安全地批量添加行
|
|||
/// </summary>
|
|||
public void BulkAddRows(IEnumerable<object[]> rowsData) |
|||
{ |
|||
ExecuteWriteOperation(table => |
|||
{ |
|||
foreach (var rowData in rowsData) |
|||
{ |
|||
DataRow newRow = table.NewRow(); |
|||
for (int i = 0; i < Math.Min(rowData.Length, table.Columns.Count); i++) |
|||
{ |
|||
newRow[i] = rowData[i] ?? DBNull.Value; |
|||
} |
|||
table.Rows.Add(newRow); |
|||
} |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 安全地清空所有数据
|
|||
/// </summary>
|
|||
public void Clear() |
|||
{ |
|||
ExecuteWriteOperation(table => |
|||
{ |
|||
table.Clear(); |
|||
}); |
|||
} |
|||
#endregion
|
|||
|
|||
#region 数据导出
|
|||
/// <summary>
|
|||
/// 获取数据快照（线程安全的副本）
|
|||
/// </summary>
|
|||
public DataTable GetSnapshot() |
|||
{ |
|||
return ExecuteReadOperation(table => |
|||
{ |
|||
return table.Copy(); // 创建数据的独立副本
|
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 获取只读视图
|
|||
/// </summary>
|
|||
public DataView GetReadOnlyView() |
|||
{ |
|||
return ExecuteReadOperation(table => |
|||
{ |
|||
return new DataView(table) { AllowEdit = false }; |
|||
}); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 导出到新 DataTable（包含结构）
|
|||
/// </summary>
|
|||
public DataTable ExportToDataTable() |
|||
{ |
|||
return ExecuteReadOperation(table => |
|||
{ |
|||
DataTable exported = table.Clone(); |
|||
exported.Merge(table, false, MissingSchemaAction.Add); |
|||
return exported; |
|||
}); |
|||
} |
|||
#endregion
|
|||
|
|||
#region 锁操作核心
|
|||
/// <summary>
|
|||
/// 执行读取操作（共享锁）
|
|||
/// </summary>
|
|||
private T ExecuteReadOperation<T>(Func<DataTable, T> operation) |
|||
{ |
|||
_lock.EnterReadLock(); |
|||
try |
|||
{ |
|||
return operation(_dataTable); |
|||
} |
|||
finally |
|||
{ |
|||
_lock.ExitReadLock(); |
|||
} |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 执行写入操作（独占锁）
|
|||
/// </summary>
|
|||
private void ExecuteWriteOperation(Action<DataTable> operation) |
|||
{ |
|||
_lock.EnterWriteLock(); |
|||
try |
|||
{ |
|||
operation(_dataTable); |
|||
_dataTable.AcceptChanges(); |
|||
} |
|||
finally |
|||
{ |
|||
_lock.ExitWriteLock(); |
|||
} |
|||
} |
|||
|
|||
private T ExecuteWriteOperation<T>(Func<DataTable, T> operation) |
|||
{ |
|||
_lock.EnterWriteLock(); |
|||
try |
|||
{ |
|||
var result = operation(_dataTable); |
|||
_dataTable.AcceptChanges(); |
|||
return result; |
|||
} |
|||
finally |
|||
{ |
|||
_lock.ExitWriteLock(); |
|||
} |
|||
} |
|||
|
|||
private bool IsValidRowIndex(DataTable table, int rowIndex) |
|||
{ |
|||
return rowIndex >= 0 && rowIndex < table.Rows.Count; |
|||
} |
|||
#endregion
|
|||
|
|||
#region 高级功能
|
|||
/// <summary>
|
|||
/// 带超时的安全操作
|
|||
/// </summary>
|
|||
public bool TryUpdateCell(int rowIndex, string columnName, object value, int timeoutMs = 5000) |
|||
{ |
|||
if (_lock.TryEnterWriteLock(timeoutMs)) |
|||
{ |
|||
try |
|||
{ |
|||
if (IsValidRowIndex(_dataTable, rowIndex) && _dataTable.Columns.Contains(columnName)) |
|||
{ |
|||
_dataTable.Rows[rowIndex][columnName] = value ?? DBNull.Value; |
|||
_dataTable.AcceptChanges(); |
|||
return true; |
|||
} |
|||
return false; |
|||
} |
|||
finally |
|||
{ |
|||
_lock.ExitWriteLock(); |
|||
} |
|||
} |
|||
return false; |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// 执行事务性操作
|
|||
/// </summary>
|
|||
public void ExecuteTransaction(Action<DataTable> transaction) |
|||
{ |
|||
_lock.EnterWriteLock(); |
|||
try |
|||
{ |
|||
// 开始事务
|
|||
_dataTable.BeginLoadData(); |
|||
|
|||
try |
|||
{ |
|||
transaction(_dataTable); |
|||
_dataTable.AcceptChanges(); |
|||
} |
|||
catch |
|||
{ |
|||
_dataTable.RejectChanges(); |
|||
throw; |
|||
} |
|||
finally |
|||
{ |
|||
_dataTable.EndLoadData(); |
|||
} |
|||
} |
|||
finally |
|||
{ |
|||
_lock.ExitWriteLock(); |
|||
} |
|||
} |
|||
#endregion
|
|||
|
|||
#region 属性和清理
|
|||
public string TableName |
|||
{ |
|||
get => ExecuteReadOperation(table => table.TableName); |
|||
set => ExecuteWriteOperation(table => table.TableName = value); |
|||
} |
|||
|
|||
public DataColumnCollection Columns |
|||
{ |
|||
get => ExecuteReadOperation(table => table.Columns); |
|||
} |
|||
|
|||
public void Dispose() |
|||
{ |
|||
if (!_disposed) |
|||
{ |
|||
_lock?.Dispose(); |
|||
_dataTable?.Dispose(); |
|||
_disposed = true; |
|||
} |
|||
GC.SuppressFinalize(this); |
|||
} |
|||
|
|||
~DataTableManager() |
|||
{ |
|||
Dispose(); |
|||
} |
|||
#endregion
|
|||
} |
|||
} |
|||
Loading…
Reference in new issue