1、新建注解接口VerifyAccount.java:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
package com.zero4j.annotation; import java.lang.annotation.Documented; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; @Target({ElementType.PARAMETER}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface VerifyAccount { String paramName() default "token"; String permission() default ""; } |
2、创建注解对应的切面类VerifyAccountAspect.java:
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 46 47 48 49 50 51 52 53 54 55 |
package com.zero4j.annotation; import net.sf.json.JSONObject; import org.aspectj.lang.annotation.Aspect; import org.springframework.core.MethodParameter; import org.springframework.stereotype.Component; import org.springframework.web.bind.support.WebDataBinderFactory; import org.springframework.web.context.request.NativeWebRequest; import org.springframework.web.method.support.HandlerMethodArgumentResolver; import org.springframework.web.method.support.ModelAndViewContainer; import org.springframework.web.multipart.support.MissingServletRequestPartException; import com.zero4j.model.account.Account; import com.zero4j.model.token.util.TokenStaticUtil; @Aspect @Component public class VerifyAccountAspect implements HandlerMethodArgumentResolver{ @Override public boolean supportsParameter(MethodParameter parameter) { return parameter.hasParameterAnnotation(VerifyAccount.class); } @Override public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception { String token = webRequest.getParameter("token"); JSONObject out = new JSONObject(); if(token==null||token.equals("")){ out.put("status", 401); out.put("message", "请先注册并登录后再进行此操作"); out.put("debug", "缺少参数token或为空"); //ResponseStaticUtil.write(response,out); System.out.println(out.toString()); //return; } Account account = TokenStaticUtil.getAccount(token); if(account==null){ out.put("status", 401); out.put("message", "请先注册并登录后再进行此操作"); out.put("debug", "token对应的account为空"); //ResponseStaticUtil.write(response,out); System.out.println(out.toString()); throw new MissingServletRequestPartException("account"); //return; } return account; } } |
注意要实现接口:HandlerMethodArgumentResolver
3、最后要在SpringMVC的配置java类中加入对应的代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
package com.zero4j.config; import java.util.List; import org.springframework.context.annotation.Configuration; import org.springframework.web.method.support.HandlerMethodArgumentResolver; import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; import com.zero4j.annotation.VerifyAccountAspect; @Configuration public class WebAppConfigurer implements WebMvcConfigurer { @Override public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) { resolvers.add(new VerifyAccountAspect()); WebMvcConfigurer.super.addArgumentResolvers(resolvers); } } |
注意这行:resolvers.add(new VerifyAccountAspect());