-
Notifications
You must be signed in to change notification settings - Fork 0
/
writer.ts
57 lines (48 loc) · 1.49 KB
/
writer.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
import {Metric, metricToString} from "./metric";
import {MergeMode} from "./merging";
import S3 from 'aws-sdk/clients/s3';
import {MetricKey} from "./metric";
import {Builder, IBuilder} from 'builder-pattern';
/**
* Class that writes metrics into an S3 bucket of your choice.
*/
class Writer {
s3Bucket!: string;
mergeMode!: MergeMode;
s3Client!: S3;
constructor(s3Bucket: string, mergeMode: MergeMode, s3Client: S3) {
this.s3Client = s3Client;
this.s3Bucket = s3Bucket;
this.mergeMode = mergeMode;
}
static builder(): IBuilder<Writer> {
return Builder(Writer, new Writer('N/A', MergeMode.merge, new S3()));
}
/**
* Writes the metric into the S3 bucket. Metrics are located in the bucket
* in a path based on their resolution, then group and finally key. Depending on resolution
* there might then be sub folders created to save the values.
*/
writeMetrics(metricKey: MetricKey, metrics: Metric[]): Promise<any> {
return new Promise((resolve, reject) => {
const key = metricKey.getAwsPrefix() + 'all';
const body =
metrics
.map(metricToString)
.join('\n');
const request: S3.Types.PutObjectRequest = {
Bucket: this.s3Bucket,
Key: key,
Body: body
};
this.s3Client.putObject(request, (err) => {
if(err) {
reject(err);
} else {
resolve();
}
});
});
}
}
export default Writer;