DMS/DMS.Infrastructure/Repositories/VarDataRepository.cs

using System.Collections.Generic;
using System.Diagnostics;
using System.Threading.Tasks;
using DMS.Extensions;
using SqlSugar;
using AutoMapper;
using Dm.util;
using DMS.Infrastructure.Entities;
using DMS.Helper;
using DMS.Models;

namespace DMS.Infrastructure.Repositories;

/// <summary>
/// VariableData仓储类，用于操作DbVariableData实体
/// </summary>
public class VarDataRepository
{
    private readonly IMapper _mapper;

    public VarDataRepository(IMapper mapper)
    {
        _mapper = mapper;
    }

    /// <summary>
    /// 根据ID获取VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <returns></returns>
    public async Task<DbVariable> GetByIdAsync(int id)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await GetByIdAsync(id, db);
            stopwatch.Stop();
            NlogHelper.Info($"根据ID '{id}' 获取VariableData耗时：{stopwatch.ElapsedMilliseconds}ms");
            return result;
        }
    }

    /// <summary>
    /// 根据ID获取VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<DbVariable> GetByIdAsync(int id, SqlSugarClient db)
    {
        return await db.Queryable<DbVariable>()
                       .In(id)
                       .SingleAsync();
    }

    /// <summary>
    /// 获取所有VariableData
    /// </summary>
    /// <returns></returns>
    public async Task<List<Variable>> GetAllAsync()
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await GetAllAsync(db);
            stopwatch.Stop();
            NlogHelper.Info($"获取所有VariableData耗时：{stopwatch.ElapsedMilliseconds}ms");
            return result;
        }
    }

    /// <summary>
    /// 获取所有VariableData
    /// </summary>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<List<Variable>> GetAllAsync(SqlSugarClient db)
    {
        var result = await db.Queryable<DbVariable>()
                             .Includes(d => d.VariableTable)
                             .Includes(d => d.VariableTable.Device)
                             .ToListAsync();
        return result.Select(d => _mapper.Map<Variable>(d))
                     .ToList();
    }

    /// <summary>
    /// 获取所有VariableData
    /// </summary>
    /// <returns></returns>
    public async Task<List<Variable>> GetByVariableTableIdAsync(int varTableId)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await GetByVariableTableIdAsync(varTableId, db);
            stopwatch.Stop();
            NlogHelper.Info($"获取变量表的所有变量{result.Count()}个耗时：{stopwatch.ElapsedMilliseconds}ms");
            return result;
        }
    }

    /// <summary>
    /// 获取所有VariableData
    /// </summary>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<List<Variable>> GetByVariableTableIdAsync(int varTableId, SqlSugarClient db)
    {
        var result = await db.Queryable<DbVariable>()
                             .Where(d => d.VariableTableId == varTableId)
                             .ToListAsync();
        return result.Select(d => _mapper.Map<Variable>(d))
                     .ToList();
    }

    /// <summary>
    /// 新增VariableData
    /// </summary>
    /// <param name="variable">VariableData实体</param>
    /// <returns></returns>
    public async Task<Variable> AddAsync(Variable variable)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var varData = await AddAsync(variable, db);
            stopwatch.Stop();
            NlogHelper.Info($"新增VariableData '{variable.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
            return varData;
        }
    }


    /// <summary>
    /// 新增VariableData
    /// </summary>
    /// <param name="variable">VariableData实体</param>
    /// <returns></returns>
    public async Task<Variable> AddAsync(Variable variable, SqlSugarClient db)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        var dbVarData = await db.Insertable(_mapper.Map<DbVariable>(variable))
                                .ExecuteReturnEntityAsync();
        stopwatch.Stop();
        NlogHelper.Info($"新增VariableData '{variable.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return _mapper.Map<Variable>(dbVarData);
    }

    /// <summary>
    /// 新增VariableData
    /// </summary>
    /// <param name="variableData">VariableData实体</param>
    /// <returns></returns>
    public async Task<int> AddAsync(IEnumerable<Variable> variableDatas)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var varData = await AddAsync(variableDatas, db);
            stopwatch.Stop();
            NlogHelper.Info($"新增VariableData '{variableDatas.Count()}'个， 耗时：{stopwatch.ElapsedMilliseconds}ms");
            return varData;
        }
    }


    /// <summary>
    /// 新增VariableData
    /// </summary>
    /// <param name="variableData">VariableData实体</param>
    /// <returns></returns>
    public async Task<int> AddAsync(IEnumerable<Variable> variableDatas, SqlSugarClient db)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        Stopwatch stopwatch2 = new Stopwatch();
        stopwatch2.Start();
        var dbList = variableDatas.Select(vb => _mapper.Map<DbVariable>(vb))
                                  .ToList();
        stopwatch2.Stop();
        NlogHelper.Info($"复制 Variable'{variableDatas.Count()}'个， 耗时：{stopwatch2.ElapsedMilliseconds}ms");

        var res = await db.Insertable<DbVariable>(dbList)
                          .ExecuteCommandAsync();

        stopwatch.Stop();
        NlogHelper.Info($"新增VariableData '{variableDatas.Count()}'个， 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return res;
    }


    /// <summary>
    /// 更新VariableData
    /// </summary>
    /// <param name="variable">VariableData实体</param>
    /// <returns></returns>
    public async Task<int> UpdateAsync(Variable variable)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using (var db = DbContext.GetInstance())
        {
            var result = await UpdateAsync(variable, db);
            stopwatch.Stop();
            NlogHelper.Info($"更新VariableData '{variable.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
            return result;
        }
    }

    /// <summary>
    /// 更新VariableData
    /// </summary>
    /// <param name="variable">VariableData实体</param>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<int> UpdateAsync(Variable variable, SqlSugarClient db)
    {
        var result = await db.Updateable<DbVariable>(_mapper.Map<DbVariable>(variable))
                             .ExecuteCommandAsync();
        return result;
    }

    /// <summary>
    /// 更新VariableData
    /// </summary>
    /// <param name="variableData">VariableData实体</param>
    /// <returns></returns>
    public async Task<int> UpdateAsync(List<Variable> variableDatas)
    {
        using var _db = DbContext.GetInstance();
        return await UpdateAsync(variableDatas, _db);
    }

    /// <summary>
    /// 更新VariableData
    /// </summary>
    /// <param name="variableData">VariableData实体</param>
    /// <returns></returns>
    public async Task<int> UpdateAsync(List<Variable> variableDatas, SqlSugarClient db)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();

        var dbVarDatas = variableDatas.Select(vd => _mapper.Map<DbVariable>(vd));
        var result = await db.Updateable<DbVariable>(dbVarDatas.ToList())
                             .ExecuteCommandAsync();

        stopwatch.Stop();
        NlogHelper.Info($"更新VariableData  {variableDatas.Count()}个 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return result;
    }

    /// <summary>
    /// 删除VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <returns></returns>
    public async Task<int> DeleteAsync(Variable variable)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using var db = DbContext.GetInstance();
        var result = await DeleteAsync(variable, db);
        stopwatch.Stop();
        NlogHelper.Info($"删除VariableData: '{variable.Name}' 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return result;
    }

    /// <summary>
    /// 根据ID删除VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<int> DeleteAsync(Variable variable, SqlSugarClient db)
    {
        var result = await db.Deleteable<DbVariable>()
                             .Where(d => d.Id == variable.Id)
                             .ExecuteCommandAsync();
        return result;
    }

    /// <summary>
    /// 删除VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <returns></returns>
    public async Task<int> DeleteAsync(IEnumerable<Variable> variableDatas)
    {
        Stopwatch stopwatch = new Stopwatch();
        stopwatch.Start();
        using var db = DbContext.GetInstance();

        var result = await DeleteAsync(variableDatas, db);

        stopwatch.Stop();
        NlogHelper.Info($"删除VariableData: '{variableDatas.Count()}'个 耗时：{stopwatch.ElapsedMilliseconds}ms");
        return result;
    }

    /// <summary>
    /// 删除VariableData
    /// </summary>
    /// <param name="id">主键ID</param>
    /// <param name="db">SqlSugarClient实例</param>
    /// <returns></returns>
    public async Task<int> DeleteAsync(IEnumerable<Variable> variableDatas, SqlSugarClient db)
    {
        var dbList = variableDatas.Select(vd => _mapper.Map<DbVariable>(vd))
                                  .ToList();
        var result = await db.Deleteable<DbVariable>(dbList)
                             .ExecuteCommandAsync();
        return result;
    }

    // public VarDataRepository(IMapper mapper)
    // {
    //     _mapper = mapper;
    // }

    /// <summary>
    /// 为变量添加MQTT服务器关联，并指定别名。
    /// </summary>
    /// <param name="variableMqttList"></param>
    /// <param name="variableDatas">要添加MQTT服务器的变量数据列表。</param>
    /// <returns>成功添加或更新关联的数量。</returns>
     public async Task<int> AddMqttToVariablesAsync(IEnumerable<VariableMqtt> variableMqttList)
    {
        using var db = DbContext.GetInstance();
        await db.BeginTranAsync();

        try
        {
            int affectedCount = 0;
            var variableIds = variableMqttList.Select(vm => vm.Variable.Id).Distinct().ToList();
            var mqttIds = variableMqttList.Select(vm => vm.Mqtt.Id).Distinct().ToList();

            // 1. 一次性查询所有相关的现有别名
            var existingAliases = await db.Queryable<DbVariableMqtt>()
                                          .Where(it => variableIds.Contains(it.VariableId) && mqttIds.Contains(it.MqttId))
                                          .ToListAsync();

            var existingAliasesDict = existingAliases
                .ToDictionary(a => (a.VariableId, a.MqttId), a => a);

            var toInsert = new List<DbVariableMqtt>();
            var toUpdate = new List<DbVariableMqtt>();

            foreach (var variableMqtt in variableMqttList)
            {
                var key = (variableMqtt.Variable.Id, variableMqtt.Mqtt.Id);
                if (existingAliasesDict.TryGetValue(key, out var existingAlias))
                {
                    // 如果存在但别名不同，则准备更新
                    if (existingAlias.MqttAlias != variableMqtt.MqttAlias)
                    {
                        existingAlias.MqttAlias = variableMqtt.MqttAlias;
                        existingAlias.UpdateTime = DateTime.Now;
                        toUpdate.Add(existingAlias);
                    }
                }
                else
                {
                    // 如果不存在，则准备插入
                    toInsert.Add(new DbVariableMqtt
                    {
                        VariableId = variableMqtt.Variable.Id,
                        MqttId = variableMqtt.Mqtt.Id,
                        MqttAlias = variableMqtt.MqttAlias,
                        CreateTime = DateTime.Now,
                        UpdateTime = DateTime.Now
                    });
                }
            }

            // 2. 批量更新
            if (toUpdate.Any())
            {
                var updateResult = await db.Updateable(toUpdate).ExecuteCommandAsync();
                affectedCount += updateResult;
            }

            // 3. 批量插入
            if (toInsert.Any())
            {
                var insertResult = await db.Insertable(toInsert).ExecuteCommandAsync();
                affectedCount += insertResult;
            }

            await db.CommitTranAsync();
            NlogHelper.Info($"成功为 {variableMqttList.Count()} 个变量请求添加/更新了MQTT服务器关联，实际影响 {affectedCount} 个。");
            return affectedCount;
        }
        catch (Exception ex)
        {
            await db.RollbackTranAsync();
            NlogHelper.Error($"为变量添加MQTT服务器关联时发生错误: {ex.Message}", ex);
            // 根据需要，可以向上层抛出异常
            throw;
        }
    }
}