优化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

@@ -1,37 +1,32 @@
-// ***********************************************************************
+// ***********************************************************************
-// Assembly         : OpenAuth.App
+// Assembly         : OpenAuth.App
-// Author           : Yubao Li
+// Author           : Yubao Li
-// Created          : 12-01-2015
+// Created          : 12-01-2015
-//
+//
-// Last Modified By : Yubao Li
+// Last Modified By : Yubao Li
-// Last Modified On : 12-01-2015
+// Last Modified On : 12-01-2015
-// ***********************************************************************
+// ***********************************************************************
-// <copyright file="LoginUserVM.cs" company="">
+// <copyright file="LoginUserVM.cs" company="">
-//     Copyright (c) . All rights reserved.
+//     Copyright (c) . All rights reserved.
-// </copyright>
+// </copyright>
-// <summary>登陆视图模型</summary>
+// <summary>登陆视图模型</summary>
-// ***********************************************************************
+// ***********************************************************************
-
+
-using System.Collections.Generic;
+using System.Collections.Generic;
-using OpenAuth.Domain;
+using OpenAuth.Domain;
-
+
-namespace OpenAuth.App.ViewModel
+namespace OpenAuth.App.ViewModel
-{
+{
-    /// <summary>
+    /// <summary>
-    /// 登陆用户视图模型
+    /// 登陆用户视图模型
-    /// </summary>
+    /// </summary>
-    public class LoginUserVM 
+    public class LoginUserVM 
-    {
+    {
-        public User User { get; set; }
+        public User User { get; set; }
-        /// <summary>
+        /// <summary>
-        /// 用户可以访问到的模块（包括所属角色与自己的所有模块）
+        /// 用户可以访问到的模块（包括所属角色与自己的所有模块）
-        /// </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

@@ -1,29 +1,34 @@
-using System;
+using System;
-using System.Text;
+using System.Text;
-using System.Collections.Generic;
+using System.Collections.Generic;
-using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
-using OpenAuth.App;
+using OpenAuth.App;
-using OpenAuth.Repository;
+using OpenAuth.Domain;
-
+using OpenAuth.Repository;
-namespace OpenAuth.UnitTest
+
-{
+namespace OpenAuth.UnitTest
-    /// <summary>
+{
-    /// TestLogin 的摘要说明
+    /// <summary>
-    /// </summary>
+    /// TestLogin 的摘要说明
-    [TestClass]
+    /// </summary>
-    public class TestLogin
+    [TestClass]
-    {
+    public class TestLogin
-        
+    {
-
+        
-        [TestMethod]
+
-        public void Test()
+        [TestMethod]
-        {
+        public void Test()
-            var login = new LoginApp(new UserRepository(), new ModuleRepository(),  new RelevanceRepository());
+        {
-            var user = login.Login("admin", "admin");
+            var login = new LoginApp(new UserRepository(), 
-            foreach (var module in user.Modules)
+                new ModuleRepository(), 
-            {
+                new RelevanceRepository(),
-                Console.WriteLine(module.Id +"\t" + module.Name);
+                new BaseRepository<ModuleElement>()
-            }
+                );
-        }
+            var user = login.Login("admin", "admin");
-    }
+            foreach (var module in user.Modules)
-}
+            {
+                Console.WriteLine(module.Id +"\t" + module.Name);
+            }
+        }
+    }
+}