AutoMapper的一些简单应用和扩展

最近在项目中用到了AutoMapper，在我项目中只用到了AutoMapper的一些简单功能。如对象转换、将sql查询出来的Datatable转换为实体集合。将实体集合转换为Datatable，我看了半天，好像AutoMapper不支持...，于是就自己实现了一个功能。
下面是我将用到的部分功能做了扩展，主要还是用的扩展方法，方便调用。

 using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
 using AutoMapper;
 using System.Collections;
 using System.Data;
 using System.Reflection;
 
 
namespace LY.AutoMapper
 {
 public static class AutoMapperExtension
 {
 /// 

 /// 实体对象转换
 /// 
 /// 
 /// 
 /// 
 public static TDestination MapTo(this object o)
 {
 if (o == null)
 throw new ArgumentNullException();
 
 
        Mapper.CreateMap(o.GetType(), typeof(TDestination));

        return Mapper.Map(o); ;
    }


    /// 

    /// 集合转换
    /// 
    /// 
    /// 
    /// 
    public static List MapTo(this IEnumerable o)
    {
        if (o == null)
            throw new ArgumentNullException();


        foreach (var item in o)
        {
            Mapper.CreateMap(item.GetType(), typeof(TDestination));

            break;
        }
        return Mapper.Map>(o);
    }


    /// 
  
    /// 将 DataTable 转为实体对象  
    /// 
  
    ///   
    ///   
    ///   
    public static List MapTo(this DataTable dt)
    {
        if (dt == null || dt.Rows.Count == 0)
            return default(List);

        Mapper.CreateMap();
        return Mapper.Map>(dt.CreateDataReader());
    }


    /// 

    /// 将List转换为Datatable
    /// 
    /// 
    /// 
    /// 
    public static DataTable MapToTable(this IEnumerable list)
    {
        if (list == null)
            return default(DataTable);

        //创建属性的集合
        List pList = new List();
        //获得反射的入口
        Type type = typeof(T);
        DataTable dt = new DataTable();
        //把所有的public属性加入到集合 并添加DataTable的列
        Array.ForEach(type.GetProperties(), p => { pList.Add(p); dt.Columns.Add(p.Name, p.PropertyType); });
        foreach (var item in list)
        {
            //创建一个DataRow实例
            DataRow row = dt.NewRow();
            //给row 赋值
            pList.ForEach(p => row[p.Name] = p.GetValue(item, null));
            //加入到DataTable
            dt.Rows.Add(row);
        }
        return dt;
    }

}
 
 
}
 

将List转换为DataTable，因为我没看到AutoMapper支持（或许研究的不深...），所以这里用了反射的实现。这样的话，这个扩展类就支持从Datatable to List，List to Datatable了。

下面我们进行测试。
新建两个类，模拟Entity与Model。

 namespace LY.AutoMapper.Test
 {
 public class StudentEntity
 {
 public string Name { get; set; }
 
 
    public int Age { get; set; }
}
 
 
}
 

 namespace LY.AutoMapper.Test
 {
 public class StudentModel
 {
 
 
    public string Name { get; set; }

    public int Age { get; set; }
}
 
 
}
 

下面，是使用方法。

 private void btnTest_Click(object sender, EventArgs e)
 {
 StudentEntity stuEntity = new StudentEntity() { Name = "ly", Age = 29 };
 
 
        //实体转换
        StudentModel stuModel = stuEntity.MapTo();

        //集合转换
        List strEntityList = new List()
          {
              new StudentEntity(){Name="aa",Age=15}
          };

        List stuModelList = strEntityList.MapTo();

        #region Get DataTable
        DataTable tb = new DataTable();
        tb.Columns.Add("Name");
        tb.Columns.Add("Age", typeof(Int32));

        DataRow row1 = tb.NewRow();
        row1[0] = "ly";
        row1[1] = 27;

        DataRow row2 = tb.NewRow();
        row2[0] = "yy";
        row2[1] = 26;

        tb.Rows.Add(row1);
        tb.Rows.Add(row2);
        #endregion

        //Datatable转换为List
        List stuEntitys = tb.MapTo();

        //List转换为Datatable
        DataTable tbTest = stuEntitys.MapToTable();
 
    }
 
 

当然，如果项目中使用了EF，利用AutoMapper可以很方便的将数据持久层转换为我们定义的Model，节省了很多代码量。