黑马程序员技术交流社区

标题: Spring-security User扩展 [打印本页]

---

作者: Socuss    时间: 2019-9-12 12:24
标题: Spring-security User扩展
Spring security中的系统的User只有username,password,enabled等几个属性,对于实际开发中,这些属性用的概率非常低
比如: 用户查看自己的头像,手机号,修改昵称等等, 在spring-security默认的User对象中并不存在这些自定义的字段.怎么办呢?
这就需要我们自己扩展User类了.

思路:
我自己定义的MyUser 继承了spring-security中的User类,对其进行扩展.

1. package com.qingcheng.controller;
   
2. 
   
3. import org.springframework.security.core.GrantedAuthority;
   
4. import org.springframework.security.core.userdetails.User;
   
5. 
   
6. import java.util.Collection;
   
7. 
   
8. public class MyUser extends User {
   
9.     private Integer id;
   
10.     private String loginName;
    
11.     private String image;
    
12. 
    
13.     public MyUser(String username, String password, Collection<? extends GrantedAuthority> authorities) {
    
14.         super(username, password, authaorities);
    
15.     }
    
16. 
    
17.     public MyUser(String username, String password, boolean enabled, boolean accountNonExpired, boolean credentialsNonExpired, boolean accountNonLocked, Collection<? extends GrantedAuthority> authorities) {
    
18.         super(username, password, enabled, accountNonExpired, credentialsNonExpired, accountNonLocked, authorities);
    
19.     }
    
20. 
    
21.     public Integer getId() {
    
22.         return id;
    
23.     }
    
24. 
    
25.     public void setId(Integer id) {
    
26.         this.id = id;
    
27.     }
    
28. 
    
29.     public String getLoginName() {
    
30.         return loginName;
    
31.     }
    
32. 
    
33.     public void setLoginName(String loginName) {
    
34.         this.loginName = loginName;
    
35.     }
    
36. 
    
37.     public String getImage() {
    
38.         return image;
    
39.     }
    
40. 
    
41.     public void setImage(String image) {
    
42.         this.image = image;
    
43.     }
    
44. }
    

复制代码

然后在UserDetailService的实现类UserDetailServiceImpl中,将匹配到的用户信息, set到自定义的MyUser对象中,再返回,如下:

1. package com.qingcheng.controller;
   
2. 
   
3. import com.alibaba.dubbo.config.annotation.Reference;
   
4. import com.qingcheng.pojo.system.Admin;
   
5. import com.qingcheng.service.system.AdminService;
   
6. import org.springframework.security.core.GrantedAuthority;
   
7. import org.springframework.security.core.authority.SimpleGrantedAuthority;
   
8. import org.springframework.security.core.userdetails.UserDetails;
   
9. import org.springframework.security.core.userdetails.UserDetailsService;
   
10. import org.springframework.security.core.userdetails.UsernameNotFoundException;
    
11. 
    
12. import java.util.ArrayList;
    
13. import java.util.HashMap;
    
14. import java.util.List;
    
15. import java.util.Map;
    
16. 
    
17. public class UserDetailsServiceImpl implements UserDetailsService {
    
18. 
    
19.     @Reference
    
20.     private AdminService adminService;
    
21. 
    
22.     @Override
    
23.     public UserDetails loadUserByUsername(String s) throws UsernameNotFoundException {
    
24. 
    
25.         Map  searchMap = new HashMap();
    
26.         searchMap.put("loginName",s);
    
27.         searchMap.put("status","1");
    
28.         List<Admin> list = adminService.findList(searchMap);
    
29.         if(list.size() == 0){
    
30.             return null;
    
31.         }
    
32. 
    
33.         ArrayList<GrantedAuthority> grantedAuthorities = new ArrayList<>();
    
34.         grantedAuthorities.add(new SimpleGrantedAuthority("ROLE_ADMIN"));
    
35.         Admin admin = list.get(0);
    
36. //        return new User(s,list.get(0).getPassword(),grantedAuthorities);
    
37.         MyUser myUser = new MyUser(s, admin.getPassword(), grantedAuthorities);
    
38.         myUser.setId(admin.getId());
    
39.         myUser.setLoginName(admin.getLoginName());
    
40.         myUser.setImage(admin.getImage());
    
41.         return myUser;
    
42.     }
    
43. }
    

复制代码

第一次发帖, 也是java小白, 不要注重于代码,写得比较low.谢谢哈!

---

欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2