magazine-service/service-gateway2/app/controller/home.js

'use strict';

const Controller = require('egg').Controller;
const jsonwebtoken = require('jsonwebtoken');
const { createProxyMiddleware, responseInterceptor } = require('http-proxy-middleware');
class HomeController extends Controller {
  async index() {
    const { modules } = this.app.config;
    // 路径参数
    const { agent, service, module, method, item } = this.ctx.params;
    // 请求地址
    const address = modules[service];
    // 请求方法
    const requestMethod = this.ctx.request.method;
    // 请求路径
    const path = this.ctx.path;
    const exampleProxy = createProxyMiddleware(
      `/${agent}/${service}/`,
      {
        target: `${address}${path}`,
        changeOrigin: true,
        selfHandleResponse: true,
        onProxyRes: responseInterceptor(async (buffer, proxyRes, req, res) => {
          console.log(buffer, 'buffer');
          console.log(proxyRes, 'proxyRes');
          console.log(req, 'req');
          console.log(res, 'res');
        }),
      });
    this.app.use(exampleProxy);
    this.app.listen(3000);
    // // jwt 验证
    // const auth = await this.auth({ service, path });
    // // 验证失败
    // if (auth.errcode !== 0) {
    //   this.ctx.status = 401;
    //   this.ctx.body = { errcode: 401, errmsg: '身份验证失败', details: auth.details };
    //   return false;
    // }
    this.ctx.body = 111;
  }

  // 日志记录
  async setLog({ agent, service, module, method, item, details, result, auth }) {
    // 配置
    const { modules, apipath } = this.app.config;
    // 请求路径
    const path = this.ctx.path;
    // 根据service类型 获取配置文件
    const logmodules = apipath[service];
    // 没有配置文件默认放过
    if (logmodules) {
      // const items = logmodules.find(e => e.url.includes(path));
      const items = await this.rewrite_path(logmodules, path);
      if (items && items.log) {
        // 此处开始记录日志
        let userName = null;
        let name = null;
        if (auth.decode) {
          userName = auth.decode.userName;
          name = auth.decode.name;
        }
        try {
          await this.ctx.curl(`${modules.log}/api/log/log/create`, {
            method: 'POST',
            dataType: 'json',
            data: { agent, service, module, method, item, details, result, userName, name },
          });
        } catch (error) {
          this.ctx.logger.error(`日志添加失败: ${error}`);
        }
      }
    }
  }

  // 验证函数
  async auth({ service, path }) {
    const { jwt, apipath } = this.app.config;
    // 根据service类型 获取配置文件
    const modules = apipath[service];
    // 没有配置文件默认放过
    if (!modules) return { errcode: 0, details: '' };
    // 获取与当前路径匹配的项
    // const item = modules.filter(e => e.url.includes(path));
    const item = await this.rewrite_path(modules, path);
    // 没有匹配数据  默认放过
    if (item) return { errcode: 0, details: '' };
    // 不验证jwt放过
    if (!item.jwt) return { errcode: 0, details: '' };
    // 获取token
    const token = this.ctx.header.authorization.split('Bearer ')[1];
    try {
      // 解密jwt
      const decode = jsonwebtoken.verify(token, jwt.secret);
      // 如果有验证签发者
      if (item && item.issuer) {
        // 比对签发者
        if (!item.issuer.includes(decode.iss)) return { errcode: -4001, details: 'issuer fail verification' };
      }
      return { errcode: 0, details: '', decode };
    } catch (error) {
      return { errcode: -4001, details: error.message };
    }
  }
  // 重写路径
  async rewrite_path(modules, path) {
    return modules.find(e => {
      const { url } = e;
      let data = null;
      const urlList = url.split('/');
      const pathIist = path.split('/');
      const urlItem = urlList.findIndex(j => j.includes(':'));
      if (urlItem > 0) data = pathIist[urlItem];
      if (data !== null) urlList[urlItem] = data;
      const pathItem = urlList.join('/');
      if (pathItem === path) return e;
      return false;
    });
  }
}

module.exports = HomeController;