优化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>();
+            if (obj == null) return default(T);
-            return Mapper.Map<TResult>(source);
+            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);
         }
     }
 }

OpenAuth.App/LoginApp.cs

@@ -14,14 +14,17 @@ namespace OpenAuth.App
         private IUserRepository _repository;
         private IModuleRepository _moduleRepository;
         private IRelevanceRepository _relevanceRepository;
+        private IRepository<ModuleElement> _moduleElementRepository; 
 
         public LoginApp(IUserRepository repository,
             IModuleRepository moduleRepository,
-            IRelevanceRepository relevanceRepository)
+            IRelevanceRepository relevanceRepository,
+            IRepository<ModuleElement>  moduleElementRepository )
         {
             _repository = repository;
             _moduleRepository = moduleRepository;
             _relevanceRepository = relevanceRepository;
+            _moduleElementRepository = moduleElementRepository;
         }
 
         public LoginUserVM Login(string userName, string password)
@@ -48,7 +51,7 @@ namespace OpenAuth.App
                         (u.FirstId == user.Id && u.Key == "UserModule") ||
                         (u.Key == "RoleModule" && userRoleIds.Contains(u.FirstId))).Select(u =>u.SecondId).ToList();
             //得出最终用户拥有的模块
-            loginVM.Modules = _moduleRepository.Find(u => moduleIds.Contains(u.Id)).ToList();
+            loginVM.Modules = _moduleRepository.Find(u => moduleIds.Contains(u.Id)).MapToList<ModuleView>();
             
            return loginVM;
         }
@@ -65,7 +68,7 @@ namespace OpenAuth.App
                     Name = "开发者账号"
                 }
             };
-            loginUser.Modules = _moduleRepository.Find(null).ToList();
+            loginUser.Modules = _moduleRepository.Find(null).MapToList<ModuleView>();
             return loginUser;
         }
     }

OpenAuth.App/ViewModel/LoginUserVM.cs

@@ -26,12 +26,7 @@ namespace OpenAuth.App.ViewModel
         /// <summary>
         /// 用户可以访问到的模块（包括所属角色与自己的所有模块）
         /// </summary>
-        public List<Module> Modules { get; set; }
+        public List<ModuleView> Modules { get; set; }
-
-        /// <summary>
-        /// 用户可以访问到的模块中的元素
-        /// </summary>
-        public List<ModuleElement> ModuleElements { get; set; } 
     }
 
 }

OpenAuth.App/ViewModel/ModuleView.cs

@@ -1,6 +1,6 @@
-using System.Collections.Generic;
+using Infrastructure;
-using Infrastructure;
 using OpenAuth.Domain;
+using System.Collections.Generic;
 
 namespace OpenAuth.App.ViewModel
 {
@@ -39,17 +39,21 @@ namespace OpenAuth.App.ViewModel
         /// <summary>
         /// 子节点
         /// </summary>
-        public List<ModuleView> Childern  = new List<ModuleView>();
+        public List<ModuleView> Childern = new List<ModuleView>();
+
+        /// <summary>
+        /// 模块中的元素
+        /// </summary>
+        public List<ModuleElement> Elements = new List<ModuleElement>();
 
         public static implicit operator ModuleView(Module module)
         {
-            return AutoMapperExt.ConvertTo<Module, ModuleView>(module);
+            return module.MapTo<ModuleView>();
         }
 
         public static implicit operator Module(ModuleView view)
         {
-            return AutoMapperExt.ConvertTo<ModuleView, Module>(view);
+            return view.MapTo<Module>();
         }
-
     }
 }

OpenAuth.App/ViewModel/RoleVM.cs

@@ -52,12 +52,12 @@ namespace OpenAuth.App.ViewModel
 
         public static implicit operator RoleVM(Role role)
         {
-            return AutoMapperExt.ConvertTo<Role,RoleVM>(role);
+            return role.MapTo<RoleVM>();
         }
 
         public static implicit operator Role(RoleVM rolevm)
         {
-            return AutoMapperExt.ConvertTo<RoleVM, Role>(rolevm);
+            return rolevm.MapTo<Role>();
         }
 
     }

OpenAuth.App/ViewModel/UserView.cs

@@ -71,12 +71,12 @@ namespace OpenAuth.App.ViewModel
 
         public static implicit operator UserView(User user)
         {
-            return AutoMapperExt.ConvertTo<User, UserView>(user);
+            return user.MapTo<UserView>();
         }
 
         public static implicit operator User(UserView view)
         {
-            return AutoMapperExt.ConvertTo<UserView, User>(view);
+            return view.MapTo<User>();
         }
     }
 }

OpenAuth.Mvc/Views/Home/Index.cshtml

@@ -1,4 +1,4 @@
-@model List<OpenAuth.Domain.Module>
+@model List<OpenAuth.App.ViewModel.ModuleView>
 <!DOCTYPE html>
 <html lang="zh">
 <head>

OpenAuth.UnitTest/TestLogin.cs

@@ -3,6 +3,7 @@ using System.Text;
 using System.Collections.Generic;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using OpenAuth.App;
+using OpenAuth.Domain;
 using OpenAuth.Repository;
 
 namespace OpenAuth.UnitTest
@@ -18,7 +19,11 @@ namespace OpenAuth.UnitTest
         [TestMethod]
         public void Test()
         {
-            var login = new LoginApp(new UserRepository(), new ModuleRepository(),  new RelevanceRepository());
+            var login = new LoginApp(new UserRepository(), 
+                new ModuleRepository(), 
+                new RelevanceRepository(),
+                new BaseRepository<ModuleElement>()
+                );
             var user = login.Login("admin", "admin");
             foreach (var module in user.Modules)
             {