SqlHelper 实现类,支持多数据库,提供异步操作、自动重试、事务、存储过程、分页、缓存等功能。

cs 复制代码
/// <summary>
/// SqlHelper 实现类,支持多数据库,提供异步操作、自动重试、事务、存储过程、分页、缓存等功能。
/// </summary>
public class SqlHelper : IDbHelper
{
    private readonly IDbConnectionFactory _connectionFactory;
    private readonly ILogger<SqlHelper>? _logger;
    private readonly AsyncPolicy _retryPolicy;
    private readonly IRedisCacheHelper? _redisCacheHelper;

    /// <summary>
    /// 构造函数,注入数据库连接工厂,日志记录器,自动重试次数与延迟,以及可选 Redis 缓存。
    /// </summary>
    /// <param name="connectionFactory">数据库连接工厂</param>
    /// <param name="logger">日志记录器</param>
    ///<param name="retryCount">自动重试次数</param> 
    /// <param name="retryDelayMs">重试延迟(毫秒)</param> 
    /// <param name="redisCacheHelper">Redis 缓存(可空)</param> 
    public SqlHelper( IDbConnectionFactory connectionFactory, ILogger<SqlHelper>? logger = null, int retryCount = 3, int retryDelayMs = 200, IRedisCacheHelper? redisCacheHelper = null) 
    { 
        _connectionFactory = connectionFactory; 
        _logger = logger;
        _redisCacheHelper = redisCacheHelper;
        _retryPolicy = Policy.Handle<Exception>()
        .WaitAndRetryAsync(retryCount, _ => TimeSpan.FromMilliseconds(retryDelayMs),
        (ex, timespan, retryCount, context) =>
        {
            _logger?.LogWarning($"Retry {retryCount} in {timespan.TotalMilliseconds}ms due to: {ex.Message}");
        });
    }

    /// <summary>
    /// 获取数据库连接实例
    /// </summary>
    /// <returns>DbConnection 实例</returns>
    private DbConnection GetConnection() => _connectionFactory.CreateConnection();

    public async Task<IEnumerable<T>> QueryAsync<T>(string sql, object? parameters = null)
    {
        var sw = Stopwatch.StartNew();
        await using var conn = GetConnection();
        await conn.OpenAsync();

        var result = await _retryPolicy.ExecuteAsync(() => conn.QueryAsync<T>(sql, parameters));
        sw.Stop();

        _logger?.LogInformation($"[Query] {sw.ElapsedMilliseconds}ms - SQL: {sql}");
        return result;
    }

    public async Task<T> QueryFirstOrDefaultAsync<T>(string sql, object? parameters = null)
    {
        var sw = Stopwatch.StartNew();
        await using var conn = GetConnection();
        await conn.OpenAsync();

        var result = await _retryPolicy.ExecuteAsync(() => conn.QueryFirstOrDefaultAsync<T>(sql, parameters));
        sw.Stop();

        _logger?.LogInformation($"[QueryFirstOrDefault] {sw.ElapsedMilliseconds}ms - SQL: {sql}");
        return result;
    }

    public async Task<int> ExecuteAsync(string sql, object? parameters = null)
    {
        var sw = Stopwatch.StartNew();
        await using var conn = GetConnection();
        await conn.OpenAsync();

        var result = await _retryPolicy.ExecuteAsync(() => conn.ExecuteAsync(sql, parameters));
        sw.Stop();

        _logger?.LogInformation($"[Execute] {sw.ElapsedMilliseconds}ms - SQL: {sql}");
        return result;
    }

    public async Task<IEnumerable<T>> RunProcedureAsync<T>(string procName, object? parameters = null)
    {
        var sw = Stopwatch.StartNew();
        await using var conn = GetConnection();
        await conn.OpenAsync();

        var result = await _retryPolicy.ExecuteAsync(() => conn.QueryAsync<T>(procName, parameters, commandType: CommandType.StoredProcedure));
        sw.Stop();

        _logger?.LogInformation($"[RunProcedure] {sw.ElapsedMilliseconds}ms - Procedure: {procName}");
        return result;
    }

    public async Task ExecuteInTransactionAsync(Func<IDbHelper, Task> transactionalOperations)
    {
        await using var conn = GetConnection();
        await conn.OpenAsync();
        await using var tran = await conn.BeginTransactionAsync();

        try
        {
            var tranHelper = new SqlHelperWithTransaction(conn, tran, _connectionFactory.DbProvider, _logger, _retryPolicy, _redisCacheHelper);
            await transactionalOperations(tranHelper);
            await tran.CommitAsync();
        }
        catch (Exception ex)
        {
            await tran.RollbackAsync();
            _logger?.LogError(ex, "Transaction failed.");
            throw;
        }
    }

    public Task<string> BuildPagedQuery(string baseSql, string orderBy, int pageIndex, int pageSize)
    {
        int offset = (pageIndex - 1) * pageSize;
        string pagedSql = _connectionFactory.DbProvider.ToLower() switch
        {
            "mysql" => $"{baseSql} ORDER BY {orderBy} LIMIT {offset}, {pageSize}",
            "postgresql" => $"{baseSql} ORDER BY {orderBy} OFFSET {offset} LIMIT {pageSize}",
            _ => $"{baseSql} ORDER BY {orderBy} OFFSET {offset} ROWS FETCH NEXT {pageSize} ROWS ONLY"
        };
        return Task.FromResult(pagedSql);
    }

    public async Task<IEnumerable<T>> QueryCachedAsync<T>(string cacheKey, string sql, object? parameters = null, TimeSpan? expiry = null)
    {
        if (_redisCacheHelper == null)
            throw new InvalidOperationException("Redis cache helper not configured.");

        return await _redisCacheHelper.GetOrSetAsync(cacheKey, async () =>
        {
            await using var conn = GetConnection();
            await conn.OpenAsync();
            return await conn.QueryAsync<T>(sql, parameters);
        }, expiry);
    }

}

完整源码:【免费】.Net7封装操作数据库,支持MySql、MSSQL、postgresql多数据库操作资源-CSDN文库

相关推荐
Bioinfo Guy1 小时前
高分Panel复现系列|转录因子网络太乱?用模块框把重点圈出来
数据库·生物信息学·生信可视化
ltl3 小时前
Disaggregated DB 合集:Socrates、PolarDB、Taurus 的共同模式
数据库
椒颜皮皮虾྅4 小时前
OpenVINO C# API 3.3.1:支持OpenVINO 2026.3及GenAI增强
人工智能·c#·openvino
Blossom i5 小时前
大数据预处理与采集实验一:使用Python操作MySQL数据库
数据库·mysql
雾时之林6 小时前
Linux--软件管理、源码包安装
linux·服务器·数据库
半亩码田7 小时前
C#转Python第3.1篇:Python 的 class 没有访问修饰符?面向对象的另一条路
开发语言·python·c#
老纪的技术唠嗑局7 小时前
超级干货分享:中小企业使用 OceanBase 实践经验汇总
数据库
CLOUD ACE7 小时前
谷歌云代理|零售商Target 如何利用 Spanner Graph 提升零售发现体验并将数据库维护成本降低 50%
数据库·零售
AI大模型-小雄8 小时前
Codex写分页接口为什么越翻越慢?用Cursor Pagination解决重复与漏数据
大数据·数据库·elasticsearch·搜索引擎·chatgpt·后端开发·codex
少晓年8 小时前
从 MySQL 迁移到人大金仓 KingbaseES:完整指南与实践
数据库·mysql