EF Core 5.0 使用过滤器进行查询拦截的方法和示例,以及一些备忘

1、查询拦截和结果替换(取消执行)的示例:

    /// <summary>
    /// 查询拦截器
    /// </summary>
    public class ShardingDbCommandInterceptor : Microsoft.EntityFrameworkCore.Diagnostics.DbCommandInterceptor
    {
        /// <summary>
        /// 在这里做拦截、判断、取消查询、赋值操作
        /// </summary>
        /// <param name="command"></param>
        /// <param name="eventData"></param>
        /// <param name="result"></param>
        /// <returns></returns>
        public override InterceptionResult<DbDataReader> ReaderExecuting(DbCommand command, CommandEventData eventData, InterceptionResult<DbDataReader> result)
        {
            Console.WriteLine(command.CommandText);

            //取参数列表
            foreach (System.Data.Common.DbParameter p in command.Parameters)
            {
                Console.WriteLine($"    参数名:{p.ParameterName},参数值:{p.Value},参数方向:{p.Direction}");
            }

            Console.WriteLine($"当前数据库名称:{command.Connection.Database},数据源为:{command.Connection.DataSource},服务器版本:{command.Connection.ServerVersion}");


            //定义一个Table,如果执行下面的代码,即拦截成功,自动被取消执行
            var dt = new System.Data.DataTable();
            dt.Columns.Add("Id", typeof(string));
            dt.Columns.Add("AccountId", typeof(Int64));
            var rs = dt.NewRow();
            rs["Id"] = Guid.NewGuid().ToString();
            rs["AccountId"] = 1234;
            dt.Rows.Add(rs);
            result = InterceptionResult<DbDataReader>.SuppressWithResult(dt.CreateDataReader());

            return result;
        }

        /// <summary>
        /// 执行完毕后会返回
        /// </summary>
        /// <param name="command"></param>
        /// <param name="eventData"></param>
        /// <param name="result"></param>
        /// <returns></returns>
        public override DbDataReader ReaderExecuted(DbCommand command, CommandExecutedEventData eventData, DbDataReader result)
        {
            return base.ReaderExecuted(command, eventData, result);
        }
    }

  2、注册过滤器的代码:

    /// <summary>
    /// 多库、分库、分表、Redis自动化 上下文对象
    /// </summary>
    public class DataContexts:DbContext
    {

        public DataContexts() : base()
        {
            
        }

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            optionsBuilder.UseSqlServer("Data Source=192.168.3.162;Initial Catalog=hmt_report;Persist Security Info=True;User ID=sa;Password=123").AddInterceptors(new ShardingDbCommandInterceptor());
        }

        /// <summary>
        /// 重写该方法,用于实现批量生成SQL进行批量操作,例如批量插入,以及做Redis的更新
        /// </summary>
        /// <returns></returns>
        public override int SaveChanges()
        {
            return base.SaveChanges();
        }

        /// <summary>
        /// 声明当前数据库上下文使用的全部模型和反射字段列表
        /// </summary>
        public static Dictionary<string, System.Reflection.PropertyInfo[]> modelProperties { get; set; } = new Dictionary<string, System.Reflection.PropertyInfo[]>();


        /// <summary>
        /// 表 AudienceAccountReport_DataOnDay
        /// </summary>
        public DbSet<ShardingEntityFramework.DataModels.AudienceAccountReport_DataOnDay> AudienceAccountReport_DataOnDay { get; set; }

    }

  EF Framework也是一样,有了该拦截器后,可以扩展为Redis自动化,只要识别出来查询语句中的目标表,以及键字段有具体值,就可以直接从redis里取,然后,读写分离也是基于拦截器来改变数据库连接字符串,就可以实现读写的分离,另外,要自动更新redis或者读写分离,则通过覆盖重写

SaveChanges 函数来实现,同时,分库分表也涉及到这个地方,实际上,特别是分表稍为麻烦一些,分表因为有多种不同规则,分表方法,核心难点是解析SQL里的条件,得出表名和对不同表的where字段及值,进行做实际表的替换,来实现自动化分表哦。

仅仅是思路。
posted @ 2021-09-07 09:39  soleds  阅读(643)  评论(0编辑  收藏  举报