spring boot 利用注解实现权限验证的实现代码

这里使用 aop 来实现权限验证

引入依赖

<dependency>
  <groupid>org.springframework.boot</groupid>
  <artifactid>spring-boot-starter-aop</artifactid>
</dependency>

定义注解

package com.lmxdawn.api.admin.annotation;

import java.lang.annotation.elementtype;
import java.lang.annotation.retention;
import java.lang.annotation.retentionpolicy;
import java.lang.annotation.target;

/**
 * 后台登录授权/权限验证的注解
 */
//此注解只能修饰方法
@target(elementtype.method)
//当前注解如何去保持
@retention(retentionpolicy.runtime)
public @interface authruleannotation {
  string value();
}

拦截实现登录和权限验证

package com.lmxdawn.api.admin.aspect;

import com.lmxdawn.api.admin.annotation.authruleannotation;
import com.lmxdawn.api.admin.enums.resultenum;
import com.lmxdawn.api.admin.exception.jsonexception;
import com.lmxdawn.api.admin.service.auth.authloginservice;
import com.lmxdawn.api.common.utils.jwtutils;
import io.jsonwebtoken.claims;
import lombok.extern.slf4j.slf4j;
import org.aspectj.lang.joinpoint;
import org.aspectj.lang.annotation.aspect;
import org.aspectj.lang.annotation.before;
import org.aspectj.lang.annotation.pointcut;
import org.aspectj.lang.reflect.methodsignature;
import org.springframework.stereotype.component;
import org.springframework.web.context.request.requestcontextholder;
import org.springframework.web.context.request.servletrequestattributes;

import javax.annotation.resource;
import javax.servlet.http.httpservletrequest;
import java.lang.reflect.method;
import java.util.list;

/**
 * 登录验证 aop
 */
@aspect
@component
@slf4j
public class authorizeaspect {

  @resource
  private authloginservice authloginservice;

  @pointcut("@annotation(com.lmxdawn.api.admin.annotation.authruleannotation)")
  public void adminloginverify() {
  }

  /**
   * 登录验证
   *
   * @param joinpoint
   */
  @before("adminloginverify()")
  public void doadminauthverify(joinpoint joinpoint) {

    servletrequestattributes attributes = (servletrequestattributes) requestcontextholder.getrequestattributes();
    if (attributes == null) {
      throw new jsonexception(resultenum.not_network);
    }
    httpservletrequest request = attributes.getrequest();

    string id = request.getheader("x-adminid");

    long adminid = long.valueof(id);

    string token = request.getheader("x-token");
    if (token == null) {
      throw new jsonexception(resultenum.login_verify_fall);
    }

    // 验证 token
    claims claims = jwtutils.parse(token);
    if (claims == null) {
      throw new jsonexception(resultenum.login_verify_fall);
    }
    long jwtadminid = long.valueof(claims.get("admin_id").tostring());
    if (adminid.compareto(jwtadminid) != 0) {
      throw new jsonexception(resultenum.login_verify_fall);
    }

    // 判断是否进行权限验证
    methodsignature signature = (methodsignature) joinpoint.getsignature();
    //从切面中获取当前方法
    method method = signature.getmethod();
    //得到了方,提取出他的注解
    authruleannotation action = method.getannotation(authruleannotation.class);
    // 进行权限验证
    authruleverify(action.value(), adminid);
  }

  /**
   * 权限验证
   *
   * @param authrule
   */
  private void authruleverify(string authrule, long adminid) {

    if (authrule != null && authrule.length() > 0) {

      list<string> authrules = authloginservice.listrulebyadminid(adminid);
      // admin 为最高权限
      for (string item : authrules) {
        if (item.equals("admin") || item.equals(authrule)) {
          return;
        }
      }
      throw new jsonexception(resultenum.auth_failed);
    }

  }

}

controller 中使用

使用 authruleannotation 注解, value 值就是在数据库里面定义的 权限规则名称

/**
 * 获取管理员列表
 */
@authruleannotation("admin/auth/admin/index")
@getmapping("/admin/auth/admin/index")
public resultvo index(@valid authadminqueryform authadminqueryform,
           bindingresult bindingresult) {

  if (bindingresult.haserrors()) {
    return resultvoutils.error(resultenum.param_verify_fall, bindingresult.getfielderror().getdefaultmessage());
  }

  if (authadminqueryform.getroleid() != null) {
    list<authroleadmin> authroleadmins = authroleadminservice.listbyroleid(authadminqueryform.getroleid());
    list<long> ids = new arraylist<>();
    if (authroleadmins != null && !authroleadmins.isempty()) {
      ids = authroleadmins.stream().map(authroleadmin::getadminid).collect(collectors.tolist());
    }
    authadminqueryform.setids(ids);
  }
  list<authadmin> authadminlist = authadminservice.listadminpage(authadminqueryform);

  // 查询所有的权限
  list<long> adminids = authadminlist.stream().map(authadmin::getid).collect(collectors.tolist());
  list<authroleadmin> authroleadminlist = authroleadminservice.listbyadminidin(adminids);

  // 视图列表
  list<authadminvo> authadminvolist = authadminlist.stream().map(item -> {
    authadminvo authadminvo = new authadminvo();
    beanutils.copyproperties(item, authadminvo);
    list<long> roles = authroleadminlist.stream()
        .filter(authroleadmin -> authadminvo.getid().equals(authroleadmin.getadminid()))
        .map(authroleadmin::getroleid)
        .collect(collectors.tolist());
    authadminvo.setroles(roles);
    return authadminvo;
  }).collect(collectors.tolist());

  pageinfo<authadmin> authadminpageinfo = new pageinfo<>(authadminlist);
  pagesimplevo<authadminvo> authadminpagesimplevo = new pagesimplevo<>();
  authadminpagesimplevo.settotal(authadminpageinfo.gettotal());
  authadminpagesimplevo.setlist(authadminvolist);

  return resultvoutils.success(authadminpagesimplevo);

}

相关地址

github 地址:

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。