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; protected Expression> _filter; protected Func _filterCompile; public BaseRepository(IFreeSql fsql, Expression> filter) : base() { _fsql = fsql ?? throw new NullReferenceException("fsql 参数不可为空"); _filter = filter; _filterCompile = filter?.Compile(); } public ISelect Select => _fsql.Select().Where(_filter); public IUpdate UpdateDiy => _fsql.Update().Where(_filter); public int Delete(Expression> predicate) => _fsql.Delete().Where(_filter).Where(predicate).ExecuteAffrows(); public int Delete(TEntity entity) { ValidatorEntityAndThrow(entity); return _fsql.Delete(entity).Where(_filter).ExecuteAffrows(); } public Task DeleteAsync(Expression> predicate) => _fsql.Delete().Where(_filter).Where(predicate).ExecuteAffrowsAsync(); public Task DeleteAsync(TEntity entity) { ValidatorEntityAndThrow(entity); return _fsql.Delete(entity).Where(_filter).ExecuteAffrowsAsync(); } public virtual TEntity Insert(TEntity entity) { ValidatorEntityAndThrow(entity); switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return _fsql.Insert().AppendData(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) { ValidatorEntityAndThrow(entitys); switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return _fsql.Insert().AppendData(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) { ValidatorEntityAndThrow(entity); switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return (await _fsql.Insert().AppendData(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) { ValidatorEntityAndThrow(entitys); switch (_fsql.Ado.DataType) { case DataType.SqlServer: case DataType.PostgreSQL: return _fsql.Insert().AppendData(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) { ValidatorEntityAndThrow(entity); return _fsql.Update().SetSource(entity).Where(_filter).ExecuteAffrows(); } public Task UpdateAsync(TEntity entity) { ValidatorEntityAndThrow(entity); return _fsql.Update().SetSource(entity).Where(_filter).ExecuteAffrowsAsync(); } protected void ValidatorEntityAndThrow(TEntity entity) => ValidatorEntityAndThrow(new[] { entity }); protected virtual void ValidatorEntityAndThrow(IEnumerable entitys) { foreach (var entity in entitys) { if (_filterCompile?.Invoke(entity) == false) throw new Exception($"FreeSql.Repository Insert 失败,因为设置了 {_filter},插入的数据不符合"); } } } public abstract class BaseRepository : BaseRepository, IRepository where TEntity : class { public BaseRepository(IFreeSql fsql, Expression> filter) : base(fsql, filter) { } public int Delete(TKey id) => _fsql.Delete(id).Where(_filter).ExecuteAffrows(); public Task DeleteAsync(TKey id) => _fsql.Delete(id).Where(_filter).ExecuteAffrowsAsync(); public TEntity Find(TKey id) => _fsql.Select(id).Where(_filter).ToOne(); public Task FindAsync(TKey id) => _fsql.Select(id).Where(_filter).ToOneAsync(); public TEntity Get(TKey id) => Find(id); public Task GetAsync(TKey id) => FindAsync(id); } }