cube.js 自定义checkAuth 响应状态码

如果阅读源码了,会发现默认是基于UserError 进行处理的,但是默认只有403,解决方法

参考源码

通过以下源码可以看到内部处理

 protected checkAuth: RequestHandler = async (req, res, next) => {
    const token = this.extractAuthorizationHeaderWithSchema(req);
 
    try {
      await this.checkAuthFn(req, token);
      if (next) {
        next();
      }
    } catch (e) {
      if (e instanceof UserError) {
         // 此处需要修改为e.status, 对于需要自定义状态码的可以自己搞一个error 实现
        res.status(403).json({ error: e.message });
      } else {
        this.log({
          type: 'Auth Error',
          token,
          error: e.stack || e.toString()
        }, <any>req);
        res.status(500).json({ error: e.toString() });
      }
    }
  };

自定义Error 解决

const CubejsServer = require('@cubejs-backend/server');
const {UserError} = require("@cubejs-backend/api-gateway")
 
 
class MyUserError extends UserError {
    constructor(message){
       super(message)
        // 改写自定义异常状态码
       this.status=401
    }
}
const server = new CubejsServer({
    checkAuth: (ob) => {
        throw new MyUserError('Could not authenticate user from LDAP');
    }
});
 
server.listen().then(({ version, port }) => {
  console.log(`🚀 Cube.js server (${version}) is listening on ${port}`);
});

说明

官方目前也已经支持了次功能了,参考官方pr https://github.com/cube-js/cube.js/pull/2568

posted on 2021-04-16 23:23  荣锋亮  阅读(147)  评论(0编辑  收藏  举报

导航