/
util.ts
126 lines (113 loc) · 3.67 KB
/
util.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/*
* Copyright The OpenTelemetry Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import * as url from 'url';
import * as http from 'http';
import * as https from 'https';
import * as zlib from 'zlib';
import { pipeline, Readable } from 'stream';
import * as collectorTypes from '../../types';
import { CollectorExporterNodeBase } from './CollectorExporterNodeBase';
import { CollectorExporterNodeConfigBase } from '.';
import { diag } from '@opentelemetry/api';
const gzip = zlib.createGzip();
/**
* Sends data using http
* @param collector
* @param data
* @param contentType
* @param onSuccess
* @param onError
*/
export function sendWithHttp<ExportItem, ServiceRequest>(
collector: CollectorExporterNodeBase<ExportItem, ServiceRequest>,
data: string | Buffer,
contentType: string,
compress: boolean,
onSuccess: () => void,
onError: (error: collectorTypes.CollectorExporterError) => void
): void {
const parsedUrl = new url.URL(collector.url);
const options: http.RequestOptions | https.RequestOptions = {
hostname: parsedUrl.hostname,
port: parsedUrl.port,
path: parsedUrl.pathname,
method: 'POST',
headers: {
'Content-Length': Buffer.byteLength(data),
'Content-Type': contentType,
'Content-Encoding': 'gzip',
...collector.headers,
},
agent: collector.agent,
};
const request = parsedUrl.protocol === 'http:' ? http.request : https.request;
const req = request(options, (res: http.IncomingMessage) => {
let responseData = '';
res.on('data', chunk => (responseData += chunk));
res.on('end', () => {
if (res.statusCode && res.statusCode < 299) {
diag.debug(`statusCode: ${res.statusCode}`, responseData);
onSuccess();
} else {
const error = new collectorTypes.CollectorExporterError(
res.statusMessage,
res.statusCode,
responseData
);
onError(error);
}
});
});
req.on('error', (error: Error) => {
onError(error);
});
if (compress) {
const dataStream = Readable.from(data);
pipeline(dataStream, gzip, req, onGzipError(onError));
} else {
req.write(data);
req.end();
}
}
function onGzipError(onError: (error: collectorTypes.CollectorExporterError) => void) {
return (err: NodeJS.ErrnoException | null) => {
const error = new collectorTypes.CollectorExporterError(
err?.message,
500,
'Compressing the request body for collector exporter failed.'
);
onError(error)
}
}
export function createHttpAgent(
config: CollectorExporterNodeConfigBase
): http.Agent | https.Agent | undefined {
if (config.httpAgentOptions && config.keepAlive === false) {
diag.warn('httpAgentOptions is used only when keepAlive is true');
return undefined;
}
if (config.keepAlive === false || !config.url) return undefined;
try {
const parsedUrl = new url.URL(config.url as string);
const Agent = parsedUrl.protocol === 'http:' ? http.Agent : https.Agent;
return new Agent({ keepAlive: true, ...config.httpAgentOptions });
} catch (err) {
diag.error(
`collector exporter failed to create http agent. err: ${err.message}`
);
return undefined;
}
}