Compare commits
1 Commits
developmen
...
bf/S3C-102
Author | SHA1 | Date |
---|---|---|
Rahul Padigela | 4214282faf |
|
@ -1,7 +1,12 @@
|
|||
const url = require('url');
|
||||
const { eachSeries } = require('async');
|
||||
const ipCheck = require('../ipCheck');
|
||||
const errors = require('../errors');
|
||||
|
||||
const responseErr = new Error();
|
||||
responseErr.code = 'ResponseError';
|
||||
responseErr.message = 'response closed by client request before all data sent';
|
||||
|
||||
/**
|
||||
* setCommonResponseHeaders - Set HTTP response headers
|
||||
* @param {object} headers - key and value of new headers to add
|
||||
|
@ -242,54 +247,78 @@ function okContentHeadersResponse(overrideParams, resHeaders,
|
|||
return response;
|
||||
}
|
||||
|
||||
function retrieveData(locations, dataRetrievalFn,
|
||||
response, logger, errorHandlerFn) {
|
||||
if (locations.length === 0) {
|
||||
return response.end();
|
||||
}
|
||||
if (errorHandlerFn === undefined) {
|
||||
// eslint-disable-next-line
|
||||
errorHandlerFn = () => { response.connection.destroy(); };
|
||||
}
|
||||
const current = locations.shift();
|
||||
if (current.azureStreamingOptions) {
|
||||
// pipe data directly from source to response
|
||||
response.on('error', err => {
|
||||
logger.error('error piping data from source');
|
||||
errorHandlerFn(err);
|
||||
});
|
||||
return dataRetrievalFn(current, response, logger, err => {
|
||||
function retrieveData(locations, retrieveDataFn, response, log) {
|
||||
let responseDestroyed = false;
|
||||
const _destroyResponse = () => {
|
||||
// destroys the socket if available
|
||||
response.destroy();
|
||||
responseDestroyed = true;
|
||||
};
|
||||
response.once('close', () => {
|
||||
log.debug('received close event before response end');
|
||||
_destroyResponse();
|
||||
});
|
||||
|
||||
eachSeries(locations,
|
||||
(current, next) => retrieveDataFn(current, log, (err, readable) => {
|
||||
let cbCalled = false;
|
||||
const _next = err => {
|
||||
// Avoid multiple callbacks since it's possible that response's
|
||||
// close event and the readable's end event are emitted at
|
||||
// the same time.
|
||||
if (!cbCalled) {
|
||||
cbCalled = true;
|
||||
next(err);
|
||||
}
|
||||
};
|
||||
|
||||
if (err) {
|
||||
logger.error('failed to get object from source', {
|
||||
error: err,
|
||||
method: 'retrieveData',
|
||||
backend: 'Azure',
|
||||
});
|
||||
return errorHandlerFn(err);
|
||||
}
|
||||
return undefined;
|
||||
});
|
||||
}
|
||||
return dataRetrievalFn(current, response, logger,
|
||||
(err, readable) => {
|
||||
if (err) {
|
||||
logger.error('failed to get object', {
|
||||
log.error('failed to get object', {
|
||||
error: err,
|
||||
method: 'retrieveData',
|
||||
});
|
||||
return errorHandlerFn(err);
|
||||
_destroyResponse();
|
||||
return _next(err);
|
||||
}
|
||||
readable.on('error', err => {
|
||||
logger.error('error piping data from source');
|
||||
errorHandlerFn(err);
|
||||
if (responseDestroyed) {
|
||||
log.debug('response destroyed before readable could stream');
|
||||
readable.emit('close');
|
||||
return _next(responseErr);
|
||||
}
|
||||
if (current.azureStreamingOptions) {
|
||||
// azure-sdk handles the streams
|
||||
return _next();
|
||||
}
|
||||
// client closed the connection abruptly
|
||||
response.once('close', () => {
|
||||
log.debug('received close event before readable end');
|
||||
if (!responseDestroyed) {
|
||||
_destroyResponse();
|
||||
}
|
||||
readable.emit('close');
|
||||
return _next(responseErr);
|
||||
});
|
||||
// readable stream successfully consumed
|
||||
readable.on('end', () => {
|
||||
process.nextTick(retrieveData,
|
||||
locations, dataRetrievalFn, response, logger);
|
||||
log.debug('readable stream end reached');
|
||||
return _next();
|
||||
});
|
||||
readable.pipe(response, { end: false });
|
||||
return undefined;
|
||||
});
|
||||
// errors on server side with readable stream
|
||||
readable.on('error', err => {
|
||||
log.error('error piping data from source');
|
||||
return _next(err);
|
||||
});
|
||||
return readable.pipe(response, { end: false });
|
||||
}), err => {
|
||||
if (err) {
|
||||
log.debug('abort response due to client error', {
|
||||
error: err.code, errMsg: err.message });
|
||||
}
|
||||
// call end for all cases (error/success) per node.js docs
|
||||
// recommendation
|
||||
response.end();
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
function _responseBody(responseBackend, errCode, payload, response, log,
|
||||
|
|
Loading…
Reference in New Issue