一步步实现自己的ORM(五)

上一张优化了ORM的INSERT、UPDATE、DELETE,但将数据库里的值填充到实体类这块还没优化。另外有博友在网上咨询说你这个都是查询所有字段的,而他的需求是按需查询字段,不是一次性取出来所有字段的,在这里我请这位朋友耐心等待,这个会在后面章节提到的。这次我们先优化datareader->entity,将数据库里的值Mapping到实体类里,我们常用的有两种办法第一种是采用EMIT方式,第二种是采用Expression tree表达式,这两种方案在性能上差异不大,但写法上第二种较第一种比较容易,我在这里采用Expression tree表达式来完成实体类转换操作。

让我们先来看下Expression 表达式如何将DataReader转换成Object:

            IDataReader reader = null;
            Expression<Func<IDataReader, User>> expr = (r) => new User()
            {
                UserId = r.GetInt32(0),
                CreatedTime = r.GetDateTime(1),
                Email = r.GetString(2),
            };

            var func = expr.Compile();
            func(reader);

在new User的时候采用对象初始化方式给属性赋值,而在实际项目中我们会遇到更复杂的,比如还要判断是否为DbNull,要不然转换会出错,在这里我定义一个GetValue<T>(int i)方法,专门做取值操作。

    class DbFieldReader
    {
        private IDataReader reader;
        public DbFieldReader(IDataReader reader)
        {
            this.reader = reader;
        }

        public T GetValue<T>(int index)
        {
            object value = reader.GetValue(index);

            if (value == DBNull.Value)
                return default(T);

            return (T)reader.GetValue(index);
        }
    }

现在将代码修改下,

            IDataReader reader = null;
            DbFieldReader fr = new DbFieldReader(reader);
            Expression<Func<DbFieldReader, User>> expr = (r) => new User()
            {
                UserId = r.GetValue<int>(0),
                CreatedTime = r.GetValue<DateTime>(1),
                Email = r.GetValue<string>(2),
            };

            var func = expr.Compile();
            func(fr);

是不是简洁了很多?把判断代码转移到DbFieldReader类中处理,现在我们要将这个方法做成通用实体类转换,先分解下Expression表达式(如吐有错误,恳请大神指教)。

第一步定义参数,类型为DbFieldReader,参数名:r

ParameterExpression parExpr = Expression.Parameter(typeof(DbFieldReader), "r");

第二步,调用User类的构造函数,代码如下:

var newExpr = Expression.New(type.GetConstructors().First());

合并这2个表达式和运行结果:

Expression<Func<DbFieldReader, User>> expr = (Expression<Func<DbFieldReader, User>>)Expression.Lambda(newExpr, parExpr);

基本的雏形出来了,现在要做的就是给属性赋值,怎么来呢?先来个简单的,就是给固定值。

var userid = Expression.Bind(type.GetProperty("UserId"), Expression.Constant(1, typeof(int)));

这就相当于UserId = 1的赋值操作,继续合并表达式:

          var type = typeof(User);
            ParameterExpression parExpr = Expression.Parameter(typeof(DbFieldReader), "r");

            var newExpr = Expression.New(type.GetConstructors().First());

            var userid = Expression.Bind(type.GetProperty("UserId"), Expression.Constant(1, typeof(int)));
            var CreatedTime = Expression.Bind(type.GetProperty("CreatedTime"), Expression.Constant(DateTime.Now, typeof(DateTime)));

            var init = Expression.MemberInit(newExpr, userid, CreatedTime);
            Expression<Func<DbFieldReader, User>> expr = (Expression<Func<DbFieldReader, User>>)Expression.Lambda(init, parExpr);

常量赋值没问题后,下面就要来调用方法了。

            var method = typeof(DbFieldReader).GetMethods().Where(c => c.Name == "GetValue" && c.IsGenericMethod).First();
            var callExpr = Expression.Call(parExpr, method.MakeGenericMethod(typeof(int)), Expression.Constant(0));
       var userid = Expression.Bind(type.GetProperty("UserId"), callExpr);

这句话相当于 r.GetValue<int>(0);

先获取GetValue方法,然后调用MakeGenericMethod生成泛型方法,Expression.Constant(0)就是参数值。

完整版代码:

            IDataReader reader = null;
            DbFieldReader fr = new DbFieldReader(reader);

            var entityMapping = AttributeMapping.Get<User>();
            ParameterExpression parExpr = Expression.Parameter(typeof(DbFieldReader), "r");


            var newExpr = Expression.New(entityMapping.EntityType.GetConstructors().First());

            var method = typeof(DbFieldReader).GetMethods().Where(c => c.Name == "GetValue" && c.IsGenericMethod).First();

            List<MemberBinding> memberBindings = new List<MemberBinding>();
            int index = 0;
            foreach (var item in entityMapping.Members)
            {
                var callExpr = Expression.Call(parExpr, method.MakeGenericMethod(item.Member.PropertyType), Expression.Constant(index));
                var memberAssignment = Expression.Bind(item.Member, callExpr);
                memberBindings.Add(memberAssignment);
                index++;
            }

            var init = Expression.MemberInit(newExpr, memberBindings);
            Expression<Func<DbFieldReader, User>> expr = (Expression<Func<DbFieldReader, User>>)Expression.Lambda(init, parExpr);

当然我们还需要修改SQL语句,不能用SELECT * 语法,而是要用SELECT userid,email...from这样的语法

最终的代码代码下载:

http://files.cnblogs.com/files/sobaby/ORM05.zip

原文地址:https://www.cnblogs.com/sobaby/p/4380654.html