mirror of
https://gitee.com/durcframework/SOP.git
synced 2025-08-11 21:57:56 +08:00
使用es6+axios
This commit is contained in:
Changeden
@@ -1,184 +1,213 @@
|
||||
const needle = require('needle');
|
||||
const axios = require('axios');
|
||||
const moment = require('moment');
|
||||
const qs = require('qs');
|
||||
|
||||
const {Class} = require('./Class');
|
||||
const {RequestType} = require('./RequestType');
|
||||
const {SignUtil} = require('./SignUtil');
|
||||
const {BaseRequest} = require('../request/BaseRequest');
|
||||
const RequestType = require('./RequestType');
|
||||
const SignUtil = require('./SignUtil');
|
||||
const BaseRequest = require('./BaseRequest');
|
||||
|
||||
const IS_RUN_IN_BROWSER = typeof window !== 'undefined' && this === window;
|
||||
|
||||
const HEADERS = {'Accept-Encoding': 'identity'};
|
||||
|
||||
const getHeaders = function (headers) {
|
||||
if (!headers) {
|
||||
return HEADERS;
|
||||
}
|
||||
for (const key in HEADERS) {
|
||||
headers[key] = HEADERS[key];
|
||||
}
|
||||
return headers;
|
||||
const getHeaders = (headers = {}) => {
|
||||
return Object.assign({}, headers, HEADERS);
|
||||
};
|
||||
|
||||
const OpenClient = Class.create({
|
||||
const parseResponse = (error, response, request) => {
|
||||
if (!error && response.status === 200) {
|
||||
return request.parseResponse(response.data);
|
||||
} else {
|
||||
return { // 重新封装请求异常回调，以防中断
|
||||
msg: '请求异常',
|
||||
code: '502',
|
||||
sub_msg: `${error}`,
|
||||
sub_code: 'isv.invalid-server'
|
||||
};
|
||||
}
|
||||
};
|
||||
|
||||
const buildParams = (instance, request, token) => {
|
||||
const {appId, privateKey} = instance;
|
||||
const allParams = {
|
||||
'app_id': appId,
|
||||
'method': request.getMethod(),
|
||||
'charset': 'UTF-8',
|
||||
'sign_type': 'RSA2',
|
||||
'timestamp': moment().format('YYYY-MM-DD HH:mm:ss'),
|
||||
'version': request.getVersion(),
|
||||
'biz_content': JSON.stringify(request.bizModel)
|
||||
};
|
||||
|
||||
if (token) {
|
||||
allParams['app_auth_token'] = token;
|
||||
}
|
||||
// 创建签名
|
||||
allParams.sign = SignUtil.createSign(allParams, privateKey, 'RSA2');
|
||||
return allParams;
|
||||
};
|
||||
|
||||
const executeRequest = async (instance = {}, request, token, callback, customOptions = {}) => {
|
||||
const params = buildParams(instance, request, token);
|
||||
const {url} = instance;
|
||||
let {headers} = customOptions;
|
||||
const config = {
|
||||
url,
|
||||
method: 'POST',
|
||||
params: undefined,
|
||||
data: undefined
|
||||
};
|
||||
headers = getHeaders(headers);
|
||||
const requestType = request.getRealRequestType();
|
||||
switch (requestType) {
|
||||
case RequestType.GET: {
|
||||
config.method = 'GET';
|
||||
config.params = params;
|
||||
}
|
||||
break;
|
||||
case RequestType.POST_FORM: {
|
||||
headers = Object.assign(headers, {
|
||||
'Content-Type': 'application/x-www-form-urlencoded'
|
||||
});
|
||||
config.data = qs.stringify(params);
|
||||
}
|
||||
break;
|
||||
case RequestType.POST_JSON: {
|
||||
config.data = params;
|
||||
}
|
||||
break;
|
||||
case RequestType.POST_FILE: {
|
||||
let formData;
|
||||
if (IS_RUN_IN_BROWSER) {
|
||||
formData = new window.FormData()
|
||||
(request.files || []).forEach(({name, path}) => {
|
||||
formData.append(name, path, {
|
||||
contentType: 'application/octet-stream'
|
||||
});
|
||||
});
|
||||
} else {
|
||||
const fs = require('fs');
|
||||
const fd = require('form-data');
|
||||
formData = new fd();
|
||||
(request.files || []).forEach(({name, path}) => {
|
||||
formData.append(name, fs.createReadStream(path), {
|
||||
contentType: 'application/octet-stream'
|
||||
});
|
||||
});
|
||||
headers = Object.assign(headers, formData.getHeaders());
|
||||
}
|
||||
Object.keys(params).forEach(key => {
|
||||
const value = params[key];
|
||||
if (!(typeof key === 'undefined' || typeof value === 'undefined')) {
|
||||
formData.append(key, params[key]);
|
||||
}
|
||||
});
|
||||
config.data = formData;
|
||||
}
|
||||
break;
|
||||
default: {
|
||||
callback(parseResponse(new Error('request.getRequestType()类型不正确'), undefined, request));
|
||||
return;
|
||||
}
|
||||
}
|
||||
try {
|
||||
config['headers'] = headers;
|
||||
const response = await axios.request(config);
|
||||
callback(parseResponse(undefined, response, request));
|
||||
} catch (error) {
|
||||
callback(parseResponse(error, undefined, request));
|
||||
}
|
||||
};
|
||||
|
||||
module.exports = class OpenClient {
|
||||
/**
|
||||
* 初始化客户端
|
||||
* @param appId 应用ID
|
||||
* @param privateKey 应用私钥，2048位，PKCS8
|
||||
* @param url 请求url
|
||||
*/
|
||||
init: function (appId, privateKey, url) {
|
||||
this.appId = appId;
|
||||
this.privateKey = privateKey;
|
||||
this.url = url;
|
||||
},
|
||||
constructor(appId, privateKey, url) {
|
||||
this.appId = appId || '';
|
||||
this.privateKey = privateKey || '';
|
||||
this.url = url || '';
|
||||
}
|
||||
|
||||
setAppId(appId) {
|
||||
this.appId = appId || '';
|
||||
return this;
|
||||
}
|
||||
|
||||
setPrivateKey(privateKey) {
|
||||
this.privateKey = privateKey || '';
|
||||
return this;
|
||||
}
|
||||
|
||||
setUrl(url) {
|
||||
this.url = url || '';
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送请求
|
||||
* @param request 请求类
|
||||
* @param callback 回调函数，参数json（undefined则使用executeSync）
|
||||
* @param options 自定义参数，如headers
|
||||
*/
|
||||
execute: function (request, callback) {
|
||||
execute(request, callback, options) {
|
||||
if (typeof callback == 'function') {
|
||||
this.executeToken(request, null, callback);
|
||||
return this.executeToken(request, null, callback, options);
|
||||
} else {
|
||||
return this.executeSync(request);
|
||||
return this.executeSync(request, options);
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送同步请求
|
||||
* @param request 请求类
|
||||
* @param options 自定义参数，如headers
|
||||
* */
|
||||
executeSync: function (request) {
|
||||
executeSync(request, options) {
|
||||
return new Promise((resolve) => {
|
||||
this.execute(request, res => {
|
||||
const _ = this.execute(request, res => {
|
||||
resolve(res);
|
||||
});
|
||||
}, options);
|
||||
});
|
||||
},
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送请求
|
||||
* @param request 请求类
|
||||
* @param token token
|
||||
* @param callback 回调函数，参数json（undefined则使用executeTokenSync）
|
||||
* @param options 自定义参数，如headers
|
||||
*/
|
||||
executeToken: function (request, token, callback) {
|
||||
async executeToken(request, token, callback, options) {
|
||||
if (!(request instanceof BaseRequest)) {
|
||||
throw 'request类未继承BaseRequest';
|
||||
}
|
||||
if (typeof callback == 'function') {
|
||||
const requestType = request.getRequestType();
|
||||
if (request.files) {
|
||||
this._postFile(request, callback);
|
||||
} else {
|
||||
switch (requestType) {
|
||||
case RequestType.GET:
|
||||
this._get(request, callback);
|
||||
break;
|
||||
case RequestType.POST_FORM:
|
||||
this._postForm(request, callback);
|
||||
break;
|
||||
case RequestType.POST_JSON:
|
||||
this._postJson(request, callback);
|
||||
break;
|
||||
case RequestType.POST_FILE:
|
||||
this._postFile(request, callback);
|
||||
break;
|
||||
default: {
|
||||
throw 'request.getRequestType()类型不正确';
|
||||
}
|
||||
}
|
||||
const files = request.files;
|
||||
if (files && files.length > 0) {
|
||||
request.setForceRequestType(RequestType.POST_FILE);
|
||||
}
|
||||
return await executeRequest(this, request, token, callback, options);
|
||||
} else {
|
||||
return this.executeTokenSync(request, token);
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送同步请求
|
||||
* @param request 请求类
|
||||
* @param token token
|
||||
* @param options 自定义参数，如headers
|
||||
*/
|
||||
executeTokenSync: function (request, token) {
|
||||
executeTokenSync(request, token, options) {
|
||||
return new Promise((resolve) => {
|
||||
this.executeToken(request, token, res => {
|
||||
const _ = this.executeToken(request, token, res => {
|
||||
resolve(res);
|
||||
});
|
||||
}, options);
|
||||
});
|
||||
},
|
||||
_get: function (request, callback) {
|
||||
const allParams = this._buildParams(request);
|
||||
const that = this;
|
||||
// needle.request(method, url, data[, options][, callback])
|
||||
needle.request('GET', this.url, allParams, {
|
||||
headers: getHeaders()
|
||||
}, function (error, response) {
|
||||
callback(that._parseResponse(error, response, request));
|
||||
});
|
||||
},
|
||||
_postForm: function (request, callback) {
|
||||
const allParams = this._buildParams(request);
|
||||
const that = this;
|
||||
needle.request('POST', this.url, allParams, {
|
||||
headers: getHeaders({
|
||||
'Content-Type': 'application/x-www-form-urlencoded'
|
||||
})
|
||||
}, function (error, response) {
|
||||
callback(that._parseResponse(error, response, request));
|
||||
});
|
||||
},
|
||||
_postJson: function (request, callback) {
|
||||
const allParams = this._buildParams(request);
|
||||
const that = this;
|
||||
needle.request('POST', this.url, allParams, {
|
||||
headers: getHeaders(), json: true
|
||||
}, function (error, response) {
|
||||
callback(that._parseResponse(error, response, request));
|
||||
});
|
||||
},
|
||||
_postFile: function (request, callback) {
|
||||
const allParams = this._buildParams(request);
|
||||
const files = request.files;
|
||||
files.forEach(row => {
|
||||
// 设置成{ file: row.path, content_type: 'application/octet-stream' }格式
|
||||
// needle会认为是上传文件
|
||||
allParams[row.name] = {file: row.path, content_type: 'application/octet-stream'};
|
||||
});
|
||||
const that = this;
|
||||
needle.request('POST', this.url, allParams, {
|
||||
headers: getHeaders(), multipart: true
|
||||
}, function (error, response) {
|
||||
callback(that._parseResponse(error, response, request));
|
||||
});
|
||||
},
|
||||
_parseResponse: function (error, response, request) {
|
||||
if (!error && response.statusCode === 200) {
|
||||
return request.parseResponse(response.body);
|
||||
} else {
|
||||
// throw '请求异常：' + error
|
||||
return { // 重新封装请求异常回调，以防中断
|
||||
msg: '请求异常',
|
||||
code: '502',
|
||||
sub_msg: `${error}`,
|
||||
sub_code: 'isv.invalid-server'
|
||||
};
|
||||
}
|
||||
},
|
||||
_buildParams: function (request, token) {
|
||||
const allParams = {
|
||||
'app_id': this.appId,
|
||||
'method': request.getMethod(),
|
||||
'charset': 'UTF-8',
|
||||
'sign_type': 'RSA2',
|
||||
'timestamp': moment().format('YYYY-MM-DD HH:mm:ss'),
|
||||
'version': request.getVersion(),
|
||||
'biz_content': JSON.stringify(request.bizModel)
|
||||
};
|
||||
|
||||
if (token) {
|
||||
allParams['app_auth_token'] = token;
|
||||
}
|
||||
// 创建签名
|
||||
const sign = SignUtil.createSign(allParams, this.privateKey, 'RSA2');
|
||||
allParams.sign = sign;
|
||||
return allParams;
|
||||
}
|
||||
});
|
||||
|
||||
module.exports = OpenClient;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user