公司新來了一個(gè)同事，把權(quán)限系統(tǒng)設(shè)計(jì)的爐火純青！

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-security</artifactId></dependency>

package com.example.demo.web;
import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RestController;
@RestController@RequestMapping("/test")public class Test {    @RequestMapping("/test")    public String test(){return "test";    }}

用戶名：user
密碼 984cccf2-ba82-468e-a404-7d32123d0f9c

spring:  security:    user:      name: ming      password: 123456      roles: admin

package com.example.demo.config;
import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;import org.springframework.security.crypto.password.NoOpPasswordEncoder;import org.springframework.security.crypto.password.PasswordEncoder;
@Configurationpublic class MyWebSecurityConfig extends WebSecurityConfigurerAdapter {    @Bean    PasswordEncoder passwordEncoder(){        return NoOpPasswordEncoder.getInstance();    }
    @Override    protected void configure(AuthenticationManagerBuilder auth) throws Exception {        auth.inMemoryAuthentication()                .withUser("admin").password("123").roles("admin");    }}

package com.ming.demo.interceptor;
import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.http.HttpMethod;import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity;import org.springframework.security.config.annotation.web.builders.HttpSecurity;import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;import org.springframework.security.crypto.password.PasswordEncoder;import org.springframework.security.web.authentication.rememberme.TokenBasedRememberMeServices;
@Configuration@EnableWebSecuritypublic class SecurityConfig  extends WebSecurityConfigurerAdapter {    //基于內(nèi)存的用戶存儲(chǔ)    @Override    public void configure(AuthenticationManagerBuilder auth) throws Exception {        auth.inMemoryAuthentication()                .withUser("itguang").password("123456").roles("USER").and()                .withUser("admin").password("{noop}" + "123456").roles("ADMIN");    }
    //請求攔截    @Override    protected void configure(HttpSecurity http) throws Exception {        http.authorizeRequests()                .anyRequest().permitAll()                .and()                .formLogin()                .permitAll()                .and()                .logout()                .permitAll();    }}

 <!-- https://mvnrepository.com/artifact/io.jsonwebtoken/jjwt -->        <dependency>            <groupId>io.jsonwebtoken</groupId>            <artifactId>jjwt</artifactId>            <version>0.9.1</version>        </dependency>        <!-- https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-security -->        <dependency>            <groupId>org.springframework.boot</groupId>            <artifactId>spring-boot-starter-security</artifactId>            <version>2.3.1.RELEASE</version>        </dependency>

package com.example.demo;
import org.springframework.security.core.GrantedAuthority;import org.springframework.security.core.userdetails.UserDetails;
import java.util.Collection;
public class JwtUser implements UserDetails {    private String username;    private String password;    private Integer state;    private Collection<? extends GrantedAuthority> authorities;    public JwtUser(){
    }
    public JwtUser(String username, String password, Integer state,  Collection<? extends GrantedAuthority> authorities){        this.username = username;        this.password = password;        this.state = state;        this.authorities = authorities;    }
    @Override    public Collection<? extends GrantedAuthority> getAuthorities() {        return authorities;    }
    @Override    public String getPassword() {        return this.password;    }
    @Override    public String getUsername() {        return this.username;    }
    @Override    public boolean isAccountNonExpired() {        return true;    }
    @Override    public boolean isAccountNonLocked() {        return true;    }
    @Override    public boolean isCredentialsNonExpired() {        return true;    }
    @Override    public boolean isEnabled() {        return true;    }}

package com.example.demo;
import io.jsonwebtoken.Claims;import io.jsonwebtoken.Jwts;import io.jsonwebtoken.SignatureAlgorithm;import org.springframework.security.core.userdetails.UserDetails;
import java.io.Serializable;import java.util.Date;import java.util.HashMap;import java.util.Map;
public class JwtTokenUtil implements Serializable {    private String secret;    private Long expiration;    private String header;
    private String generateToken(Map<String, Object> claims) {        Date expirationDate = new Date(System.currentTimeMillis() + expiration);        return Jwts.builder().setClaims(claims).setExpiration(expirationDate).signWith(SignatureAlgorithm.HS512, secret).compact();    }    private Claims getClaimsFromToken(String token) {        Claims claims;        try {            claims = Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();        } catch (Exception e) {            claims = null;        }        return claims;    }
    public String generateToken(UserDetails userDetails) {        Map<String, Object> claims = new HashMap<>(2);        claims.put("sub", userDetails.getUsername());        claims.put("created", new Date());        return generateToken(claims);    }
    public String getUsernameFromToken(String token) {        String username;        try {            Claims claims = getClaimsFromToken(token);            username = claims.getSubject();        } catch (Exception e) {            username = null;        }        return username;    }
    public Boolean isTokenExpired(String token) {        try {            Claims claims = getClaimsFromToken(token);            Date expiration = claims.getExpiration();            return expiration.before(new Date());        } catch (Exception e) {            return false;        }    }    public String refreshToken(String token) {        String refreshedToken;        try {            Claims claims = getClaimsFromToken(token);            claims.put("created", new Date());            refreshedToken = generateToken(claims);        } catch (Exception e) {            refreshedToken = null;        }        return refreshedToken;    }
    public Boolean validateToken(String token, UserDetails userDetails) {        JwtUser user = (JwtUser) userDetails;        String username = getUsernameFromToken(token);        return (username.equals(user.getUsername()) && !isTokenExpired(token));    }}

package com.example.demo;
import org.apache.commons.lang.StringUtils;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;import org.springframework.security.core.context.SecurityContextHolder;import org.springframework.security.core.userdetails.UserDetails;import org.springframework.security.core.userdetails.UserDetailsService;import org.springframework.security.web.authentication.WebAuthenticationDetailsSource;import org.springframework.stereotype.Component;import org.springframework.web.filter.OncePerRequestFilter;
import javax.servlet.FilterChain;import javax.servlet.ServletException;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import java.io.IOException;
@Componentpublic class JwtAuthenticationTokenFilter  extends OncePerRequestFilter {    @Autowired    private UserDetailsService userDetailsService;    @Autowired    private JwtTokenUtil jwtTokenUtil;
    @Override    protected void doFilterInternal(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, FilterChain filterChain) throws ServletException, IOException {        String authHeader = httpServletRequest.getHeader(jwtTokenUtil.getHeader());        if (authHeader != null && StringUtils.isNotEmpty(authHeader)) {            String username = jwtTokenUtil.getUsernameFromToken(authHeader);            if (username != null && SecurityContextHolder.getContext().getAuthentication() == null) {                UserDetails userDetails = this.userDetailsService.loadUserByUsername(username);                if (jwtTokenUtil.validateToken(authHeader, userDetails)) {                    UsernamePasswordAuthenticationToken authentication  =                    new UsernamePasswordAuthenticationToken(userDetails,null,userDetails.getAuthorities());                    authentication.setDetails(new WebAuthenticationDetailsSource().buildDetails(httpServletRequest));                    SecurityContextHolder.getContext().setAuthentication(authentication);
                }            }        }        filterChain.doFilter(httpServletRequest, httpServletResponse);    }}

package com.example.demo;
import org.springframework.beans.factory.annotation.Autowired;import org.springframework.security.core.authority.SimpleGrantedAuthority;import org.springframework.security.core.userdetails.User;import org.springframework.security.core.userdetails.UserDetails;import org.springframework.security.core.userdetails.UserDetailsService;import org.springframework.security.core.userdetails.UsernameNotFoundException;import org.springframework.stereotype.Service;
import javax.management.relation.Role;import java.util.List;
@Servicepublic class JwtUserDetailsServiceImpl  implements UserDetailsService {    @Autowired    private UserMapper userMapper;
    @Override    public UserDetails loadUserByUsername(String s) throws UsernameNotFoundException {        User user = userMapper.selectByUserName(s);        if (user == null) {            throw new UsernameNotFoundException(String.format("'%s'.這個(gè)用戶不存在", s));
        }        List<SimpleGrantedAuthority> collect = user.getRoles().stream().map(Role::getRolename).map(SimpleGrantedAuthority::new).collect(Collectors.toList());        return new JwtUser(user.getUsername(), user.getPassword(), user.getState(), collect);    }}

@Servicepublic class UserServiceImpl  implements UserService {    @Autowired    private UserMapper userMapper;
    @Autowired    private AuthenticationManager authenticationManager;
    @Autowired    private UserDetailsService userDetailsService;
    @Autowired    private JwtTokenUtil jwtTokenUtil;
    public User findByUsername(String username) {        User user = userMapper.selectByUserName(username);        return user;    }
    public RetResult login(String username, String password) throws AuthenticationException {        UsernamePasswordAuthenticationToken upToken = new UsernamePasswordAuthenticationToken(username, password);        final Authentication authentication = authenticationManager.authenticate(upToken);        SecurityContextHolder.getContext().setAuthentication(authentication);        UserDetails userDetails = userDetailsService.loadUserByUsername(username);        return new RetResult(RetCode.SUCCESS.getCode(),jwtTokenUtil.generateToken(userDetails));    }}

@EnableGlobalMethodSecurity(prePostEnabled = true)@EnableWebSecuritypublic class WebSecurity extends WebSecurityConfigurerAdapter {    @Autowired    private UserDetailsService userDetailsService;    @Autowired    private JwtAuthenticationTokenFilter jwtAuthenticationTokenFilter;
    @Autowired    public void configureAuthentication(AuthenticationManagerBuilder authenticationManagerBuilder) throws Exception {        authenticationManagerBuilder.userDetailsService(this.userDetailsService).passwordEncoder(passwordEncoder());
    }
    @Bean(name = BeanIds.AUTHENTICATION_MANAGER)
    @Override    public AuthenticationManager authenticationManagerBean() throws Exception {        return super.authenticationManagerBean();    }        @Bean    public PasswordEncoder passwordEncoder() {        return new BCryptPasswordEncoder();    }
    @Override    protected void configure(HttpSecurity http) throws Exception {        http.csrf().disable().sessionManagement().sessionCreationPolicy(SessionCreationPolicy.STATELESS)                .and().authorizeRequests()                .antMatchers(HttpMethod.OPTIONS, "/**").permitAll()                .antMatchers("/auth/**").permitAll()                .anyRequest().authenticated()                .and().headers().cacheControl();
        http.addFilterBefore(jwtAuthenticationTokenFilter, UsernamePasswordAuthenticationFilter.class);
        ExpressionUrlAuthorizationConfigurer<HttpSecurity>.ExpressionInterceptUrlRegistry registry = http.authorizeRequests();
        registry.requestMatchers(CorsUtils::isPreFlightRequest).permitAll();
    }
    @Bean    public CorsFilter corsFilter() {        final UrlBasedCorsConfigurationSource urlBasedCorsConfigurationSource = new UrlBasedCorsConfigurationSource();        final CorsConfiguration cors = new CorsConfiguration();        cors.setAllowCredentials(true);        cors.addAllowedOrigin("*");        cors.addAllowedHeader("*");        cors.addAllowedMethod("*");        urlBasedCorsConfigurationSource.registerCorsConfiguration("/**", cors);        return new CorsFilter(urlBasedCorsConfigurationSource);
    }}

public class CustomAuthenticationFilter extends UsernamePasswordAuthenticationFilter {
    @Override    public Authentication attemptAuthentication(HttpServletRequest request, HttpServletResponse response) throws AuthenticationException {
        //attempt Authentication when Content-Type is json        if(request.getContentType().equals(MediaType.APPLICATION_JSON_UTF8_VALUE)                ||request.getContentType().equals(MediaType.APPLICATION_JSON_VALUE)){
            //use jackson to deserialize json            ObjectMapper mapper = new ObjectMapper();            UsernamePasswordAuthenticationToken authRequest = null;            try (InputStream is = request.getInputStream()){                AuthenticationBean authenticationBean = mapper.readValue(is,AuthenticationBean.class);                authRequest = new UsernamePasswordAuthenticationToken(                        authenticationBean.getUsername(), authenticationBean.getPassword());            }catch (IOException e) {                e.printStackTrace();                authRequest = new UsernamePasswordAuthenticationToken(                        "", "");            }finally {                setDetails(request, authRequest);                return this.getAuthenticationManager().authenticate(authRequest);            }        }
        //transmit it to UsernamePasswordAuthenticationFilter        else {            return super.attemptAuthentication(request, response);        }    }}

@Overrideprotected void configure(HttpSecurity http) throws Exception {    http            .cors().and()            .antMatcher("/**").authorizeRequests()            .antMatchers("/", "/login**").permitAll()            .anyRequest().authenticated()            //這里必須要寫formLogin()，不然原有的UsernamePasswordAuthenticationFilter不會(huì)出現(xiàn)，也就無法配置我們重新的UsernamePasswordAuthenticationFilter            .and().formLogin().loginPage("/")            .and().csrf().disable();
    //用重寫的Filter替換掉原有的UsernamePasswordAuthenticationFilter    http.addFilterAt(customAuthenticationFilter(),    UsernamePasswordAuthenticationFilter.class);}
//注冊自定義的UsernamePasswordAuthenticationFilter@BeanCustomAuthenticationFilter customAuthenticationFilter() throws Exception {    CustomAuthenticationFilter filter = new CustomAuthenticationFilter();    filter.setAuthenticationSuccessHandler(new SuccessHandler());    filter.setAuthenticationFailureHandler(new FailureHandler());    filter.setFilterProcessesUrl("/login/self");
    //這句很關(guān)鍵，重用WebSecurityConfigurerAdapter配置的AuthenticationManager，不然要自己組裝AuthenticationManager    filter.setAuthenticationManager(authenticationManagerBean());    return filter;}

/**     * 密碼加密     */    @Bean    public BCryptPasswordEncoder passwordEncoder(){        return new BCryptPasswordEncoder();    }

@Service@Transactionalpublic class UserServiceImpl implements UserService {
    @Resource    private UserRepository userRepository;
    @Resource    private BCryptPasswordEncoder bCryptPasswordEncoder;  //注入bcryct加密    @Override    public User add(User user) {        user.setPassword(bCryptPasswordEncoder.encode(user.getPassword())); //對密碼進(jìn)行加密        User user2 = userRepository.save(user);        return user2;    }    @Override    public ResultInfo login(User user) {        ResultInfo resultInfo=new ResultInfo();        User user2 = userRepository.findByName(user.getName());        if (user2==null) {            resultInfo.setCode("-1");            resultInfo.setMessage("用戶名不存在");            return resultInfo;        }
        //判斷密碼是否正確        if (!bCryptPasswordEncoder.matches(user.getPassword(),user2.getPassword())) {            resultInfo.setCode("-1");            resultInfo.setMessage("密碼不正確");            return resultInfo;        }        resultInfo.setMessage("登錄成功");        return resultInfo;    }}

@Configurablepublic class WebSecurityConfig extends WebSecurityConfigurerAdapter {    @Autowired    private UserService userService;    // service 層注入
    @Bean    PasswordEncoder passwordEncoder(){        return new BCryptPasswordEncoder();    }
    @Override    protected void configure(AuthenticationManagerBuilder auth) throws Exception {        // 參數(shù)傳入Service，進(jìn)行驗(yàn)證        auth.userDetailsService(userService);    }
    @Override    protected void configure(HttpSecurity http) throws Exception {        http.authorizeRequests()                .antMatchers("/admin/**").hasRole("admin")                .anyRequest().authenticated()                .and()                .formLogin()                .loginProcessingUrl("/login").permitAll()                .and()                .csrf().disable();    }}