一步步实现自己的ORM（四）

通过前3章文章，大致对ORM有一定的了解，但也存在效率低下（大量用了反射）和重复代码，今天我们要对ORM进行优化。

具体流程如下：

我们优化的第一个就是减少反射调用，我的思路是定义一个Mapping，把表名、字段名信息缓存起来，EntityMapping 表示实体类信息对应数据库中的table，MemberMapping表示实体类的属性对应数据库中的Column。

EntityMapping代码：

    class EntityMapping

    {

        /// <summary>

        /// 对应实体类类型

        /// </summary>

        public Type EntityType { get; internal set; }

        /// <summary>

        /// 表名

        /// </summary>

        public string TableName { get; set; }

        /// <summary>

        /// 实体属性

        /// </summary>

        public List<MemberMapping> Members;

        /// <summary>

        /// 主键

        /// </summary>

        public List<MemberMapping> PrimaryKey { get; set; }

    }

MemberMapping代码：

    class MemberMapping

    {

        /// <summary>

        /// 列名

        /// </summary>

        public string ColumnName { get; internal set; }

        /// <summary>

        /// 属性名

        /// </summary>

        public PropertyInfo Member { get; set; }

        /// <summary>

        /// 是否为主键

        /// </summary>

        public bool IsPrimaryKey { get; set; }

        /// <summary>

        /// 是否为数据库自动生成

        /// </summary>

        public bool IsDbGenerated { get; set; }

    }

有Mapping后，我们现在要做的是把实体类里的信息存到Mapping类中，我在这里定义了一个AttributeMapping，里面方法如下：

    class AttributeMapping

    {

        private static Dictionary<Type, EntityMapping> entityMappings = new Dictionary<Type, EntityMapping>();

        public static EntityMapping Get<T>()

        public static EntityMapping Get(Type type)

        private static EntityMapping CreateMapping(Type type)

    }

其中Get方法就是根据实体类返回Mapping信息，而CreateMapping则是创建Mapping，具体代码见附件下载。

Mapping信息有了，我们需要做的是把重构之前的EntityHelper类，而这个类里有大量的链接数据库、操作数据库代码，我们把这些信息抽离到一个单独类，就好比我们常用的DbHelper类，而这个类我命名为DbProvider。

    class DbProvider

    {

        private IDbConnection conntion;

        public DbProvider(IDbConnection conntion)

        public virtual int ExecuteNonQuery(string sql, Dictionary<string, object> parameters)

        public virtual IDataReader ExecuteReader(string sql, Dictionary<string, object> parameters)

    }

EntityHelper类还有大量的SQL生成语句，我们为了简化EntityHelper类把SQL生成的代码放到单独的DbSqlBuilder类，为啥要这么做呢？除了简化代码外，还有就是为了适应不同的Db。

    class DbSqlBuilder

    {

        public string BuildInsertSql(EntityMapping entityMapping, List<string> columnNames)

        public string BuildUpdateSql(EntityMapping entityMapping, List<string> updateColumns,List<string> whereColumns)

        public string BuildDeleteSql(EntityMapping entityMapping, List<string> whereColumns)

　　　　　public string BuildSelectSql(EntityMapping entityMapping,string strWhere, string orderBy)

    }

为什么参数里要传递List<string>呢？因为我们拼SQL语句要用到列名的，修改后的EntityHelper（我在这里将名字改成DbContext）。

