Java注解之自定义注解并实现实战（初学者）_自定义注解如何使用实战模式

作者：你好赵伟 | 2024-03-15 11:13:36

踩

自定义注解如何使用实战模式

业务需求

半个月前，测试和我说：“如果商家被禁用了，那他就只能提现和查看自己的信息，不能修改。”

这把当时的我难住了。因为我们的商家是在一张表里的，所有的登录却是查的用户表，那怎么判断商家登录的他的商店是否被禁用了呢？我想到了一个以前从来没有尝试过的办法：AOP。

设计是这样的：被禁用的商家想要登录，没问题，可以，但他要想发货，操作商品，那不行，只有解封了才可以操作。在他禁用这段时间，他只能查看他的订单和商品情况，噢，还有可以提现。

于是乎，我东琢磨西琢磨，便琢磨出了这个注解。

代码赏析

1. 依赖

		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-aop</artifactId>
		</dependency>
1
2
3
4

2. 自定义注解

/**
 * 商家操作限制
 *
 * @author: YCL
 * @date: 2021-09-09 12:43:35
 */
@Target({
		ElementType.TYPE,//作用于类上
		ElementType.METHOD,//作用于方法上
})
@Retention(RetentionPolicy.RUNTIME)
public @interface OperationIntercept {

	/**
	 * 默认拦截
	 * 
	 * @return
	 */
	boolean intercept() default true;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

3. 注解实现

A. 首先，定义一个类，啥都不继承

/**
 * 操作限制实现
 *
 * @author: YCL
 * @date: 2021-09-09 12:49:50
 */
@Aspect //标明是切面编程
@Configuration
@Component
public class OperationInterceptAspect {

	@Resource
	private StringRedisTemplate stringRedisTemplate;
	@Resource
	private MallStoreService mallStoreService;

}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

B. 解析注解

	
	/**
	 * 这里不能用@annotation(), 不然类上面使用这个注解无效, 得使用@within()
	 */
	@Pointcut("@within(ycl.important.annocation.OperationIntercept)")
	public void pointcut(){
	}

	/**
	 * 一般来说，鼠标悬停在 pointcut() 上加上ctrl键点击会跑到上面的方法，如果没跳就看看类开头的 @Aspect 加了没有
	 * 
	 * 导包：
	 * org.aspectj.lang.JoinPoint
	 * org.aspectj.lang.reflect.MethodSignature
	 * java.lang.reflect.Method
	 * org.springframework.web.context.request.RequestContextHolder
	 */
	@Before("pointcut()")
	
	public void before(JoinPoint joinPoint){
		MethodSignature signature = (MethodSignature) joinPoint.getSignature();
		Method method = signature.getMethod();

		OperationIntercept annotation = method.getAnnotation(OperationIntercept.class);
		//是否校验, 我这边设置了为true就校验
		boolean intercept = true;
		if (annotation != null)
			intercept = annotation.intercept();
		if (!intercept)
			return;
			
		//获取request，项目的请求头token是被request带过来的，所以这里必须要request
		RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
		if (attributes == null)
			return;
		HttpServletRequest request = (HttpServletRequest) attributes.resolveReference(RequestAttributes.REFERENCE_REQUEST);
		if (request == null)
			return;
	
		Long id = mallStoreService.getUserId(request);
		Long mall = mallStoreService.selectIdByUserId(id);
		String s = stringRedisTemplate.opsForValue().get(RedisConstant.MERCHANT_DISABLE + mall);
		if (StrUtil.isNotBlank(s))
			throw new IllegalArgumentException("抱歉, 您被禁用, 无法操作");		
	}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45