使用 AutoMapper 映射 IDataReader、DataSet、DataTable 到实体类

痛定思痛。 2024-02-17 15:49 48阅读 0赞

AutoMapper是一个.NET的对象映射工具。

项目地址：[https://github.com/AutoMapper/AutoMapper][https_github.com_AutoMapper_AutoMapper]。

帮助文档：[https://github.com/AutoMapper/AutoMapper/wiki][https_github.com_AutoMapper_AutoMapper_wiki]

主要用途

领域对象与DTO之间的转换、数据库查询结果映射至实体对象。

这里主要说下使用 AutoMapper 将 IDataReader、DataSet、DataTable 转为实体的方法。

依赖文件：AutoMapper.dll、AutoMapper.Net4.dll 两个 dll 文件

AutoMapper.Net4.dll 这个文件可以下载代码自己编译，该文件是封装了对 IDataReader 的支持

简单说下 AutoMapper 使用

第一步：声明映射约定

Mapper.CreateMap<IDataReader, menuModel>();//将 IDataReader 映射为 menuModel 实体

第二步：转换实体

//IDataReader =>menuModel

using (IDataReader dr = .....)  
\{  
var list = Mapper.Map<List<menuModel>>(dr);  
dr.Close();  
\}

以下是自己封装的一个 AutoMapper 帮助类

使用方法：

第一步：将要转换的模型在 静态构造函数中 约定下

/// <summary>  
/// 注册 Mapper 转换规则约定  
/// </summary>  
static void Configure()  
\{  
Mapper.CreateMap<IDataReader, menuModel>();

...

Mapper.CreateMap<IDataReader, xxxxxModel>();  
\}

第二步：在项目中应用程序集，使用已经写好的扩展方法

using Utitity.AutoMapper

IDataReader dr = ...;  
var list1 = dr.GetEntity<List<menuModel>>();  
DataSet ds = ...;  
var list2 = ds.GetEntity<List<menuModel>>();  
DataTable dt = ...;  
var list3 = dt.GetEntity<List<menuModel>>();

MapperHelper 源码

using AutoMapper;
    using System.Data;
    
    namespace Utitity.AutoMapper
    {
        /// <summary>
        /// 实体映射帮助类
        /// </summary>
        public static class MapperHelper
        {
            #region 配置映射规则
            /// <summary>
            /// 确保映射配置只注册一次
            /// </summary>
            static MapperHelper()
            {
                Configure();
            }
            /// <summary>
            /// 注册 Mapper 转换规则约定
            /// </summary>
            static void Configure()
            {
                Mapper.CreateMap<IDataReader, xxxxModel>();//只需要约定基础类型，不要要写成List<xxxxModel>这种形式
            }
    
            #endregion
    
    
            #region 实体映射扩展方法
            /// <summary>
            /// 将 IDataReader 转为实体对象
            /// </summary>
            /// <typeparam name="T"></typeparam>
            /// <param name="dr"></param>
            /// <returns></returns>
            public static T GetEntity<T>(this IDataReader dr)
            {
                return Mapper.Map<T>(dr);
            }
            /// <summary>
            /// 将 DataSet 转为实体对象
            /// </summary>
            /// <typeparam name="T"></typeparam>
            /// <param name="ds"></param>
            /// <returns></returns>
            public static T GetEntity<T>(this DataSet ds)
            {
                if (ds == null || ds.Tables.Count == 0 || ds.Tables[0].Rows.Count == 0)
                    return default(T);
                var dr = ds.Tables[0].CreateDataReader();
                return Mapper.Map<T>(dr);
            }
            /// <summary>
            /// 将 DataTable 转为实体对象
            /// </summary>
            /// <typeparam name="T"></typeparam>
            /// <param name="dt"></param>
            /// <returns></returns>
            public static T GetEntity<T>(this DataTable dt)
            {
                if (dt == null || dt.Rows.Count == 0)
                    return default(T);
                var dr = dt.CreateDataReader();
                return Mapper.Map<T>(dr);
            }
    
            #endregion
        }
    }

[https_github.com_AutoMapper_AutoMapper]: https://github.com/AutoMapper/AutoMapper
[https_github.com_AutoMapper_AutoMapper_wiki]: https://github.com/AutoMapper/AutoMapper/wiki