using System; using System.Collections.Generic; using System.ComponentModel.DataAnnotations; using System.ComponentModel.DataAnnotations.Schema; 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 IRaCIS.Core.Infrastructure; using IRaCIS.Core.Infrastructure.Extention; using Microsoft.Data.SqlClient; using Z.EntityFramework.Plus; namespace IRaCIS.Core.Infra.EFCore { public class Repository : IRepository where TEntity : Entity { public IMapper _mapper { get; set; } public IRaCISDBContext _dbContext { get; set; } public DbSet _dbSet => _dbContext.Set(); public Repository(IRaCISDBContext dbContext, IMapper mapper) { _dbContext = dbContext; _mapper = mapper; } #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); if (autoSave) { await SaveChangesAsync(); } 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) { foreach (var propertyInfo in from.GetType().GetProperties()) { if (propertyInfo.GetValue(from) == null) { _dbContext.Entry(dbEntity).Property(propertyInfo.Name).IsModified = false; } } } if (autoSave) { await SaveChangesAsync(); } return dbBeforEntity; } //public TEntity UpdatePartialFields(Guid id, Expression> updateFactory, bool autoSave = false, bool ignoreEntityNullProperty = true, params EntityVerifyExp[] verify) //{ // EntityVerifyAsync(false, verify, id).RunSynchronously(); // var entity = updateFactory.Compile().Invoke(new TEntity() { Id = id }); // var entityEntry = _dbContext.Entry(entity); // entityEntry.State = EntityState.Detached; //} /// /// 部分字段更新 (只更新传递的字段名) /// /// /// 更新的字段数组 /// /// /// /// public TEntity UpdatePartialFields(TEntity entity, string[] propertyNames, bool autoSave = false, bool ignoreEntityNullProperty = true, params EntityVerifyExp[] verify) { EntityVerifyAsync(false, verify, entity.Id).RunSynchronously(); 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; } /// /// 更新 排除某些字段的更新 /// /// /// /// /// /// /// public TEntity UpdateExcludeFields(TEntity entity, string[] propertyNames, bool autoSave = false, bool ignoreEntityNullProperty = true, params EntityVerifyExp[] verify) { EntityVerifyAsync(false, verify, entity.Id).RunSynchronously(); 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 UpdateAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default) { _dbSet.Update(entity); if (autoSave) { return await SaveChangesAsync(cancellationToken); } else { return false; } } 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 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 DeleteManyAsync(IEnumerable entities, bool autoSave = false, CancellationToken cancellationToken = default) { _dbSet.RemoveRange(entities); 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); } // Z.EntityFramework.Plus.EFCore public async Task DeleteFromQueryAsync(Expression> deleteFilter) { return await _dbSet.IgnoreQueryFilters().Where(deleteFilter).DeleteFromQueryAsync() > 0; } public async Task UpdateFromQueryAsync(Expression> where, Expression> updateFactory) { return await _dbSet.IgnoreQueryFilters().Where(where).UpdateFromQueryAsync(updateFactory) > 0; } #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 } }