1. 程式人生 > >EF通用資料層封裝類(支援讀寫分離,一主多從)

EF通用資料層封裝類(支援讀寫分離,一主多從)

原文: EF通用資料層封裝類(支援讀寫分離,一主多從)

淺談orm

  記得四年前在學校第一次接觸到 Ling to Sql,那時候瞬間發現不用手寫sql語句是多麼的方便,後面慢慢的接觸了許多orm框架,像 EF,Dapper,Hibernate,ServiceStack.OrmLite 等。當然每種orm都有各自的優勢,也有不足的地方。園子裡也有很多大神開源了他們寫的orm,如SqlSugar,Chloe.ORM,CYQ.Data 等。先不說這些開源的orm使用度怎麼樣,我覺得起碼從開源的精神上就很可嘉了,我也曾下載過這幾位大神的原始碼進行學習。
  所有orm最終體現的一點就是方便,減少程式設計師重複性的工作,當然目前還有一些公司還是使用手寫sql的方式來做,我覺得整個專案都使用手寫sql來做真的是有點閒到蛋疼,並不是不推薦手寫sql的方式,只是個人覺得最基本的增刪改查這些都手寫的話,那其實考驗不是能力,而是耐力。有人說手寫sql的方式可控性強,效能高,我想說的是orm也能做到,關鍵是你怎麼去使用。
  orm的優點非常明顯,開發便捷,但或許也是由於這個優點,讓很多偷懶的程式設計師也會漸漸忘了sql語句的寫法,我遇到過很多的程式設計師朋友用了EF後,手寫sql,檢視、儲存過程這些都不想用了,我個人覺手寫sql這種還是必要的。不然某一天你看到別人的程式裡面寫著 “exec xxxx”,你就會突然覺得“啊,好像在哪裡見過.....”。所以我想說的是“該出手時還是得出手"。

淺談Entity Framework

  Entity Framework 是微軟家的orm框架,隨著 Entity Framework 不斷的完善強化,目前相信使用的比例相對其他的orm來說還是較高的。像我目前使用的最多的就是EF和Dapper。確實,EF用起來開發過程中會方便很多,畢竟EF走過了這麼年頭,無論是成熟度,還是效能等都提高了很多,也有很多開發者為EF提供了擴充套件功能,如entity framework extended 等。而且作為.net開發者來說專案通用性也很強,資料也多,微軟在這塊的更新力度也很給力。不過之前剛出的EF Core也存在一些坑,畢竟還是初期階段,相信現在後面會越來越好的。

  Entity Framework  提供了三種開發模式,code first,db first,model first。目前用的最多的就屬code first了。至於這三種模式的簡單使用和區別,大家可以參考下這篇文章

  我曾聽一些朋友說過說EF使用起來效能很差,生成的sql語句很難看等。我覺得說這種話之前還是先檢查下程式碼或者多看下一些EF文章吧,要先確保自己沒給自己挖坑,然後才能指責別人的不好。如果真心覺得EF或者其他的orm用起來很不爽,那就自己寫一個吧,我也曾經和同事用Dapper擴充套件一個通用的orm,當時是出於一種學習和使用方便的角度。

Entity Framework 通用資料層類

  這裡提供下 EF 通用資料層父類方法,其實網上也有很多人提供了自己專案中的 EF 通用資料層父類方法,所以這裡提供的並不是最優和最好的選擇,只能說是可以通用的類,方便大家學習和使用,具體程式碼如下:

DbContextFactory DbContext工廠類

    public class DbContextFactory
    {
        public  DbContext GetDbContext()
        {
            string key = typeof(DBContext.DbContextFactory).Name + "XJHDbContext";
            DbContext dbContext = CallContext.GetData(key) as DbContext;
            if (dbContext == null)
            {
                dbContext = new XJHDbContext();
                CallContext.SetData(key, dbContext);
            }
            return dbContext;
        }
    }

