SpringBoot自定义注解使用AOP实现请求参数解密以及响应数据加密
SpringBoot自定义注解使用AOP实现请求参数解密以及响应数据加密
一、前言
本篇文章将依托与SpringBoot平台,自定义注解用来标识接口请求是否实现加密解密。使用AOP切面来具体操作解密加密,实现对源代码的低耦合,不在原基础上做很大的改动。
本篇文章的所有示例,都上传到我的github中,欢迎大家拉取测试,欢迎star github实现要求:
- 自定义一个注解@Secret,用来标识需要实现加密解密
- 作用在Controller类上,表示此Controller类的所有接口都实现加密解密
- 作用来单一方法上,表示此接口方法需要实现加密解密
- 使用AOP切面编程实现
- 在接口方法执行之前将前端的加密参数解密并重新赋给接口参数
- 在接口方法响应之后,将返回的数据进行加密返回
- 在配置文件中配置,是否开启全局的加密解密操作
文章图片
Image.png
- 前端请求的接口将请求参数json通过AES加密生成加密字符串,然后将加密字符串通过名为encyptStr字段传递给后端。
- AOP前置方法拦截,将encyptStr字符串通过AES解密得到原始请求参数json,将json映射为请求方法的参数对象User。
- 接口通过参数成功响应,并将响应数据直接返回。
- AOP后置方式拦截,将响应参数data字段里的数据AES加密,并返回给前端
- 前端收到请求响应,通过code判断请求是否成功,AES加密data字段得到需要的数据。
1. 创建SpringBoot项目 创建一个SpringBoot项目,导入必要的maven依赖。
- 使用AOP切面需要导入AOP的启动器
- lombok是一个通过注解简化代码的工具,在idea中使用需要安装lombok插件
- json转换工具,apache工具类
pom.xml
org.springframework.boot
spring-boot-starter-web
org.springframework.boot
spring-boot-starter-aop
org.projectlombok
lombok
true
com.alibaba
fastjson
1.2.52.sec06
org.apache.commons
commons-lang3
3.9
2. 自定注解@Secret 我们通过自定义的注解,来标识类或接口,告诉AOP哪些类或方法需要执行加密解密操作,更加的灵活。
Secret.java
package com.agger.springbootaopdemo.annotation;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* @classname: Secret
* @description: 自定义注解,用来标识请求类 或者方法是否使用AOP加密解密
* @author chenhx
* @date 2019-12-05 13:48:03
*/
@Target({ElementType.TYPE,ElementType.METHOD})// 可以作用在类上和方法上
@Retention(RetentionPolicy.RUNTIME)// 运行时起作用
public @interface Secret {// 参数类(用来传递加密数据,只有方法参数中有此类或此类的子类才会执行加解密)
Class value();
// 参数类中传递加密数据的属性名,默认encryptStr
String encryptStrName() default "encryptStr";
}
自定义注解很简单,只需要确定注解的作用位置和运行时机。其中有两个变量value和encryptStrName。
value没有默认值,是必传的参数,用来表示需要加解密的参数类或父类。AOP中或用到。
encryptStrName默认值为"encryptStr",用来表示前端传递的加密参数名称是什么,value类中必须存在此字段
3. Controller中使用 定义好@Secret注解后,我们就可以在Controller中使用了,不过现在只相当于是一个标注,还没有起任何作用,需要我们再定义好AOP后才会起作用。
- @Secret注解作用来类上
UserController.java
package com.agger.springbootaopdemo.controller;
import com.agger.springbootaopdemo.annotation.Secret;
import com.agger.springbootaopdemo.vo.BaseVO;
import com.agger.springbootaopdemo.vo.ResultVO;
import com.agger.springbootaopdemo.vo.UserVO;
import org.springframework.web.bind.annotation.*;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
/**
* @program: UserController
* @description: 用户控制类
* @author: chenhx
* @create: 2019-12-03 15:22
**/
@Secret(BaseVO.class)//接口参数和返回要进行加解密
@RestController
@RequestMapping("user")
public class UserController {//采用内部类的实例代码块方式初始化map
HashMap userMap = new HashMap(){
{
put(1,new UserVO(1,"张三"));
put(2,new UserVO(2,"李四"));
put(3,new UserVO(3,"王五"));
}
};
// 通过id查询用户
@GetMapping("getUserName/{id}")
public ResultVO getUserName(@PathVariable("id")Integer id){
return new ResultVO(0,"查询成功",userMap.get(id));
}// 通过name查询用户id
@GetMapping("getUserId")
public ResultVO getUserId(@RequestParamString name){
Iterator> iterator = userMap.entrySet().iterator();
UserVO u = null;
while (iterator.hasNext()){
Map.Entry entry = iterator.next();
if(entry.getValue().getName().equals(name)){
u = entry.getValue();
break;
}
}
return new ResultVO(0,"查询成功",u);
}// 新增用户
@PostMapping("addUser")
public ResultVO addUser(@RequestBody UserVO user){
return new ResultVO(0,"新增成功",user);
}// 更改用户
@PostMapping("updateUser")
public ResultVO updateUser(@RequestBody UserVO user) throws Throwable {
if(user==null||user.getId()==null){
throw new NullPointerException();
}else{
return new ResultVO(0,"修改成功",user);
}
}
}
@Secret(BaseVO.class)定义在了UserController类上,表示整个类下面的方法都会实现AOP加密解密。BaseVO是所有vo的基类,其中只定义了一个字段encryptStr,也就是前端传递的加密参数字段。
BaseVO.java
package com.agger.springbootaopdemo.vo;
import lombok.Data;
/**
* @program: BaseVO
* @description: 基类
* @author: chenhx
* @create: 2019-12-05 15:15
**/
@Data
public class BaseVO {
// 加密密文
private String encryptStr;
}
UserVO.java
package com.agger.springbootaopdemo.vo;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* @program: User
* @description: 用户
* @author: chenhx
* @create: 2019-12-03 15:31
**/
@Data
@NoArgsConstructor
@AllArgsConstructor
public class UserVO extends BaseVO{
private Integer id;
private String name;
}
- @Secret注解作用在方法上,表示只有此方法才需要执行AOP加密解密
DeptController.java
package com.agger.springbootaopdemo.controller;
import com.agger.springbootaopdemo.annotation.Secret;
import com.agger.springbootaopdemo.vo.DeptVO;
import com.agger.springbootaopdemo.vo.ResultVO;
import org.springframework.web.bind.annotation.*;
/**
* @program: Dept
* @description: 部门类
* @author: chenhx
* @create: 2019-12-03 15:26
**/
@RestController
@RequestMapping("dept")
public class DeptController {@GetMapping("getDeptName/{id}")
public ResultVO getDeptName(@PathVariable("id") String id){
return new ResultVO(0,"查询成功","财务部" + id);
}// 注解在方法上,并传递了encryptStrName自己定义的加密字符串名称encryptJson
@Secret(value = https://www.it610.com/article/DeptVO.class,encryptStrName ="encryptJson")
@PostMapping("addDept")
public ResultVO addDept(@RequestBody DeptVO dept){
return new ResultVO(0,"新增成功",dept);
}}
DeptVO类没有继承BaseVO类,自己写了一个前端需要传递的加密字符串字段,并传递给注解。ResultVO为接口响应类。
DeptVO.java
package com.agger.springbootaopdemo.vo;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* @program: Dept
* @description: 部门类
* @author: chenhx
* @create: 2019-12-03 15:32
**/
@Data
@NoArgsConstructor
@AllArgsConstructor
public class DeptVO{private Integer id;
private String deptName;
// 自己实现的一个参数,用来给前端传递加密字符串
private String encryptJson;
}
ResultVO.java
package com.agger.springbootaopdemo.vo;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* @program: ResultVO
* @description: 响应类
* @author: chenhx
* @create: 2019-12-03 15:34
**/
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ResultVO {
private Integer code;
private String msg;
private Object data;
}
4. 定义AOP切面 万事具备,咱只欠定义一个AOP切面来实现加密和解密操作了。
SecretAOPController.java
package com.agger.springbootaopdemo.aop;
import com.agger.springbootaopdemo.annotation.Secret;
import com.agger.springbootaopdemo.utils.AESUtils;
import com.agger.springbootaopdemo.vo.ResultVO;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
/**
* @program: SecretAOPController
* @description: 切面加密解密
* @author: chenhx
* @create: 2019-12-05 13:43
**/
@Aspect
@Component
@Slf4j
public class SecretAOPController {// 是否进行加密解密,通过配置文件注入(不配置默认为true)
@Value("${isSecret:true}")
boolean isSecret;
// 定义切点,使用了@Secret注解的类 或 使用了@Secret注解的方法
@Pointcut("@within(com.agger.springbootaopdemo.annotation.Secret) || @annotation(com.agger.springbootaopdemo.annotation.Secret)")
public void pointcut(){}// 环绕切面
@Around("pointcut()")
public ResultVO around(ProceedingJoinPoint point){
ResultVO result = null;
// 获取被代理方法参数
Object[] args = point.getArgs();
// 获取被代理对象
Object target = point.getTarget();
// 获取通知签名
MethodSignature signature = (MethodSignature )point.getSignature();
try {
// 获取被代理方法
Method pointMethod = target.getClass().getMethod(signature.getName(), signature.getParameterTypes());
// 获取被代理方法上面的注解@Secret
Secret secret = pointMethod.getAnnotation(Secret.class);
// 被代理方法上没有,则说明@Secret注解在被代理类上
if(secret==null){
secret = target.getClass().getAnnotation(Secret.class);
}if(secret!=null){
// 获取注解上声明的加解密类
Class clazz = secret.value();
// 获取注解上声明的加密参数名
String encryptStrName = secret.encryptStrName();
for (int i = 0;
i < args.length;
i++) {
// 如果是clazz类型则说明使用了加密字符串encryptStr传递的加密参数
if(clazz.isInstance(args[i])){
Object cast = clazz.cast(args[i]);
//将args[i]转换为clazz表示的类对象
// 通过反射,执行getEncryptStr()方法,获取加密数据
Method method = clazz.getMethod(getMethedName(encryptStrName));
// 执行方法,获取加密数据
String encryptStr = (String) method.invoke(cast);
// 加密字符串是否为空
if(StringUtils.isNotBlank(encryptStr)){
// 解密
String json = AESUtils.decrypt(encryptStr);
// 转换vo
args[i] = JSON.parseObject(json, (Type) args[i].getClass());
}
}
// 其他类型,比如基本数据类型、包装类型就不使用加密解密了
}
}// 执行请求
result = (ResultVO) point.proceed(args);
// 判断配置是否需要返回加密
if(isSecret){
// 获取返回值json字符串
String jsonString = JSON.toJSONString(result.getData());
// 加密
String s = AESUtils.encrypt(jsonString);
result.setData(s);
}} catch (NoSuchMethodException e) {
log.error("@Secret注解指定的类没有字段:encryptStr,或encryptStrName参数字段不存在");
e.printStackTrace();
} catch (Throwable throwable) {
throwable.printStackTrace();
}
return result;
}// 转化方法名
private String getMethedName(String name){
String first = name.substring(0,1);
String last = name.substring(1);
first = StringUtils.upperCase(first);
return "get" + first + last;
}
}
- 定义字段isSecret通过@Value从配置文件中注入true/false,来规定是否执行全局的AOP加密解密,在开发测试环境我们可以配置为不加密解密,方便查找错误。当项目上线运行时,可以配置true,实现加密操作。默认不配置为true
application.yml
# 配置是否开启AOP参数加密解密,不配置默认为true
isSecret: true
- 定义了AOP的切入点@Pointcut("@within(Secret)||@annotation(Secret)"),@Within表示匹配类上的指定注解,@annotation表示匹配方法上的指定注解
- 使用环绕通知切面来实现加密和解密。point.proceed(args); 表示执行请求方法,此方法之前表示前置切点,此方法之后表示后置切点。args是已经解密后的参数重新赋值传入。
- 通过被代理的接口方法反射找到@Secret注解是在类上还是在方法上,并获取@Secret注解类对象,找到传入的value和encryptStrName
- 我默认只有是单一参数的接口使用vo接收参数的惨进行加密解密,如果是单一参数比如
getUser(Integer id) { ... }
这种形式的接口建议全部用vo来接收,并且继承BaseVO基类,具有encryptStr字段,或者你自定义的加密字段,在你使用@Secret的时候,别忘了填写
// 注解在方法上,并传递了encryptStrName自己定义的加密字符串名称
@Secret(value = https://www.it610.com/article/DeptVO.class,encryptStrName ="encryptJson")
5. 执行效果 我们可以通过postman或其他前端工具来调取接口,在这里我使用的是idea自带的接口调试工具。
选择工具栏的Tools > HTTP Client > TestRESTfull Web Service
文章图片
Image1.png 点击后,就会在底部打开这个工具,跟postman一样,可以编辑请求方法,请求参数等等。
文章图片
Image2.png 不过,看点上面的提示没有?This REST Client is deprecated.Try our new HTTP Client in the editor
已经过时了,请使用新的客户端编辑,点击右边的按钮,idea就会给我们生成一个.http结尾的请求文件,我们可以像编辑配置文件一样,编辑请求,然后点击执行!!每个请求以三个#好分割,并可以分别执行。
文章图片
Image3.png
- 编写好我的请求文件,执行addUser方法试试。
文章图片
Image4.png
- 执行addDept方法测试一下,这个接口我们吧@Secret注解在了接口方法上,并传递了自定义的加密字段encryptJson。所以前端传递的加密参数名应该为encryptJson
文章图片
Image5.png
【SpringBoot自定义注解使用AOP实现请求参数解密以及响应数据加密】AESUtils就是一个普通的AES加密工具,在此没有展示出来,需要的可以去我的 github 获取,欢迎star三、总结
- 此篇文章实现,使用了自定注解的功能,其实自定义注解无非就是这样用,通过反射来获取操作和标识作用。
- AOP面向切面,其实SpringBoot的自动配置就是使用的AOP切面来实现的,我们通过自己实现一个切面,可以了解到整个执行流程和反射的应用。
- 建议需要加密操作的接口参数都用vo对象来接收,因为切面中只能获取接口参数,不能获取到接口不存在的参数名。所以前端在传递加密参数名的时候,一定要保证接口参数具有相同的名称来接收,解密后再赋值其他参数。
- AOP切面还有很多的实现,比如@Before @After @AfterReturning等,在此不过多讲解,本片只使用了@Around环绕通知。你也可以将解密加密分别拆分到@Before和@After中去执行。
- rest-api.http文件是idea HTTPClient工具生成的api测试文件,默认没保存在项目中,我已经将此文件放在了test目录下,大家可以查看。
- 本片文章参考了博客 https://blog.csdn.net/lmb55/article/details/82470388
- 本篇文章的所有示例,都上传到我的github中,欢迎大家拉取测试,欢迎star github
推荐阅读
- Activiti(一)SpringBoot2集成Activiti6
- SpringBoot调用公共模块的自定义注解失效的解决
- python自定义封装带颜色的logging模块
- 解决SpringBoot引用别的模块无法注入的问题
- 列出所有自定义的function和view
- 生活与游戏日记(第182篇)(〔成长瞬间〕关注解决问题2019.10)
- springboot使用redis缓存
- Spring|Spring Boot 自动配置的原理、核心注解以及利用自动配置实现了自定义 Starter 组件
- 自定义MyAdapter
- springboot整合数据库连接池-->druid