create-api-server.js 2.22 KB
import checkParams from '../../utils/check-params';
import apiMaps from '../../config/api-map';
import createReport from 'report-error';

const yohoApi = global.yoho.API;
const ufoAPI = global.yoho.UfoAPI;
const serviceApi = global.yoho.ServiceAPI;


const checkApiMap = url => {
  return apiMaps[url] ? apiMaps[url] : void 0;
};
const request = async({url, method, reqParams = {}, context}) => {
  const apiInfo = checkApiMap(url);
  const {env, user} = context;

  if (!apiInfo) {
    return Promise.reject(new Error(`未找到对应的接口:${url}`));
  }
  if (!apiInfo.service) {
    Object.assign(reqParams, {
      uid: (user && user.uid) ? {
        toString: () => {
          return user.uid;
        },
        sessionKey: user.sessionKey,
        appSessionType: user.appSessionType
      } : 1,
      method: apiInfo.api,
    });
  }
  const params = checkParams.getParams(reqParams, apiInfo);
  const cache = method.toLowerCase() !== 'get' ? false : apiInfo.cache;
  const headers = {
    'X-YOHO-IP': env.clientIp,
    'X-Forwarded-For': env.clientIp,
    'User-Agent': 'yoho/nodejs'
  };

  try {
    if (apiInfo.service) {
      return await serviceApi.get(`${apiInfo.api}${apiInfo.path}`, params, {
        cache: cache,
        headers
      });
    } else if (apiInfo.ufo) {
      return await ufoAPI[method](`${apiInfo.path || ''}`, params, {
        cache: cache,
        headers
      });
    } else {
      return await yohoApi[method](`${apiInfo.path || ''}`, params, {
        cache: cache,
        headers
      });
    }
  } catch (error) {
    return Promise.reject({
      code: error.code || 500,
      message: error.message || '服务器错误'
    });
  }
};

const catchError = (context, reqParams) => {
  return result => {
    if (result && result.code === 500) {
      createReport(context, 'api')(Object.assign({
        api: reqParams.method,
      }, result));
    }
    return result;
  };
};

export const createApi = context => {
  return {
    get(url, reqParams = {}) {
      return request({url, method: 'get', reqParams, context}).then(catchError(context, reqParams));
    },
    post(url, reqParams = {}) {
      return request({url, method: 'post', reqParams, context}).then(catchError(context, reqParams));
    }
  };
};