DbBase 資料層通用操作類

 public class DbBase
    {
        protected DbContext Db = new DbContextFactory().GetDbContext();
        
        #region 自定義其他方法

        /// <summary>
        /// 執行儲存過程或自定義sql語句--返回集合(自定義返回型別)
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public List<TModel> Query<TModel>(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            //儲存過程(exec getActionUrlId @name,@ID)
            if (cmdType == CommandType.StoredProcedure)
            {
                StringBuilder paraNames = new StringBuilder();
                foreach (var sqlPara in parms)
                {
                    paraNames.Append($" @{sqlPara},");
                }
                sql = paraNames.Length > 0 ? $"exec {sql} {paraNames.ToString().Trim(',')}" : $"exec {sql} ";
            }
            var list = Db.Database.SqlQuery<TModel>(sql, parms.ToArray());
            var enityList = list.ToList();
            return enityList;
        }

        /// <summary>
        /// 自定義語句和儲存過程的增刪改--返回影響的行數
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public int Execute(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            //儲存過程(exec getActionUrlId @name,@ID)
            if (cmdType == CommandType.StoredProcedure)
            {
                StringBuilder paraNames = new StringBuilder();
                foreach (var sqlPara in parms)
                {
                    paraNames.Append($" @{sqlPara},");
                }
                sql = paraNames.Length > 0 ?
                    $"exec {sql} {paraNames.ToString().Trim(',')}" :
                    $"exec {sql} ";
            }
            int ret = Db.Database.ExecuteSqlCommand(sql, parms.ToArray());
            return ret;
        }

        #endregion 自定義其他方法
    }

    /// <summary>
    /// mssql資料庫 資料層 父類
    /// </summary>
    /// <typeparam name="T"></typeparam>
    public class DbBase<T> : DbBase where T : class, new()
    {
        #region INSERT

        /// <summary>
        /// 新增 實體
        /// </summary>
        /// <param name="model"></param>
        /// <returns></returns>
        public void Insert(T model)
        {
            Db.Set<T>().Add(model);

        }

        /// <summary>
        /// 普通批量插入
        /// </summary>
        /// <param name="datas"></param>
        public void InsertRange(List<T> datas)
        {
            Db.Set<T>().AddRange(datas);
        }

        #endregion INSERT

        #region DELETE

        /// <summary>
        /// 根據模型刪除
        /// </summary>
        /// <param name="model">包含要刪除id的物件</param>
        /// <returns></returns>
        public void Delete(T model)
        {
            Db.Set<T>().Attach(model);
            Db.Set<T>().Remove(model);
        }

        /// <summary>
        /// 刪除
        /// </summary>
        /// <param name="whereLambda"></param>
        public void Delete(Expression<Func<T, bool>> whereLambda)
        {
            Db.Set<T>().Where(whereLambda).Delete();
        }

        #endregion DELETE

        #region UPDATE

        /// <summary>
        /// 單個物件指定列修改
        /// </summary>
        /// <param name="model">要修改的實體物件</param>
        /// <param name="proNames">要修改的 屬性 名稱</param>
        /// <param name="isProUpdate"></param>
        /// <returns></returns>
        public void Update(T model, List<string> proNames, bool isProUpdate = true)
        {
            //將 物件 新增到 EF中
            Db.Set<T>().Attach(model);
            var setEntry = ((IObjectContextAdapter)Db).ObjectContext.ObjectStateManager.GetObjectStateEntry(model);
            //指定列修改
            if (isProUpdate)
            {
                foreach (string proName in proNames)
                {
                    setEntry.SetModifiedProperty(proName);
                }
            }
            //忽略類修改
            else
            {
                Type t = typeof(T);
                List<PropertyInfo> proInfos = t.GetProperties(BindingFlags.Instance | BindingFlags.Public).ToList();
                foreach (var item in proInfos)
                {
                    string proName = item.Name;
                    if (proNames.Contains(proName))
                    {
                        continue;
                    }
                    setEntry.SetModifiedProperty(proName);
                }
            }
        }

        /// <summary>
        /// 單個物件修改
        /// </summary>
        /// <param name="model"></param>
        /// <returns></returns>
        public void Update(T model)
        {
            DbEntityEntry entry = Db.Entry<T>(model);
            Db.Set<T>().Attach(model);
            entry.State = EntityState.Modified;

        }

        /// <summary>
        /// 批量修改
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <param name="updateExpression"></param>
        public void Update(Expression<Func<T, bool>> whereLambda, Expression<Func<T, T>> updateExpression)
        {
            Db.Set<T>().Where(whereLambda).Update(updateExpression);
        }

        /// <summary>
        /// 批量修改
        /// </summary>
        /// <param name="models"></param>
        /// <returns></returns>
        public void UpdateAll(List<T> models)
        {
            foreach (var model in models)
            {
                DbEntityEntry entry = Db.Entry(model);
                entry.State = EntityState.Modified;
            }


        }

        /// <summary>
        /// 批量統一修改
        /// </summary>
        /// <param name="model">要修改的實體物件</param>
        /// <param name="whereLambda">查詢條件</param>
        /// <param name="modifiedProNames"></param>
        /// <returns></returns>
        public void Update(T model, Expression<Func<T, bool>> whereLambda, params string[] modifiedProNames)
        {
            //查詢要修改的資料
            List<T> listModifing = Db.Set<T>().Where(whereLambda).ToList();
            Type t = typeof(T);
            List<PropertyInfo> proInfos = t.GetProperties(BindingFlags.Instance | BindingFlags.Public).ToList();
            Dictionary<string, PropertyInfo> dictPros = new Dictionary<string, PropertyInfo>();
            proInfos.ForEach(p =>
            {
                if (modifiedProNames.Contains(p.Name))
                {
                    dictPros.Add(p.Name, p);
                }
            });
            if (dictPros.Count <= 0)
            {
                throw new Exception("指定修改的欄位名稱有誤或為空");
            }
            foreach (var item in dictPros)
            {
                PropertyInfo proInfo = item.Value;

                //取出 要修改的值
                object newValue = proInfo.GetValue(model, null);

                //批量設定 要修改 物件的 屬性
                foreach (T oModel in listModifing)
                {
                    //為 要修改的物件 的 要修改的屬性 設定新的值
                    proInfo.SetValue(oModel, newValue, null);
                }
            }

        }

        #endregion UPDATE

        #region SELECT

        /// <summary>
        /// 根據主鍵查詢
        /// </summary>
        /// <param name="id"></param>
        /// <returns></returns>
        public T FindById(dynamic id)
        {
            return Db.Set<T>().Find(id);
        }

        /// <summary>
        /// 獲取預設一條資料,沒有則為NULL
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public T FirstOrDefault(Expression<Func<T, bool>> whereLambda = null)
        {
            if (whereLambda == null)
            {
                return Db.Set<T>().FirstOrDefault();
            }
            return Db.Set<T>().FirstOrDefault(whereLambda);
        }

        /// <summary>
        /// 獲取全部資料
        /// </summary>
        /// <returns></returns>
        public List<T> GetAll(string ordering = null)
        {
            return ordering == null
                ? Db.Set<T>().ToList()
                : Db.Set<T>().OrderBy(ordering).ToList();
        }

        /// <summary>
        /// 帶條件查詢獲取資料
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <param name="ordering"></param>
        /// <returns></returns>
        public List<T> GetAll(Expression<Func<T, bool>> whereLambda, string ordering = null)
        {
            var iQueryable = Db.Set<T>().Where(whereLambda);
            return ordering == null
                ? iQueryable.ToList()
                : iQueryable.OrderBy(ordering).ToList();
        }

        /// <summary>
        /// 帶條件查詢獲取資料
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public IQueryable<T> GetAllIQueryable(Expression<Func<T, bool>> whereLambda = null)
        {
            return whereLambda == null ? Db.Set<T>() : Db.Set<T>().Where(whereLambda);
        }

        /// <summary>
        /// 獲取數量
        /// </summary>
        /// <param name="whereLambd"></param>
        /// <returns></returns>
        public int GetCount(Expression<Func<T, bool>> whereLambd = null)
        {
            return whereLambd == null ? Db.Set<T>().Count() : Db.Set<T>().Where(whereLambd).Count();
        }

        /// <summary>
        /// 判斷物件是否存在
        /// </summary>
        /// <param name="whereLambd"></param>
        /// <returns></returns>
        public bool Any(Expression<Func<T, bool>> whereLambd)
        {
            return Db.Set<T>().Where(whereLambd).Any();
        }

        /// <summary>
        /// 分頁查詢
        /// </summary>
        /// <param name="pageIndex">當前頁碼</param>
        /// <param name="pageSize">每頁大小</param>
        /// <param name="rows">總條數</param>
        /// <param name="orderBy">排序條件(一定要有)</param>
        /// <param name="whereLambda">查詢新增(可有,可無)</param>
        /// <param name="isOrder">是否是Order排序</param>
        /// <returns></returns>
        public List<T> Page<TKey>(int pageIndex, int pageSize, out int rows, Expression<Func<T, TKey>> orderBy, Expression<Func<T, bool>> whereLambda = null, bool isOrder = true)
        {
            IQueryable<T> data = isOrder ?
                Db.Set<T>().OrderBy(orderBy) :
                Db.Set<T>().OrderByDescending(orderBy);

            if (whereLambda != null)
            {
                data = data.Where(whereLambda);
            }
            rows = data.Count();
            return data.PageBy((pageIndex - 1) * pageSize, pageSize).ToList();
        }

        /// <summary>
        /// 分頁查詢
        /// </summary>
        /// <param name="pageIndex">當前頁碼</param>
        /// <param name="pageSize">每頁大小</param>
        /// <param name="rows">總條數</param>
        /// <param name="ordering">排序條件(一定要有)</param>
        /// <param name="whereLambda">查詢新增(可有,可無)</param>
        /// <returns></returns>
        public List<T> Page(int pageIndex, int pageSize, out int rows, string ordering, Expression<Func<T, bool>> whereLambda = null)
        {
            // 分頁 一定注意: Skip 之前一定要 OrderBy
            var data = Db.Set<T>().OrderBy(ordering);
            if (whereLambda != null)
            {
                data = data.Where(whereLambda);
            }
            rows = data.Count();
            return data.PageBy((pageIndex - 1) * pageSize, pageSize).ToList();
        }

        /// <summary>
        /// 查詢轉換
        /// </summary>
        /// <typeparam name="TDto"></typeparam>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public List<TDto> Select<TDto>(Expression<Func<T, bool>> whereLambda)
        {
            return Db.Set<T>().Where(whereLambda).Select<TDto>().ToList();
        }

        #endregion SELECT

        #region ORTHER

        /// <summary>
        /// 執行儲存過程或自定義sql語句--返回集合
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public List<T> Query(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            return Query<T>(sql, parms, cmdType);
        }

        /// <summary>
        /// 提交儲存
        /// </summary>
        /// <returns></returns>
        public int SaveChanges()
        {
            return Db.SaveChanges();
        }

        /// <summary>
        /// 回滾
        /// </summary>
        public void RollBackChanges()
        {
            var items = Db.ChangeTracker.Entries().ToList();
            items.ForEach(o => o.State = EntityState.Unchanged);
        }

        #endregion ORTHER

    }
