AutoMapper的一些简单应用和扩展
最近在项目中用到了AutoMapper,在我项目中只用到了AutoMapper的一些简单功能。如对象转换、将sql查询出来的Datatable转换为实体集合。将实体集合转换为Datatable,我看了半天,好像AutoMapper不支持...,于是就自己实现了一个功能。
下面是我将用到的部分功能做了扩展,主要还是用的扩展方法,方便调用。
<pre>
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
{
/// <summary>
/// 实体对象转换
/// </summary>
/// <typeparam name="TDestination"></typeparam>
/// <param name="o"></param>
/// <returns></returns>
public static TDestination MapTo<TDestination>(this object o)
{
if (o == null)
throw new ArgumentNullException();
Mapper.CreateMap(o.GetType(), typeof(TDestination));
return Mapper.Map<TDestination>(o); ;
}
/// <summary>
/// 集合转换
/// </summary>
/// <typeparam name="TDestination"></typeparam>
/// <param name="o"></param>
/// <returns></returns>
public static List<TDestination> MapTo<TDestination>(this IEnumerable o)
{
if (o == null)
throw new ArgumentNullException();
foreach (var item in o)
{
Mapper.CreateMap(item.GetType(), typeof(TDestination));
break;
}
return Mapper.Map<List<TDestination>>(o);
}
/// <summary>
/// 将 DataTable 转为实体对象
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="dt"></param>
/// <returns></returns>
public static List<T> MapTo<T>(this DataTable dt)
{
if (dt == null || dt.Rows.Count == 0)
return default(List<T>);
Mapper.CreateMap<IDataReader, T>();
return Mapper.Map<List<T>>(dt.CreateDataReader());
}
/// <summary>
/// 将List转换为Datatable
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="list"></param>
/// <returns></returns>
public static DataTable MapToTable<T>(this IEnumerable list)
{
if (list == null)
return default(DataTable);
//创建属性的集合
List<PropertyInfo> pList = new List<PropertyInfo>();
//获得反射的入口
Type type = typeof(T);
DataTable dt = new DataTable();
//把所有的public属性加入到集合 并添加DataTable的列
Array.ForEach<PropertyInfo>(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;
}
}
}
</pre>
将List转换为DataTable,因为我没看到AutoMapper支持(或许研究的不深...),所以这里用了反射的实现。这样的话,这个扩展类就支持从Datatable to List,List to Datatable了。
下面我们进行测试。
新建两个类,模拟Entity与Model。
<pre>
namespace LY.AutoMapper.Test
{
public class StudentEntity
{
public string Name { get; set; }
public int Age { get; set; }
}
}
</pre>
<pre>
namespace LY.AutoMapper.Test
{
public class StudentModel
{
public string Name { get; set; }
public int Age { get; set; }
}
}
</pre>
下面,是使用方法。
<pre>
private void btnTest_Click(object sender, EventArgs e)
{
StudentEntity stuEntity = new StudentEntity() { Name = "ly", Age = 29 };
//实体转换
StudentModel stuModel = stuEntity.MapTo<StudentModel>();
//集合转换
List<StudentEntity> strEntityList = new List<StudentEntity>()
{
new StudentEntity(){Name="aa",Age=15}
};
List<StudentModel> stuModelList = strEntityList.MapTo<StudentModel>();
#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<StudentEntity> stuEntitys = tb.MapTo<StudentEntity>();
//List转换为Datatable
DataTable tbTest = stuEntitys.MapToTable<StudentEntity>();
}
</pre>
当然,如果项目中使用了EF,利用AutoMapper可以很方便的将数据持久层转换为我们定义的Model,节省了很多代码量。