欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2024-03-05 10:22:42
这里使用 aop 来实现权限验证 引入依赖 org.springframework.b...

这里使用 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 地址:

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