DbBase

擴充套件類,實現讀寫分離

  上面的通用類是比較基礎簡單通用的,適合於單庫讀寫操作。對於EF實現讀寫分離,之前網上找過類似的參考文章,很多人文章都是使用 DbCommandInterceptor攔截器 來實現,具體的做法是通過攔截到sql語句,然後根據具體條件去判斷是走主庫還是從庫。這種做法不是不行,只是個人感覺不是很好擴充套件,而且要在攔截器裡面做限制判斷。

  其實說白了EF本身就是一個讀寫分離的orm。用過EF的人知道,EF提供訪問資料庫的是 DbContext 這個物件,所以想實現讀寫分離的就很簡單了,只要在程式中使用兩個不同的DbContext物件,一個負責讀,一個負責寫就好了。

  所以在上面提供的通用封裝類中稍微做下修改,修改如下DbContextFactory中獲取DbContext的方法,實現一個讀的DbContext和一個寫的DbContext物件的獲取。

  這裡要注意下,對於讀的DbContext來說,要做下設定
  1.使用 Database.SetInitializer(new NullDatabaseInitializer<ReadDbContext>()); 改變ef初始化策略,因為一般我們使用讀寫分離的話從庫都是同步於主庫的,所以不使用ef的自動建立資料庫策略。
  2.重寫 SaveChanges 方法,對應從庫來說,只提供讀取的功能,所以防止誤操作,這裡禁用掉SaveChanges方法,一般需要使用從讀的儲存方法,就對外丟擲異常。

  程式碼如下:

