server/app/service/util/http-util.js

'use strict';
const { AxiosService } = require('naf-framework-mongoose-free/lib/service');
const { BusinessError, ErrorCode } = require('naf-core').Error;
const { isNullOrUndefined } = require('naf-core').Util;
const _ = require('lodash');

//
class HttpUtilService extends AxiosService {
  constructor(ctx) {
    super(ctx, {}, {});
    this.dbServerPort = this.app.config.dbServerPort;
    this.dbServerIp = this.app.config.dbServerIp;
  }


  // 替换uri中的参数变量
  merge(uri, query = {}) {
    const keys = Object.keys(query);
    const arr = [];
    for (const k of keys) {
      arr.push(`${k}=${query[k]}`);
    }
    if (arr.length > 0) {
      uri = `${uri}?${arr.join('&')}`;
    }
    return uri;
  }

  /**
   * curl-get请求
   * @param {String} uri 接口地址
   * @param {Object} query 地址栏参数
   * @param {Object} options 额外参数
   */
  async $get(uri, query, options) {
    return this.toRequest(uri, null, query, options);
  }

  /**
   * curl-post请求
   * @param {String} uri 接口地址
   * @param {Object} data post的body
   * @param {Object} query 地址栏参数
   * @param {Object} options 额外参数
   */
  async $post(uri, data = {}, query, options) {
    return this.toRequest(uri, data, query, options);
  }

  /**
   * curl-post请求
   * @param {String} uri 接口地址
   * @param {Object} data post的body
   * @param {Object} query 地址栏参数
   * @param {Object} options 额外参数
   */
  async $delete(uri, data = {}, query, options) {
    options = { ...options, method: 'delete' };
    return this.toRequest(uri, data, query, options);
  }

  async toRequest(uri, data, query, options) {
    query = _.pickBy(query, val => val && val !== '' && val !== 'undefined' && val !== 'null');
    if (!(query.page && query.size)) {
      delete query.page;
      delete query.size;
    }
    if (!uri) console.error('uri不能为空');
    if (_.isObject(query) && _.isObject(options)) {
      const params = query.params ? query.params : query;
      options = { ...options, params };
    } else if (_.isObject(query) && !query.params) {
      options = { params: query };
    } else if (_.isObject(query) && query.params) {
      options = query;
    }
    // 是否多租户模式,需要改变headers
    const headers = { 'content-type': 'application/json' };
    let url = '';
    if (uri.includes('http')) url = uri;
    else url = this.merge(`http://${this.dbServerIp}:${this.dbServerPort}${uri}`, options.params);
    let res = await this.ctx.curl(url, {
      method: isNullOrUndefined(data) ? 'get' : 'post',
      url,
      data,
      dataType: 'json',
      headers,
      ...options,
    });
    if (res.status === 200) {
      res = res.data || {};
      const { code, message } = res;
      if (code) {
        console.warn(`[${uri}] fail: ${code}-${message} `);
        return { code, message };
      }
      if (_.isArray(res.data)) return { list: res.data };
      return res.data;
    }
    const { status } = res;
    console.warn(`[${uri}] fail: ${status}-${res.data.message} `);
  }
}

module.exports = HttpUtilService;