重构项目，将项目拆分（临时提交）
This commit is contained in:
409
DMS.Infrastructure/Repositories/VarDataRepository.cs
Normal file
409
DMS.Infrastructure/Repositories/VarDataRepository.cs
Normal file
@@ -0,0 +1,409 @@
|
||||
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;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user