支援讀寫分離的 DbContextFactory 類

  public class DbContextFactory
    {
        public DbContext GetWriteDbContext()
        {
            string key = typeof(DbContextFactory).Name + "WriteDbContext";
            DbContext dbContext = CallContext.GetData(key) as DbContext;
            if (dbContext == null)
            {
                dbContext = new WriteDbContext();
                CallContext.SetData(key, dbContext);
            }
            return dbContext;
        }

        public DbContext GetReadDbContext()
        {
            string key = typeof(DbContextFactory).Name + "ReadDbContext";
            DbContext dbContext = CallContext.GetData(key) as DbContext;
            if (dbContext == null)
            {
                dbContext = new ReadDbContext();
                CallContext.SetData(key, dbContext);
            }
            return dbContext;
        }
    }

   對應的 DbBase 類也做下修改,主要將上面的Db物件改作 MasterDb  SlaveDb 物件,並且把上面的讀寫方法坐下調整,修改後如下:

支援讀寫分離的 DbBase類

public class DbBase
    {
        //是否讀寫分離(可以配置在配置檔案中)
        private static readonly bool IsReadWriteSeparation = true;

        #region EF上下文物件(主庫)

        protected DbContext MasterDb => _masterDb.Value;
        private readonly Lazy<DbContext> _masterDb = new Lazy<DbContext>(() => new DbContextFactory().GetWriteDbContext());

        #endregion EF上下文物件(主庫)

        #region EF上下文物件(從庫)

        protected DbContext SlaveDb => IsReadWriteSeparation ? _slaveDb.Value : _masterDb.Value;
        private readonly Lazy<DbContext> _slaveDb = new Lazy<DbContext>(() => new DbContextFactory().GetReadDbContext());

        #endregion EF上下文物件(從庫)

        #region 自定義其他方法

        /// <summary>
        /// 執行儲存過程或自定義sql語句--返回集合(自定義返回型別)
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public List<TModel> Query<TModel>(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            //儲存過程(exec getActionUrlId @name,@ID)
            if (cmdType == CommandType.StoredProcedure)
            {
                StringBuilder paraNames = new StringBuilder();
                foreach (var sqlPara in parms)
                {
                    paraNames.Append($" @{sqlPara},");
                }
                sql = paraNames.Length > 0 ? $"exec {sql} {paraNames.ToString().Trim(',')}" : $"exec {sql} ";
            }
            var list = SlaveDb.Database.SqlQuery<TModel>(sql, parms.ToArray());
            var enityList = list.ToList();
            return enityList;
        }

        /// <summary>
        /// 自定義語句和儲存過程的增刪改--返回影響的行數
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public int Execute(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            //儲存過程(exec getActionUrlId @name,@ID)
            if (cmdType == CommandType.StoredProcedure)
            {
                StringBuilder paraNames = new StringBuilder();
                foreach (var sqlPara in parms)
                {
                    paraNames.Append($" @{sqlPara},");
                }
                sql = paraNames.Length > 0 ?
                    $"exec {sql} {paraNames.ToString().Trim(',')}" :
                    $"exec {sql} ";
            }
            int ret = MasterDb.Database.ExecuteSqlCommand(sql, parms.ToArray());
            return ret;
        }

        #endregion 自定義其他方法
    }

    /// <summary>
    /// mssql資料庫 資料層 父類
    /// </summary>
    /// <typeparam name="T"></typeparam>
    public class DbBase<T> : DbBase where T : class, new()
    {
        #region INSERT

        /// <summary>
        /// 新增 實體
        /// </summary>
        /// <param name="model"></param>
        /// <returns></returns>
        public void Insert(T model)
        {
            MasterDb.Set<T>().Add(model);
        }

        /// <summary>
        /// 普通批量插入
        /// </summary>
        /// <param name="datas"></param>
        public void InsertRange(List<T> datas)
        {
            MasterDb.Set<T>().AddRange(datas);
        }

        #endregion INSERT

        #region DELETE

        /// <summary>
        /// 根據模型刪除
        /// </summary>
        /// <param name="model">包含要刪除id的物件</param>
        /// <returns></returns>
        public void Delete(T model)
        {
            MasterDb.Set<T>().Attach(model);
            MasterDb.Set<T>().Remove(model);
        }

        /// <summary>
        /// 刪除
        /// </summary>
        /// <param name="whereLambda"></param>
        public void Delete(Expression<Func<T, bool>> whereLambda)
        {
            MasterDb.Set<T>().Where(whereLambda).Delete();
        }

        #endregion DELETE

        #region UPDATE

        /// <summary>
        /// 單個物件指定列修改
        /// </summary>
        /// <param name="model">要修改的實體物件</param>
        /// <param name="proNames">要修改的 屬性 名稱</param>
        /// <param name="isProUpdate"></param>
        /// <returns></returns>
        public void Update(T model, List<string> proNames, bool isProUpdate = true)
        {
            //將 物件 新增到 EF中
            MasterDb.Set<T>().Attach(model);
            var setEntry = ((IObjectContextAdapter)MasterDb).ObjectContext.ObjectStateManager.GetObjectStateEntry(model);
            //指定列修改
            if (isProUpdate)
            {
                foreach (string proName in proNames)
                {
                    setEntry.SetModifiedProperty(proName);
                }
            }
            //忽略類修改
            else
            {
                Type t = typeof(T);
                List<PropertyInfo> proInfos = t.GetProperties(BindingFlags.Instance | BindingFlags.Public).ToList();
                foreach (var item in proInfos)
                {
                    string proName = item.Name;
                    if (proNames.Contains(proName))
                    {
                        continue;
                    }
                    setEntry.SetModifiedProperty(proName);
                }
            }
        }

        /// <summary>
        /// 單個物件修改
        /// </summary>
        /// <param name="model"></param>
        /// <returns></returns>
        public void Update(T model)
        {
            DbEntityEntry entry = MasterDb.Entry<T>(model);
            MasterDb.Set<T>().Attach(model);
            entry.State = EntityState.Modified;
        }

        /// <summary>
        /// 批量修改
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <param name="updateExpression"></param>
        public void Update(Expression<Func<T, bool>> whereLambda, Expression<Func<T, T>> updateExpression)
        {
            MasterDb.Set<T>().Where(whereLambda).Update(updateExpression);
        }

        /// <summary>
        /// 批量修改
        /// </summary>
        /// <param name="models"></param>
        /// <returns></returns>
        public void UpdateAll(List<T> models)
        {
            foreach (var model in models)
            {
                DbEntityEntry entry = MasterDb.Entry(model);
                entry.State = EntityState.Modified;
            }
        }

        /// <summary>
        /// 批量統一修改
        /// </summary>
        /// <param name="model">要修改的實體物件</param>
        /// <param name="whereLambda">查詢條件</param>
        /// <param name="modifiedProNames"></param>
        /// <returns></returns>
        public void Update(T model, Expression<Func<T, bool>> whereLambda, params string[] modifiedProNames)
        {
            //查詢要修改的資料
            List<T> listModifing = MasterDb.Set<T>().Where(whereLambda).ToList();
            Type t = typeof(T);
            List<PropertyInfo> proInfos = t.GetProperties(BindingFlags.Instance | BindingFlags.Public).ToList();
            Dictionary<string, PropertyInfo> dictPros = new Dictionary<string, PropertyInfo>();
            proInfos.ForEach(p =>
            {
                if (modifiedProNames.Contains(p.Name))
                {
                    dictPros.Add(p.Name, p);
                }
            });
            if (dictPros.Count <= 0)
            {
                throw new Exception("指定修改的欄位名稱有誤或為空");
            }
            foreach (var item in dictPros)
            {
                PropertyInfo proInfo = item.Value;

                //取出 要修改的值
                object newValue = proInfo.GetValue(model, null);

                //批量設定 要修改 物件的 屬性
                foreach (T oModel in listModifing)
                {
                    //為 要修改的物件 的 要修改的屬性 設定新的值
                    proInfo.SetValue(oModel, newValue, null);
                }
            }
        }

        #endregion UPDATE

        #region SELECT

        /// <summary>
        /// 根據主鍵查詢
        /// </summary>
        /// <param name="id"></param>
        /// <returns></returns>
        public T FindById(dynamic id)
        {
            return SlaveDb.Set<T>().Find(id);
        }

        /// <summary>
        /// 獲取預設一條資料,沒有則為NULL
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public T FirstOrDefault(Expression<Func<T, bool>> whereLambda = null)
        {
            if (whereLambda == null)
            {
                return SlaveDb.Set<T>().FirstOrDefault();
            }
            return SlaveDb.Set<T>().FirstOrDefault(whereLambda);
        }

        /// <summary>
        /// 獲取全部資料
        /// </summary>
        /// <returns></returns>
        public List<T> GetAll(string ordering = null)
        {
            return ordering == null
                ? SlaveDb.Set<T>().ToList()
                : SlaveDb.Set<T>().OrderBy(ordering).ToList();
        }

        /// <summary>
        /// 帶條件查詢獲取資料
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <param name="ordering"></param>
        /// <returns></returns>
        public List<T> GetAll(Expression<Func<T, bool>> whereLambda, string ordering = null)
        {
            var iQueryable = SlaveDb.Set<T>().Where(whereLambda);
            return ordering == null
                ? iQueryable.ToList()
                : iQueryable.OrderBy(ordering).ToList();
        }

        /// <summary>
        /// 帶條件查詢獲取資料
        /// </summary>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public IQueryable<T> GetAllIQueryable(Expression<Func<T, bool>> whereLambda = null)
        {
            return whereLambda == null ? SlaveDb.Set<T>() : SlaveDb.Set<T>().Where(whereLambda);
        }

        /// <summary>
        /// 獲取數量
        /// </summary>
        /// <param name="whereLambd"></param>
        /// <returns></returns>
        public int GetCount(Expression<Func<T, bool>> whereLambd = null)
        {
            return whereLambd == null ? SlaveDb.Set<T>().Count() : SlaveDb.Set<T>().Where(whereLambd).Count();
        }

        /// <summary>
        /// 判斷物件是否存在
        /// </summary>
        /// <param name="whereLambd"></param>
        /// <returns></returns>
        public bool Any(Expression<Func<T, bool>> whereLambd)
        {
            return SlaveDb.Set<T>().Where(whereLambd).Any();
        }

        /// <summary>
        /// 分頁查詢
        /// </summary>
        /// <param name="pageIndex">當前頁碼</param>
        /// <param name="pageSize">每頁大小</param>
        /// <param name="rows">總條數</param>
        /// <param name="orderBy">排序條件(一定要有)</param>
        /// <param name="whereLambda">查詢新增(可有,可無)</param>
        /// <param name="isOrder">是否是Order排序</param>
        /// <returns></returns>
        public List<T> Page<TKey>(int pageIndex, int pageSize, out int rows, Expression<Func<T, TKey>> orderBy, Expression<Func<T, bool>> whereLambda = null, bool isOrder = true)
        {
            IQueryable<T> data = isOrder ?
                SlaveDb.Set<T>().OrderBy(orderBy) :
                SlaveDb.Set<T>().OrderByDescending(orderBy);

            if (whereLambda != null)
            {
                data = data.Where(whereLambda);
            }
            rows = data.Count();
            return data.PageBy((pageIndex - 1) * pageSize, pageSize).ToList();
        }

        /// <summary>
        /// 分頁查詢
        /// </summary>
        /// <param name="pageIndex">當前頁碼</param>
        /// <param name="pageSize">每頁大小</param>
        /// <param name="rows">總條數</param>
        /// <param name="ordering">排序條件(一定要有)</param>
        /// <param name="whereLambda">查詢新增(可有,可無)</param>
        /// <returns></returns>
        public List<T> Page(int pageIndex, int pageSize, out int rows, string ordering, Expression<Func<T, bool>> whereLambda = null)
        {
            // 分頁 一定注意: Skip 之前一定要 OrderBy
            var data = SlaveDb.Set<T>().OrderBy(ordering);
            if (whereLambda != null)
            {
                data = data.Where(whereLambda);
            }
            rows = data.Count();
            return data.PageBy((pageIndex - 1) * pageSize, pageSize).ToList();
        }

        /// <summary>
        /// 查詢轉換
        /// </summary>
        /// <typeparam name="TDto"></typeparam>
        /// <param name="whereLambda"></param>
        /// <returns></returns>
        public List<TDto> Select<TDto>(Expression<Func<T, bool>> whereLambda)
        {
            return SlaveDb.Set<T>().Where(whereLambda).Select<TDto>().ToList();
        }

        #endregion SELECT

        #region ORTHER

        /// <summary>
        /// 執行儲存過程或自定義sql語句--返回集合
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="parms"></param>
        /// <param name="cmdType"></param>
        /// <returns></returns>
        public List<T> Query(string sql, List<SqlParameter> parms, CommandType cmdType = CommandType.Text)
        {
            return Query<T>(sql, parms, cmdType);
        }

        /// <summary>
        /// 提交儲存
        /// </summary>
        /// <returns></returns>
        public int SaveChanges()
        {
            return MasterDb.SaveChanges();
        }

        /// <summary>
        /// 回滾
        /// </summary>
        public void RollBackChanges()
        {
            var items = MasterDb.ChangeTracker.Entries().ToList();
            items.ForEach(o => o.State = EntityState.Unchanged);
        }

        #endregion ORTHER
    }
