优化AutoMapper代码

Infrastructure/AutoMapperExt.cs

@@ -13,15 +13,68 @@
 // ***********************************************************************
 
 using AutoMapper;
+using System.Collections;
+using System.Collections.Generic;
+using System.Data;
 
 namespace Infrastructure
 {
-    public class AutoMapperExt
+    public static class AutoMapperExt
     {
-        public static TResult ConvertTo<T, TResult>(T source)
+        /// <summary>
+        ///  类型映射
+        /// </summary>
+        public static T MapTo<T>(this object obj)
         {
-            var mapper = Mapper.CreateMap<T, TResult>();
-            return Mapper.Map<TResult>(source);
+            if (obj == null) return default(T);
+            Mapper.CreateMap(obj.GetType(), typeof(T));
+            return Mapper.Map<T>(obj);
+        }
+
+        /// <summary>
+        /// 集合列表类型映射
+        /// </summary>
+        public static List<TDestination> MapToList<TDestination>(this IEnumerable source)
+        {
+            foreach (var first in source)
+            {
+                var type = first.GetType();
+                Mapper.CreateMap(type, typeof(TDestination));
+                break;
+            }
+            return Mapper.Map<List<TDestination>>(source);
+        }
+
+        /// <summary>
+        /// 集合列表类型映射
+        /// </summary>
+        public static List<TDestination> MapToList<TSource, TDestination>(this IEnumerable<TSource> source)
+        {
+            //IEnumerable<T> 类型需要创建元素的映射
+            Mapper.CreateMap<TSource, TDestination>();
+            return Mapper.Map<List<TDestination>>(source);
+        }
+
+        /// <summary>
+        /// 类型映射
+        /// </summary>
+        public static TDestination MapTo<TSource, TDestination>(this TSource source, TDestination destination)
+            where TSource : class
+            where TDestination : class
+        {
+            if (source == null) return destination;
+            Mapper.CreateMap<TSource, TDestination>();
+            return Mapper.Map(source, destination);
+        }
+
+        /// <summary>
+        /// DataReader映射
+        /// </summary>
+        public static IEnumerable<T> DataReaderMapTo<T>(this IDataReader reader)
+        {
+            Mapper.Reset();
+            Mapper.CreateMap<IDataReader, IEnumerable<T>>();
+            return Mapper.Map<IDataReader, IEnumerable<T>>(reader);
         }
     }
-}
+}