@Currentuser注解的实现

左手的ㄟ右手 2022-05-24 07:58 164阅读 0赞

1.在springmvc.xml中添加驱动，添加内置标签mvc:argument-resolvers，注入bean

<mvc:annotation-driven validator="validator" >
        	
        	<mvc:argument-resolvers>   
                <bean class="org.csun.nc.resolvers.CurrentUserMethodArgumentResolver"/>  
            </mvc:argument-resolvers>
            
        </mvc:annotation-driven>

2.定义@CurrentUser注解

@Target(ElementType.PARAMETER)
    @Retention(RetentionPolicy.RUNTIME)
    public @interface CurrentUser {
    }

3.解析器，注入的bean具体类

public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {
    
        @Autowired
        private UserService userService;
    
        @Override
        public boolean supportsParameter(MethodParameter methodParameter) {
            //如果参数类型是User并且有CurrentUser注解则支持
            if (methodParameter.getParameterType().isAssignableFrom(User.class) &&
                    methodParameter.hasParameterAnnotation(CurrentUser.class)) {
                return true;
            }
            return false;
        }
    
        @Override
        public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer,
                                      NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
            //取出鉴权时存入的登录用户Id
            Long currentUserId = (Long) nativeWebRequest.getAttribute(Constants.CURRENT_USER_ID, RequestAttributes.SCOPE_REQUEST);
            if (currentUserId != null) {
                //从数据库中查询并返回
                return userService.getUserById(currentUserId);
            }
            throw new MissingServletRequestPartException(Constants.CURRENT_USER_ID);
        }
    }

4.在congtroller类中方法中使用该注解

@Controller
    @CrossOrigin
    @RequestMapping("Evaluate")
    public class EvaluateController {
    
    	@Resource
    	private EvaluateService evaluateService;
    	
    	// 添加评估记录
    	@Authorization
    	@ResponseBody
    	@RequestMapping(value = "/EvaluateAdd", method = RequestMethod.POST)
    	public JsonResult evaluateAdd(@CurrentUser User loginUser,
    			@Validated(value = { EvaluateValid.class }) Evaluate evaluate,BindingResult br) {
    		
    		String[] validateParam = { "oldmanId", "evaluateType", "evaluateDate",
    				"evaluator", "result", "picture" };
    		Map<String, String> errorMap = ValidateUtil.getErrorMap(br,
    				validateParam);
    		if (errorMap != null && errorMap.size() != 0) {
    			/* 返回参数校验错误详情 */
    			return new JsonResult(false, "添加评估记录参数错误", errorMap);
    		}
    		Evaluate evaluate1 = evaluateService.selectByOldManId(evaluate);
    		if(!(null == evaluate1)){
    			return new JsonResult(false, "该老人该类型的评估记录数据已存在", 0);
    		}
    		evaluate.setModifier(loginUser.getUserid());
    		evaluate.setHomeId(loginUser.getHomeid());
    		int n = evaluateService.insert(evaluate);
    		if (n <= 0) {
    			return new JsonResult(false, "添加失败", n);
    		}
    		return new JsonResult(true, "添加成功", n);
    	}