自定义注解

如果框架内置的注解无法满足你的业务需求，你还可以自定义注解注入到框架中。

1、自定义注解

假设有以下业务需求

[!INFO| label:需求场景] 自定义一个注解 @CheckAccount，具有 name、pwd 两个字段，在标注一个方法上时，要求前端必须提交相应的账号密码参数才能访问方法。

1.1、第一步，创建一个注解

/**
 * 账号校验：在标注一个方法上时，要求前端必须提交相应的账号密码参数才能访问方法。
 */
@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.METHOD, ElementType.TYPE})
public @interface CheckAccount {

    /**
     * 需要校验的账号
     */
    String name();

    /**
     * 需要校验的密码
     */
    String pwd();

}

1.2、第二步，创建注解处理器

实现 SaAnnotationHandlerInterface 接口，指定泛型为刚才自定义的注解

/**
 * 注解 CheckAccount 的处理器
 */
@Component
public class CheckAccountHandler implements SaAnnotationHandlerInterface<CheckAccount> {

    // 指定这个处理器要处理哪个注解
    @Override
    public Class<CheckAccount> getHandlerAnnotationClass() {
        return CheckAccount.class;
    }

    // 每次请求校验注解时，会执行的方法
    @Override
    public void checkMethod(CheckAccount at, AnnotatedElement element) {
        // 获取前端请求提交的参数
        String name = SaHolder.getRequest().getParamNotNull("name");
        String pwd = SaHolder.getRequest().getParamNotNull("pwd");

        // 与注解中指定的值相比较
        if(name.equals(at.name()) && pwd.equals(at.pwd()) ) {
            // 校验通过，什么也不做
        } else {
            // 校验不通过，则抛出异常
            throw new SaTokenException("账号或密码错误，未通过校验");
        }
    }

}

参考上述代码，实现类上指定了 @Component 注解，使其可以在 ioc 环境下（如 Spring）被自动扫描注册 Sa-Token 中，如果你的项目属于非 ioc 环境，则需要手动将其注册到 Sa-Token 框架中：

SaAnnotationStrategy.instance.registerAnnotationHandler(new CheckAccountHandler());

1.3、测试自定义的注解

我们在一个请求接口上指定这个注解，来测试一下效果

@RestController
@RequestMapping("/test/")
public class TestController {

	@RequestMapping("test")
	@CheckAccount(name = "sa", pwd = "123456")
	public SaResult test() {
		System.out.println("------------进来了");
		return SaResult.ok(); 
	}
	
}

启动项目，使用浏览器访问此接口。

先来个错误的账号密码访问测试一下：http://localhost:8081/test/test?name=sa&pwd=123

返回结果：

{
  "code": 500,
  "msg": "账号或密码错误，未通过校验",
  "data": null
}

使用正确账号密码测试访问：http://localhost:8081/test/test?name=sa&pwd=123456

返回结果：

{
  "code": 200,
  "msg": "ok",
  "data": null
}

2、使用自定义注解优化多账号鉴权

在之前的 [ 多账号鉴权 ] 章节，我们介绍了利用 “spring 注解处理器” 达到注解合并的目的，从而简化多账号体系下的注解鉴权写法。

此种方案比较简单，但是也有一些缺点。

1、强依赖 Spring，无法在非 Spring 环境中使用。
2、注解递归检查可能会造成一些性能下降。
3、扩展性较低，只能略微简化框架内置好的注解写法，无法灵活扩展功能。

此处我们再演示一种方案，使用自定义注解的方式达到相同的目的。

2.1、首先定义注解

/**
 * 登录认证(User版)：只有登录之后才能进入该方法 
 * <p> 可标注在函数、类上（效果等同于标注在此类的所有方法上） 
 */
@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.METHOD, ElementType.TYPE})
public @interface SaUserCheckLogin {
	
}

2.2、定义注解处理器

/**
 * 注解 SaUserCheckLogin 的处理器
 */
@Component
public class SaUserCheckLoginHandler implements SaAnnotationHandlerInterface<SaUserCheckLogin> {

    @Override
    public Class<SaUserCheckLogin> getHandlerAnnotationClass() {
        return SaUserCheckLogin.class;
    }

    @Override
    public void checkMethod(SaUserCheckLogin at, AnnotatedElement element) {
        SaCheckLoginHandler._checkMethod(StpUserUtil.TYPE);
    }

}

2.3、使用新注解

接下来就可以使用我们的自定义注解了：

// 使用 @SaUserCheckLogin 的效果等同于使用：@SaCheckLogin(type = "user")
@SaUserCheckLogin
@RequestMapping("info")
public String info() {
    return "查询用户信息";
}

注：其它注解 @SaCheckRole("xxx")、@SaCheckPermission("xxx") 同理，完整示例参考 Gitee 代码：自定义注解。

5.1 KiB Raw Blame History Unescape Escape