DbBase

  這樣簡單的讀寫分離就實現了,實現邏輯也比較清晰,方便擴充套件。

進一步改造,實現多從庫讀取

  一般做讀寫分離,都會做一主多從,特別對讀取量比較大的專案,這樣多庫讀取就能減輕讀庫的壓力。所以對於上面的方法,做下改造。
  上面可以看到,主庫和從庫都是通過 DbContextFactory 這個類來獲取的,在GetReadDbContext 方法中每次都是獲取 ReadDbContext 這個物件。那麼對於多個從庫的情況下,每次讀取到底要去哪個庫讀取資料呢?這裡就是一個演算法規則的問題了,或者說是策略吧,如果使用過nginx的朋友就知道,nginx本身內部在實現負載均衡的時候提供了多種策略,比如輪詢,加權輪詢,ip_hash等策略。其實上面獲取同一個ReadDbContext 的方法也算一種策略,叫單一策略,每次都獲取單一的物件。

  多從庫的情況下,我們簡單的來實現另一種獲取策略,隨機策略,每次都隨機獲取到一個從庫的物件,這種是最簡單的策略,當然,正式使用的話大家可以發揮自己的創造力,寫出多了的演算法策略。

首先,定義一個策略介面,方便策略的擴充套件和切換,程式碼如下:

