自定义注解-验验证接口权限

有如下需求:在没有权限管理的情况下阻挡普通用户的接口请求,可以做自定义验证来验证接口权限。

 1、拦截器

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;

/**
 * 未登录拦截器
 */
public class BusinessLoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 获取HttpSession对象
        HttpSession httpSession = request.getSession();
        // 获取我们登录后存在session中的用户信息,如果为空,表示session已经过期
        Object sessionAttribute = httpSession.getAttribute(CommonConstant.PURCHASE_SHOPKEEPER);
        // 状态码2000 表示没登录
        if (null == sessionAttribute) {
            throw new LoginToException(ReturnCode.NO_LOGIN.getCode(), CommonConstant.NO_LOGIN);
        }
        String storeId = (String) WebUtils.getSessionAttribute(request, CommonConstant.STORE_ID);
        // 不是管理员必须有店铺id
        if (!isAdmin(request) && storeId == null) {
            //非法请求,未选择店铺!
            returnJson(response, ReturnCode.ILLEGAL_REQUEST_NO_STORE_SELECTED);
            return false;
        }
        // 判断是不是管理员 使用自定义注解
        if (handler instanceof HandlerMethod) {
            //自定义注解
            CheckAdmin annotation = AnnotationUtils.findAnnotation(((HandlerMethod) handler).getMethod(), CheckAdmin.class);
            if (annotation != null) {
                if (!isAdmin(request)) {
                    //没有权限
                    returnJson(response, ReturnCode.USER_NO_PERMISSION_LOCKED_ERROR);
                    return false;
                }
            }
        }
        //已经登录
        return true;
    }

    /**
     * @Description: 判断是不是管理员
     * @Param [request]
     * @Return: boolean
     **/
    private boolean isAdmin(HttpServletRequest request) {
        BusinessLoginRequest businessLoginRequest = (BusinessLoginRequest) request.getSession().getAttribute(CommonConstant.PURCHASE_SHOPKEEPER);
        return businessLoginRequest.getAccount().equals(CommonConstant.ADMIN);
    }

    /**
     * @Description: json异常信息
     * @Param [response, result]
     * @Return: void
     **/
    private void returnJson(HttpServletResponse response, ReturnCode result) throws IOException {
        String json = JSON.toJSONString(Result.Error(result));
        response.setContentType("application/json");
        response.setCharacterEncoding("utf-8");
        response.getWriter().println(json);
    }
}

2、自定义注解

import java.lang.annotation.*;

/**
 * @Description: 判断是否admin账号
 */
@Target({ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface CheckAdmin {

}

3、接口使用(直接放在接口方法上面) 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值