lsm/OpenAuth.Net
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/OpenAuth.Net.git synced 2025-04-05 17:38:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化AutoMapper代码

Browse Source
This commit is contained in:
yubaolee 2015-12-03 23:39:27 +08:00
parent 000df6d1d3
commit 0674b108df
8 changed files with 146 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
Infrastructure/AutoMapperExt.cs
View File

@ -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);
}
}
}
}

9
OpenAuth.App/LoginApp.cs
View File

@ -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;
}
}

69
OpenAuth.App/ViewModel/LoginUserVM.cs
View File

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

18
OpenAuth.App/ViewModel/ModuleView.cs
View File

@ -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>();
}
}
}
}

4
OpenAuth.App/ViewModel/RoleVM.cs
View File

@ -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>();
}
}

4
OpenAuth.App/ViewModel/UserView.cs
View File

@ -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>();
}
}
}

2
OpenAuth.Mvc/Views/Home/Index.cshtml
View File

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

63
OpenAuth.UnitTest/TestLogin.cs
View File

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