Skip to content

Commit f82b34f

Browse files
authored
Merge pull request #25 from shutterstock/dependencies-upgrade
Update dependencies
2 parents 96879cc + b1037a3 commit f82b34f

File tree

6 files changed

+2099
-2853
lines changed

6 files changed

+2099
-2853
lines changed

lib/driver/aws-sigv4-driver-remote-connection.js

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
const debug = require('debug')('gremlin-aws-sigv4:driver');
22
const gremlin = require('gremlin');
3-
const request = require('request');
4-
const { getUrlAndHeaders, uuid } = require('../utils');
3+
const { getUrlAndHeaders, uuid, request } = require('../utils');
54

65
class AwsSigV4DriverRemoteConnection extends gremlin.driver.RemoteConnection {
76
constructor(host, port, options = {}, cbConnected = null, cbDisconnected = null, cbError = null) {
@@ -37,9 +36,11 @@ class AwsSigV4DriverRemoteConnection extends gremlin.driver.RemoteConnection {
3736
this.try += 1;
3837
const { url, headers } = getUrlAndHeaders(this.host, this.port, this.options, '/status', this.secure ? 'https' : 'http');
3938
debug(`get Neptune status: ${JSON.stringify(url, headers)} (try #${this.try})`);
40-
request.get({ url, headers, timeout: 3000 }, (error, response, body) => {
41-
this._statusCallback(error, response, body);
42-
});
39+
request(
40+
url,
41+
{ headers, timeout: 3000 },
42+
(err, response, body) => { this._statusCallback(err, response, body); },
43+
);
4344
}
4445

4546
_statusCallback(error, response, body) {

lib/utils.js

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
/* eslint-disable no-bitwise */
22

3+
const aws4 = require('aws4');
34
const crypto = require('crypto');
45
const debug = require('debug')('gremlin-aws-sigv4:utils');
5-
const aws4 = require('aws4');
6+
const http = require('http');
7+
const https = require('https');
68

79
/**
810
* Generates a random uuid
@@ -56,6 +58,25 @@ const getUrlAndHeaders = (host, port, credentials, canonicalUri, protocol) => {
5658
};
5759
};
5860

61+
/**
62+
* Callbackify native http(s) get
63+
*/
64+
const request = (url, options, cb) => {
65+
debug(`Requesting: ${url}`);
66+
(url.match('https://') ? https.get : http.get)(url, options, (response) => {
67+
let body = '';
68+
response.on('data', (d) => {
69+
body += d;
70+
});
71+
response.on('end', () => {
72+
cb(null, response, body);
73+
});
74+
})
75+
.on('error', (err) => {
76+
cb(err, null, null);
77+
});
78+
};
79+
5980
module.exports = {
60-
uuid, getUrlAndHeaders,
81+
uuid, getUrlAndHeaders, request,
6182
};

0 commit comments

Comments
 (0)