IReadDbStrategy 介面

 /// <summary>
  /// 從資料庫獲取策略介面
  /// </summary>
  public interface IReadDbStrategy
  {
      /// <summary>
      /// 獲取讀庫
      /// </summary>
      /// <returns></returns>
      DbContext GetDbContext();
  }

單從庫情況下,定義一個單一策略,程式碼如下:

單一策略

   /// <summary>
   /// 單一策略
   /// </summary>
   public class SingleStrategy : IReadDbStrategy
   {
       public DbContext GetDbContext()
       {
           return new ReadDbContext();
       }
   }

多從庫情況下,定義一個隨機策略,程式碼如下:

隨機策略

    /// <summary>
    /// 隨機策略
    /// </summary>
    public class RandomStrategy : IReadDbStrategy
    {
        //所有讀庫型別
        public static List<Type> DbTypes;
 
        static RandomStrategy()
        {
            LoadDbs();
        }
 
        //載入所有的讀庫型別
        static void LoadDbs()
        {
            DbTypes = new List<Type>();
            var assembly = Assembly.GetExecutingAssembly();
            var types = assembly.GetTypes();
            foreach (var type in types)
            {
                if (type.BaseType == typeof(BaseReadDbContext))
                {
                    DbTypes.Add(type);
                }
            }
        }
 
        public DbContext GetDbContext()
        {
            int randomIndex = new Random().Next(0, DbTypes.Count);
            var dbType = DbTypes[randomIndex];
            var dbContext = Activator.CreateInstance(dbType) as DbContext;
            return dbContext;
        }
    }

  這樣,所有從庫我們都基於策略去獲取,擴充套件也比較方便。修改下 DbContextFactory 類的 GetReadDbContext 方法,通過策略介面來獲取,程式碼如下:

支援一主多從的 DbContextFactory 類

  public class DbContextFactory
    {
        //todo:這裡可以自己通過注入的方式來實現,就會更加靈活
        private static readonly IReadDbStrategy ReadDbStrategy = new RandomStrategy();
        public DbContext GetWriteDbContext()
        {
            string key = typeof(DbContextFactory).Name + "WriteDbContext";
            DbContext dbContext = CallContext.GetData(key) as DbContext;
            if (dbContext == null)
            {
                dbContext = new WriteDbContext();
                CallContext.SetData(key, dbContext);
            }
            return dbContext;
        }

        public DbContext GetReadDbContext()
        {
            string key = typeof(DbContextFactory).Name + "ReadDbContext";
            DbContext dbContext = CallContext.GetData(key) as DbContext;
            if (dbContext == null)
            {
                dbContext = ReadDbStrategy.GetDbContext();
                CallContext.SetData(key, dbContext);
            }
            return dbContext;
        }
    }

  這樣簡單的