using System; using System.Collections.Generic; using System.Linq; using System.Linq.Expressions; using System.Threading.Tasks; namespace FreeSql { public abstract class BaseRepository : IRepository where TEntity : class { protected IFreeSql _fsql; public IDataFilter DataFilter { get; } = new DataFilter(); Func _asTableVal; protected Func AsTable { get => _asTableVal; set { _asTableVal = value; AsTableSelect = value == null ? null : new Func((a, b) => a == EntityType ? value(b) : null); } } protected Func AsTableSelect { get; private set; } internal Func AsTableSelectInternal => AsTableSelect; protected Type EntityType { get; } = typeof(TEntity); protected BaseRepository(IFreeSql fsql, Expression> filter, Func asTable = null) : base() { _fsql = fsql ?? throw new NullReferenceException(nameof(fsql)); DataFilter.Apply("", filter); AsTable = asTable; } public ISelect Select => OrmSelect(null); public IUpdate UpdateDiy => OrmUpdate(null); public int Delete(Expression> predicate) => OrmDelete(null).Where(predicate).ExecuteAffrows(); public int Delete(TEntity entity) => OrmDelete(entity).ExecuteAffrows(); public Task DeleteAsync(Expression> predicate) => OrmDelete(null).Where(predicate).ExecuteAffrowsAsync(); public Task DeleteAsync(TEntity entity) => OrmDelete(entity).ExecuteAffrowsAsync(); public virtual TEntity Insert(TEntity entity) { switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return OrmInsert(entity).ExecuteInserted().FirstOrDefault(); case DataType.MySql: case DataType.Oracle: case DataType.Sqlite: default: throw new NotImplementedException($"{_fsql.Ado.DataType}不支持类似returning或output类型的特性,请参考FreeSql插入数据的方法重新实现。"); } } public virtual List Insert(IEnumerable entitys) { switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return OrmInsert(entitys).ExecuteInserted(); case DataType.MySql: case DataType.Oracle: case DataType.Sqlite: default: throw new NotImplementedException($"{_fsql.Ado.DataType}不支持类似returning或output类型的特性,请参考FreeSql插入数据的方法重新实现。"); } } async public virtual Task InsertAsync(TEntity entity) { switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return (await OrmInsert(entity).ExecuteInsertedAsync()).FirstOrDefault(); case DataType.MySql: case DataType.Oracle: case DataType.Sqlite: default: throw new NotImplementedException($"{_fsql.Ado.DataType}不支持类似returning或output类型的特性,请参考FreeSql插入数据的方法重新实现。"); } } public virtual Task> InsertAsync(IEnumerable entitys) { switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return OrmInsert(entitys).ExecuteInsertedAsync(); case DataType.MySql: case DataType.Oracle: case DataType.Sqlite: default: throw new NotImplementedException($"{_fsql.Ado.DataType}不支持类似returning或output类型的特性,请参考FreeSql插入数据的方法重新实现。"); } } public int Update(TEntity entity) => OrmUpdate(entity).ExecuteAffrows(); public Task UpdateAsync(TEntity entity) => OrmUpdate(entity).ExecuteAffrowsAsync(); protected ISelect OrmSelect(object dywhere) { var select = _fsql.Select(dywhere); var filters = (DataFilter as DataFilter)._filters.Where(a => a.Value.IsEnabled == true); foreach (var filter in filters) select.Where(filter.Value.Expression); return select.AsTable(AsTableSelect); } protected IUpdate OrmUpdate(object dywhere) { var entityObj = dywhere as TEntity; var update = _fsql.Update(dywhere); var filters = (DataFilter as DataFilter)._filters.Where(a => a.Value.IsEnabled == true); foreach (var filter in filters) { if (entityObj != null && filter.Value.ExpressionDelegate?.Invoke(entityObj) == false) throw new Exception($"FreeSql.Repository Update 失败,因为设置了 {filter.Key}: {filter.Value.Expression},更新的数据不符合"); update.Where(filter.Value.Expression); } return update.AsTable(AsTable); } protected IDelete OrmDelete(object dywhere) { var delete = _fsql.Delete(dywhere); var filters = (DataFilter as DataFilter)._filters.Where(a => a.Value.IsEnabled == true); foreach (var filter in filters) delete.Where(filter.Value.Expression); return delete.AsTable(AsTable); } protected IInsert OrmInsert(TEntity entity) => OrmInsert(new[] { entity }); protected IInsert OrmInsert(IEnumerable entitys) { var insert = _fsql.Insert(entitys); var filters = (DataFilter as DataFilter)._filters.Where(a => a.Value.IsEnabled == true); foreach (var filter in filters) { foreach (var entity in entitys) if (entity != null && filter.Value.ExpressionDelegate?.Invoke(entity) == false) throw new Exception($"FreeSql.Repository Insert 失败,因为设置了 {filter.Key}: {filter.Value.Expression},插入的数据不符合"); } return insert.AsTable(AsTable); } protected void ApplyDataFilter(string name, Expression> exp) => DataFilter.Apply(name, exp); } public abstract class BaseRepository : BaseRepository, IRepository where TEntity : class { public BaseRepository(IFreeSql fsql, Expression> filter, Func asTable = null) : base(fsql, filter, asTable) { } public int Delete(TKey id) => OrmDelete(id).ExecuteAffrows(); public Task DeleteAsync(TKey id) => OrmDelete(id).ExecuteAffrowsAsync(); public TEntity Find(TKey id) => OrmSelect(id).ToOne(); public Task FindAsync(TKey id) => OrmSelect(id).ToOneAsync(); public TEntity Get(TKey id) => Find(id); public Task GetAsync(TKey id) => FindAsync(id); } }