将数据库操作和SQL生成代码放到单独类里，回头再看下DbContext代码，比原来的简洁了很多。

    class DbContext

    {

        private DbProvider dbProvider;

        private DbSqlBuilder sqlBuilder ;

        public DbContext(string connectionString)

        {

            SqlConnection conn = new SqlConnection(connectionString);

            dbProvider = new DbProvider(conn);

            this.sqlBuilder = new DbSqlBuilder();

        }

        public int Insert<T>(object entity)

        {

            var entityMapping = AttributeMapping.Get<T>();

            //将Entity转换成Dictionary

            var parameters = DynamicMethodBuilder.ConvertFromObject(entity);

            var sql = sqlBuilder.BuildInsertSql(entityMapping, parameters.Keys.ToList());

            return dbProvider.ExecuteNonQuery(sql, parameters);

        }

        public int Update<T>(T entity)

        {

            var entityMapping = AttributeMapping.Get<T>();

            //将Entity转换成Dictionary

            var parameters = DynamicMethodBuilder.ConvertFromObject(entity);

            var columns = entityMapping.Members.Where(m => m.IsDbGenerated == false && m.IsPrimaryKey == false).Select(c => c.ColumnName).ToArray();

            var updateColumns = new Dictionary<string, object>();

            var whereColumns = new Dictionary<string, object>();

            foreach (var item in parameters)

            {

                if (columns.Contains(item.Key))

                    updateColumns.Add(item.Key, item.Value);

                if (entityMapping.PrimaryKey.All(m => m.ColumnName == item.Key))

                    whereColumns.Add(item.Key, item.Value);

            }

            var sql = sqlBuilder.BuildUpdateSql(entityMapping, updateColumns.Keys.ToList(), whereColumns.Keys.ToList());

            return dbProvider.ExecuteNonQuery(sql, parameters);

        }

        public int DeleteByKey<T>(params object[] values)

        {

            var entityMapping = AttributeMapping.Get<T>();

            if (values.Length != entityMapping.PrimaryKey.Count)

                throw new ArgumentException("参数个数和主键数不一致");

            var parameters = new Dictionary<string, object>(StringComparer.OrdinalIgnoreCase);

            for (int i = ; i < entityMapping.PrimaryKey.Count; i++)

            {

                parameters.Add(entityMapping.PrimaryKey[i].ColumnName, values[i]);

            }

            var sql = sqlBuilder.BuildDeleteSql(entityMapping, parameters.Keys.ToList());

            return dbProvider.ExecuteNonQuery(sql, parameters);

        }

        public T Get<T>(object[] values)

        {

            var entityMapping = AttributeMapping.Get<T>();

            if (values.Length != entityMapping.PrimaryKey.Count)

                throw new ArgumentException("参数个数和主键数不一致");

            var parameters = new Dictionary<string, object>(StringComparer.OrdinalIgnoreCase);

            StringBuilder where = new StringBuilder();

            for (int i = ; i < values.Length; i++)

            {

                if (i > ) //考虑到有多个主键

                    where.Append(" AND ");

                where.Append(entityMapping.PrimaryKey[i].ColumnName).Append("=").Append("@p").Append(i);

                /*参数*/

                parameters.Add("@p" + i, values[i]);

            }

            var sql = this.sqlBuilder.BuildSelectSql(entityMapping, where.ToString(), string.Empty);

            return GetEntityList<T>(sql, parameters).FirstOrDefault();

        }

        public List<T> GetList<T>(string where, string orderBy)

        {

            var entityMapping = AttributeMapping.Get<T>();

            var sql = this.sqlBuilder.BuildSelectSql(entityMapping, where.ToString(), orderBy);

            return GetEntityList<T>(sql, null);

        }

        public List<T> GetEntityList<T>(string sql, Dictionary<string, object> parameters)

        {

            var reader = dbProvider.ExecuteReader(sql, parameters);

            List<T> list = new List<T>();

            var type = typeof(T);

            var properties = type.GetProperties();

            while (reader.Read())

            {

                var user = Activator.CreateInstance(type);

                for (int i = ; i < properties.Length; i++)

                {

                    var pi = properties[i];

                    if (reader[pi.Name] != null) //等同于 if (reader["UserId"] != null)这样的语句

                        pi.SetValue(user, reader[pi.Name], null); //等同于 user.UserId = (int)reader["UserId"];

                }

                list.Add((T)user);

            }

            return list;

        }

    }

本章结束，下一章要将DataReader转换成实体类的代码优化。

下载地址：http://files.cnblogs.com/files/sobaby/ORM04.zip

巴特西

一步步实现自己的ORM（四）

最新文章

热门文章