using System; using System.Collections.Generic; using System.ComponentModel.DataAnnotations; using System.Linq; using System.Linq.Expressions; using System.Reflection; using System.Threading; using System.Threading.Tasks; using AutoMapper; using IRaCIS.Core.Domain.Models; using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore.ChangeTracking; using AutoMapper.QueryableExtensions; using EFCore.BulkExtensions; using IRaCIS.Core.Domain.Share; using IRaCIS.Core.Infrastructure; using IRaCIS.Core.Infrastructure.Extention; using Microsoft.Data.SqlClient; namespace IRaCIS.Core.Infra.EFCore { public class Repository : IRepository where TEntity : Entity, new() { public IMapper _mapper { get; set; } public IRaCISDBContext _dbContext { get; set; } public DbSet _dbSet => _dbContext.Set(); public IUserInfo _userInfo { get; set; } public Repository(IRaCISDBContext dbContext, IMapper mapper,IUserInfo userInfo) { _dbContext = dbContext; _mapper = mapper; _userInfo = userInfo; } #region 异步部分 /// /// 仅仅供字典表插入使用,因为efcore 动态映射列的问题 /// /// /// /// /// /// public async Task InsertDictionaryAsync(TFrom from, params EntityVerifyExp[] verify) { var entity = _mapper.Map(from); foreach (var verifyItem in verify.Where(t => t.verifyType != VerifyEnum.OnlyUpdate && t.IsVerify)) { if (await _dbSet.IgnoreQueryFilters().AnyAsync(verifyItem.VerifyExp).ConfigureAwait(false)) { throw new BusinessValidationFailedException(verifyItem.VerifyMsg); } } if (typeof(TEntity) == typeof(Dictionary)) { Type type = typeof(TFrom); //以下是不要ID这个字段的 比如自增列ID 就不能像上名那样写 var properties = type.GetProperties().Where(t => t.Name != "Id"); string strSqlName = string.Join(",", properties.Select(p => $"[{p.Name}]").ToArray()); string strSqlValue = string.Join(",", properties.Select(P => $"@{P.Name}").ToArray()); string strSql = $"insert into {nameof(Dictionary)} ( " + strSqlName + " ) values (" + strSqlValue + ")"; //para Sql是参数 SqlParameter[] para = properties.Select(p => new SqlParameter($"@{p.Name}", p.GetValue(from, null))).ToArray(); _dbContext.Database.ExecuteSqlRaw(strSql, para); return entity; } else { throw new Exception("仅仅供字典表插入使用,因为efcore 动态映射列的问题"); //await _dbSet.BulkInsertAsync(new List() { entity }); //return entity; } } public async Task InsertOrUpdateAsync(TFrom from, bool autoSave = false, params EntityVerifyExp[] verify) { var entity = _mapper.Map(from); if (entity.Id == Guid.Empty) { return await InsertFromDTOAsync(from, autoSave, verify); } else { return await UpdateFromDTOAsync(from, autoSave, false, verify); } } private async Task EntityVerifyAsync(bool isAdd, EntityVerifyExp[] verify, Guid? entitydId = null) { if (isAdd) { foreach (var verifyItem in verify.Where(t => t.verifyType != VerifyEnum.OnlyUpdate && t.IsVerify)) { if (await _dbSet.IgnoreQueryFilters().AnyAsync(verifyItem.VerifyExp).ConfigureAwait(false)) { throw new BusinessValidationFailedException(verifyItem.VerifyMsg); } } } else { foreach (var verifyItem in verify.Where(t => t.verifyType != VerifyEnum.OnlyAdd && t.IsVerify)) { if (verifyItem.verifyType == VerifyEnum.OnlyUpdate) { if (await _dbSet.IgnoreQueryFilters().AnyAsync(verifyItem.VerifyExp).ConfigureAwait(false)) { throw new BusinessValidationFailedException(verifyItem.VerifyMsg); } } else if (verifyItem.verifyType == VerifyEnum.Both) { if (await _dbSet.IgnoreQueryFilters().AnyAsync(verifyItem.VerifyExp.And(t => t.Id != entitydId)).ConfigureAwait(false)) { throw new BusinessValidationFailedException(verifyItem.VerifyMsg); } } } } } public async Task InsertFromDTOAsync(TFrom from, bool autoSave = false, params EntityVerifyExp[] verify) { var entity = _mapper.Map(from); await EntityVerifyAsync(true, verify); await _dbSet.AddAsync(entity).ConfigureAwait(false); await SaveChangesAsync(autoSave); return entity; } /// /// 用前端传递的视图模型字段,更新,同时返回数据库该条记录的原始信息,方便对比某些字段是否更改,进行相应的逻辑操作 /// /// /// /// /// /// /// /// public async Task UpdateFromDTOAsync(TFrom from, bool autoSave = false, bool ignoreDtoNullProperty = true, params EntityVerifyExp[] verify) { var entity = _mapper.Map(from); await EntityVerifyAsync(false, verify, entity.Id); var dbEntity = await _dbSet.IgnoreQueryFilters().FirstOrDefaultAsync(t => t.Id == entity.Id).ConfigureAwait(false); if (dbEntity == null) { throw new BusinessValidationFailedException( " Update object not exist in db,Please check if the parameter Id is passed incorrectly"); } var dbBeforEntity = dbEntity.Clone(); _mapper.Map(from, dbEntity); //DTO null 属性不更新 防止意外操作,导致保存数据错误,或者 add 和update 用一个模型,更新的时候,只传递了部分字段,导致,不想更新的字段,因为没传递值,用null覆盖了 // Guid?属性 为null 时 映射到 Guid 时 默认会变成 Guid.Empty if (ignoreDtoNullProperty) { var dbEntityProp = typeof(TEntity).GetProperties(); foreach (var propertyInfo in from.GetType().GetProperties()) { if (propertyInfo.GetValue(from) == null && dbEntityProp.Any(t=>t.Name== propertyInfo.Name)) { _dbContext.Entry(dbEntity).Property(propertyInfo.Name).IsModified = false; } } } await SaveChangesAsync(autoSave); return dbBeforEntity; } private async Task SaveChangesAsync(bool autoSave) { if (autoSave) { return await SaveChangesAsync(); } else { return false; } } /// /// 通过主键id 和表达式树 更新部分字段 例如 Guid.Parse("8a90c96e-0776-4f7b-82a6-18933d339584"),u => new Dictionary() { ParentId = null, Code = "test" } 默认会去处理更新更新人 更新时间 /// /// /// /// /// /// public async Task UpdatePartialFields(Guid id, Expression> updateFactory, bool autoSave = false, params EntityVerifyExp[] verify) { await EntityVerifyAsync(false, verify, id); var entity = new TEntity() { Id = id }; var entityEntry = _dbContext.Entry(entity); entityEntry.State = EntityState.Detached; List list = ((MemberInitExpression)updateFactory.Body).Bindings.Select(mb => mb.Member.Name) .Select(propName => typeof(TEntity).GetProperty(propName, BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic)).ToList(); Func func = updateFactory.Compile(); TEntity applyObj = func(entity); foreach (PropertyInfo prop in list) { object value = prop.GetValue((object)applyObj); prop.SetValue((object)entity, value); _dbContext.Entry(entity).Property(prop.Name).IsModified = true; } await SaveChangesAsync(autoSave); #region Test //updateFactory.Compile()(entity); //List propNameList = ((MemberInitExpression)updateFactory.Body).Bindings.Select(mb => mb.Member.Name).ToList(); //foreach (string propName in propNameList) //{ // _dbContext.Entry(entity).Property(propName).IsModified = true; //} #endregion } /// /// 部分字段更新 (只更新传递的字段名 new[] {nameof(User.Name), nameof(User.Age)) /// new Dictionary() { ParentId = null, Code = "test",Id=Guid.Parse("8a90c96e-0776-4f7b-82a6-18933d339584")},new[] {nameof(Dictionary.Name), nameof(Dictionary.Age)) /// /// /// 更新的字段数组 /// /// /// /// public async Task UpdatePartialFields(TEntity entity, string[] propertyNames, bool autoSave = false, bool ignoreEntityNullProperty = true, params EntityVerifyExp[] verify) { await EntityVerifyAsync(false, verify, entity.Id); var entityEntry = _dbContext.Entry(entity); entityEntry.State = EntityState.Detached; foreach (var propertyName in propertyNames) { _dbContext.Entry(entity).Property(propertyName).IsModified = true; } // 忽略空值 IgnoreNullValues(ref entity, ignoreEntityNullProperty); return entityEntry.Entity; } /// /// 更新 排除某些字段的更新 排除方式: new[] {nameof(User.Name), nameof(User.Age) /// /// /// /// /// /// /// public async Task UpdateExcludeFields(TEntity entity, string[] propertyNames, bool autoSave = false, bool ignoreEntityNullProperty = true, params EntityVerifyExp[] verify) { await EntityVerifyAsync(false, verify, entity.Id); var entityEntry = _dbContext.Entry(entity); entityEntry.State = EntityState.Modified; foreach (var propertyName in propertyNames) { _dbContext.Entry(entity).Property(propertyName).IsModified = false; } // 忽略空值 IgnoreNullValues(ref entity, ignoreEntityNullProperty); return entityEntry.Entity; } /// /// 忽略空值属性 /// /// /// private void IgnoreNullValues(ref TEntity entity, bool? ignoreNullValues = null) { var isIgnore = ignoreNullValues; if (isIgnore == false) return; // 获取所有的属性 var properties = _dbSet.EntityType.GetProperties(); if (properties == null) return; foreach (var propety in properties) { var entityProperty = _dbContext.Entry(entity).Property(propety.Name); var propertyValue = entityProperty?.CurrentValue; var propertyType = entityProperty?.Metadata?.PropertyInfo?.PropertyType; // 判断是否是无效的值,比如为 null,默认时间,以及空 Guid 值 var isInvalid = propertyValue == null || (propertyType == typeof(DateTime) && propertyValue?.ToString() == new DateTime().ToString()) || (propertyType == typeof(DateTimeOffset) && propertyValue?.ToString() == new DateTimeOffset().ToString()) || (propertyType == typeof(Guid) && propertyValue?.ToString() == Guid.Empty.ToString()); if (isInvalid && entityProperty != null) { entityProperty.IsModified = false; } } } public async Task SaveChangesAsync(CancellationToken cancellationToken = default) { return await _dbContext.SaveChangesAsync(cancellationToken) > 0; } public async ValueTask FindAsync(Guid id, CancellationToken cancellationToken = default) { return await _dbContext.FindAsync(id); } //有可能是联合主键,本项目没用到,用Guid public async ValueTask FindAsync(object[] keyValues, CancellationToken cancellationToken) { return await _dbContext.FindAsync(keyValues); } public async Task FirstOrDefaultAsync(Expression> exp = null, bool ignoreQueryFilters = false) { var query = _dbSet.AsQueryable(); if (ignoreQueryFilters) { query = query.IgnoreQueryFilters(); } if (exp == null) return await query.FirstOrDefaultAsync().ConfigureAwait(false); return await query.FirstOrDefaultAsync(exp).ConfigureAwait(false); } public async ValueTask AddAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default) { await _dbSet.AddAsync(entity).ConfigureAwait(false); if (autoSave) { await SaveChangesAsync(cancellationToken); } return entity; } public async Task> AddRangeAsync(IEnumerable entities) { await _dbSet.AddRangeAsync(entities).ConfigureAwait(false); return entities; } public async Task AddRangeAsync(IEnumerable entities, bool autoSave = false, CancellationToken cancellationToken = default) { await _dbSet.AddRangeAsync(entities).ConfigureAwait(false); if (autoSave) { return await SaveChangesAsync(cancellationToken); } else { return false; } } public async Task MaxAsync(Expression> selector) { return await _dbSet.AsNoTracking().MaxAsync(selector); } public async Task AnyAsync(Expression> exp, bool ignoreQueryFilters = false) { var query = _dbSet.AsQueryable(); if (ignoreQueryFilters) { query = query.IgnoreQueryFilters(); } return await query.AsNoTracking().AnyAsync(exp); } public async Task CountAsync(Expression> whereLambda = null, bool ignoreQueryFilters = false) { var query = _dbSet.AsQueryable(); if (ignoreQueryFilters) { query = query.IgnoreQueryFilters(); } return whereLambda == null ? await query.AsNoTracking().CountAsync() : await query.AsNoTracking().CountAsync(whereLambda); } public async Task UpdateAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default) { _dbSet.Update(entity); if (autoSave) { return await SaveChangesAsync(cancellationToken); } else { return false; } } public async Task DeleteAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default) { _dbSet.Remove(entity); //var entry = _dbSet.Attach(entity); //entry.State = EntityState.Deleted; if (autoSave) { return await SaveChangesAsync(cancellationToken); } else { return false; } } public async Task BatchDeleteAsync(Expression> deleteFilter) { return await _dbSet.IgnoreQueryFilters().Where(deleteFilter).BatchDeleteAsync() > 0; } public async Task BatchUpdateAsync(Expression> where, Expression> updateFactory) { var bindings = ((MemberInitExpression)updateFactory.Body).Bindings.ToList(); if (typeof(IAuditUpdate).IsAssignableFrom(typeof(TEntity))) { bindings.Add(Expression.Bind(typeof(TEntity).GetMember(nameof(IAuditUpdate.UpdateTime))[0], Expression.Constant(DateTime.Now))); bindings.Add(Expression.Bind(typeof(TEntity).GetMember(nameof(IAuditUpdate.UpdateUserId))[0], Expression.Constant(_userInfo.Id))); } var member = Expression.MemberInit(Expression.New(typeof(TEntity)), bindings); var factory = Expression.Lambda>(member, Expression.Parameter(typeof(TEntity), "x")); return await _dbSet.IgnoreQueryFilters().Where(where).BatchUpdateAsync(factory) > 0; } /// /// 杨中科 test 有问题 对于guid? 赋值null的时候 /// /// /// /// //public async Task BatchUpdateAsync(Expression> where, // Expression> updateFactory) //{ // var builder = _dbContext.BatchUpdate(); // Func func = updateFactory.Compile(); // List list = ((MemberInitExpression)updateFactory.Body).Bindings.Select((Func)(_param1 => _param1.Member.Name)).Select((Func)(_param1 => (PropertyInfo)typeof(TEntity).GetProperty(_param1, BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic))).ToList(); // TEntity applyObj = func(new TEntity()); // foreach (PropertyInfo prop in list) // { // var propName = prop.Name; // var value = prop.GetValue(applyObj); // builder.Set(propName, value?? DBNull.Value); // //PropertyInfo property = typeof(TEntity).GetProperty(prop.Name); // //Type propertyType = property.PropertyType; // //ParameterExpression parameterExpression = Expression.Parameter(typeof(TEntity)); // //Type delegateType = typeof(Func<,>).MakeGenericType(typeof(TEntity), propertyType); // //LambdaExpression nameExpr = Expression.Lambda(delegateType, (Expression)Expression.MakeMemberAccess((Expression)parameterExpression, (MemberInfo)property), parameterExpression); // //Expression expression = (Expression)Expression.Constant(value); // //expression = (Expression)Expression.Convert(expression, propertyType); // //LambdaExpression valueExpr = Expression.Lambda(delegateType, expression, parameterExpression); // //builder.Set< >(Func nameExpr, valueExpr); // } // return await builder.Where(where).ExecuteAsync()>0; //} //// Z.EntityFramework.Plus.EFCore //public async Task BatchDeleteAsync(Expression> deleteFilter) //{ // return await _dbSet.IgnoreQueryFilters().Where(deleteFilter).DeleteFromQueryAsync() > 0; //} //public async Task BatchUpdateAsync(Expression> where, Expression> updateFactory) //{ // return await _dbSet.IgnoreQueryFilters().Where(where).UpdateFromQueryAsync(updateFactory) > 0; //} //public async Task UpdateManyAsync(IEnumerable entities, bool autoSave = false, CancellationToken cancellationToken = default) //{ // _dbSet.UpdateRange(entities); // if (autoSave) // { // return await SaveChangesAsync(cancellationToken); // } // else // { // return false; // } //} //public async Task DeleteManyAsync(IEnumerable entities, bool autoSave = false, CancellationToken cancellationToken = default) //{ // _dbSet.RemoveRange(entities); // if (autoSave) // { // return await SaveChangesAsync(cancellationToken); // } // else // { // return false; // } //} #endregion #region 非异步部分 public TEntity ImageFind(Guid id, Type type) { //重传的时候 状态为修改 上传的时候状态为添加 内存中有,就不要重复查询数据库了 var list = _dbContext.ChangeTracker.Entries() .Where(u => (u.State == EntityState.Added || u.State == EntityState.Modified) && (u.Entity.GetType() == type)).Select(t => t.Entity as TEntity); var entity = list.FirstOrDefault(t => t.Id == id); if (entity == null) { return _dbSet.FirstOrDefault(t => t.Id == id); } else { return entity; } } public IQueryable AsQueryable(bool ignoreQueryFilters = false) { var query = _dbSet.AsQueryable(); if (ignoreQueryFilters) { query = query.IgnoreQueryFilters(); } return query.AsNoTracking(); } public IQueryable Select(Expression> selector) { return _dbSet.AsNoTracking().Select(selector); } public IQueryable WhereIf(bool condition, Expression> filter) { return condition ? _dbSet.AsNoTracking().Where(filter) : _dbSet.AsNoTracking(); } public IQueryable Where(Expression> exp = null, bool isTraking = false, bool ignoreQueryFilters = false) { IQueryable query = _dbSet; if (!isTraking) { query = query.AsNoTracking(); } if (ignoreQueryFilters) { query = query.IgnoreQueryFilters(); } if (exp != null) { query = query.Where(exp); } return query; } public EntityEntry Entry(TEntity t) { return _dbContext.Entry(t); } public EntityEntry Attach(TEntity entity) { return _dbSet.Attach(entity); } public void Detached(TEntity t) { _dbContext.Entry(t).State = EntityState.Detached; } // automapper 相关 public IQueryable ProjectTo(IConfigurationProvider configuration, object parameters, params Expression>[] membersToExpand) { return _dbSet.AsNoTracking().ProjectTo(configuration, parameters, membersToExpand); } public IQueryable ProjectTo(IConfigurationProvider configuration, params Expression>[] membersToExpand) { return _dbSet.AsNoTracking().ProjectTo(configuration, membersToExpand); } public IQueryable ProjectTo(IConfigurationProvider configuration, IDictionary parameters, params string[] membersToExpand) { return _dbSet.AsNoTracking().ProjectTo(configuration, parameters, membersToExpand); } #endregion } }