using System.Diagnostics; using DMS.Core.Interfaces; using DMS.Core.Interfaces.Repositories; using DMS.Core.Models; using DMS.Infrastructure.Data; using DMS.Infrastructure.Entities; using AutoMapper; using Microsoft.Extensions.Logging; namespace DMS.Infrastructure.Repositories; /// /// 用户仓储实现类,负责用户数据的持久化操作。 /// 继承自 并实现 接口。 /// public class UserRepository : BaseRepository, IUserRepository { private readonly IMapper _mapper; /// /// 构造函数,注入 AutoMapper 和 SqlSugarDbContext。 /// /// AutoMapper 实例,用于实体模型和数据库模型之间的映射。 /// SqlSugar 数据库上下文,用于数据库操作。 /// 日志记录器实例。 public UserRepository(IMapper mapper, SqlSugarDbContext dbContext, ILogger logger) : base(dbContext, logger) { _mapper = mapper; } /// /// 异步根据ID获取单个用户。 /// /// 用户的唯一标识符。 /// 对应的用户实体,如果不存在则为null。 public async Task GetByIdAsync(int id) { var dbUser = await base.GetByIdAsync(id); return _mapper.Map(dbUser); } /// /// 异步获取所有用户。 /// /// 包含所有用户实体的列表。 public async Task> GetAllAsync() { var dbList = await base.GetAllAsync(); return _mapper.Map>(dbList); } /// /// 异步添加新用户。 /// /// 要添加的用户实体。 /// 添加成功后的用户实体(包含数据库生成的ID等信息)。 public async Task AddAsync(User entity) { var dbUser = await base.AddAsync(_mapper.Map(entity)); return _mapper.Map(dbUser, entity); } /// /// 异步更新现有用户。 /// /// 要更新的用户实体。 /// 受影响的行数。 public async Task UpdateAsync(User entity) => await base.UpdateAsync(_mapper.Map(entity)); /// /// 异步删除用户。 /// /// 要删除的用户实体。 /// 受影响的行数。 public async Task DeleteAsync(User entity) => await base.DeleteAsync(new List { _mapper.Map(entity) }); /// /// 异步根据ID删除用户。 /// /// 要删除用户的唯一标识符。 /// 受影响的行数。 public async Task DeleteByIdAsync(int id) { var stopwatch = new Stopwatch(); stopwatch.Start(); var result = await _dbContext.GetInstance().Deleteable(new DbUser() { Id = id }) .ExecuteCommandAsync(); stopwatch.Stop(); _logger.LogInformation($"Delete {typeof(DbUser)},ID={id},耗时:{stopwatch.ElapsedMilliseconds}ms"); return result; } /// /// 异步获取指定数量的用户。 /// /// 要获取的用户数量。 /// 包含指定数量用户实体的列表。 public new async Task> TakeAsync(int number) { var dbList = await base.TakeAsync(number); return _mapper.Map>(dbList); } public async Task> AddBatchAsync(List entities) { var dbEntities = _mapper.Map>(entities); var addedEntities = await base.AddBatchAsync(dbEntities); return _mapper.Map>(addedEntities); } /// /// 异步根据实体列表批量删除用户。 /// /// 要删除的用户实体列表。 /// 受影响的行数。 public async Task DeleteAsync(List entities) { var dbEntities = _mapper.Map>(entities); return await base.DeleteAsync(dbEntities); } }