using AutoMapper; using DMS.Core.Models; using DMS.Infrastructure.Interfaces; using System.Diagnostics; using DMS.Infrastructure.Entities; using DMS.Infrastructure.Data; namespace DMS.Infrastructure.Repositories; /// /// VariableData仓储类,用于操作DbVariableData实体 /// public class VarDataRepository : BaseRepository { public VarDataRepository(SqlSugarDbContext dbContext) : base(dbContext) { } public override async Task> GetAllAsync() { Stopwatch stopwatch = new Stopwatch(); stopwatch.Start(); var result = await Db.Queryable() .Includes(d => d.VariableTable) .Includes(d => d.VariableTable.Device) .ToListAsync(); stopwatch.Stop(); //NlogHelper.Info($"获取所有VariableData耗时:{stopwatch.ElapsedMilliseconds}ms"); return result; } public async Task> GetByVariableTableIdAsync(int varTableId) { Stopwatch stopwatch = new Stopwatch(); stopwatch.Start(); var result = await Db.Queryable() .Where(d => d.VariableTableId == varTableId) .ToListAsync(); stopwatch.Stop(); //NlogHelper.Info($"获取变量表的所有变量{result.Count()}个耗时:{stopwatch.ElapsedMilliseconds}ms"); return result; } // public VarDataRepository(IMapper mapper) // { // _mapper = mapper; // } /* /// /// 为变量添加MQTT服务器关联,并指定别名。 /// /// /// 要添加MQTT服务器的变量数据列表。 /// 成功添加或更新关联的数量。 public async Task AddMqttToVariablesAsync(IEnumerable variableMqttList) { 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() .Where(it => variableIds.Contains(it.VariableId) && mqttIds.Contains(it.MqttId)) .ToListAsync(); var existingAliasesDict = existingAliases .ToDictionary(a => (a.VariableId, a.Mqtt.Id), a => a); var toInsert = new List(); var toUpdate = new List(); 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; } } */ }