一步步实现自己的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