Skip to content

Commit 51a8342

Browse files
AshCrippsAshCripps
AshCripps
authored and
AshCripps
committed
update for google cloud function
1 parent fa793e8 commit 51a8342

File tree

2 files changed

+33
-14
lines changed

2 files changed

+33
-14
lines changed

ansible/roles/metrics/files/process-cloudflare/package.json

+3-1
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,13 @@
44
"description": "",
55
"main": "process-cloudflare.js",
66
"scripts": {
7-
"test": "echo \"Error: no test specified\" && exit 1"
7+
"test": "echo \"Error: no test specified\" && exit 1",
8+
"processLogs": "node -e 'require(\"./process-cloudflare.js\").processLogs()'"
89
},
910
"author": "Rod <[email protected]> (http://r.va.gg/)",
1011
"license": "Apache-2.0",
1112
"dependencies": {
13+
"@google-cloud/storage": "^4.7.0",
1214
"split2": "~3.1.1",
1315
"strftime": "~0.10.0"
1416
}

ansible/roles/metrics/files/process-cloudflare/process-cloudflare.js

+30-13
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,9 @@
33
const { pipeline, Transform } = require('stream')
44
const split2 = require('split2')
55
const strftime = require('strftime').timezone(0)
6+
const {Storage} = require('@google-cloud/storage');
7+
8+
const storage = new Storage({keyFilename: "metrics-processor-service-key.json"});
69

710
const jsonStream = new Transform({
811
readableObjectMode: true,
@@ -140,16 +143,30 @@ const logTransformStream = new Transform({
140143
}
141144
})
142145

143-
pipeline(
144-
process.stdin,
145-
split2(),
146-
jsonStream,
147-
logTransformStream,
148-
process.stdout,
149-
(err) => {
150-
if (err) {
151-
console.error('ERROR', err)
152-
process.exit(1)
153-
}
154-
}
155-
)
146+
147+
exports.processLogs = async () => {
148+
const file = data;
149+
bucketName = file.bucket;
150+
fileName = file.name;
151+
console.log("DATA " + data);
152+
console.log("BUCKET " + bucketName);
153+
console.log("FILENAME " + fileName);
154+
155+
processedFile = fileName.split(".")[0];
156+
processedFile = processedFile.split("_")[0].concat("_", processedFile.split("_")[1]);
157+
158+
console.log("PROCESSEDFILENAME " + processedFile);
159+
160+
storage.bucket(bucketName).file(file.name).createReadStream()
161+
.on('error', function(err) { console.error(err) })
162+
.pipe(split2())
163+
.pipe(jsonStream)
164+
.pipe(logTransformStream)
165+
.pipe(storage.bucket('processed-logs-nodejs').file(processedFile).createWriteStream({resumable: false, flags: 'a'})
166+
.on("error", err => {
167+
console.log("ERROR: >> ", err)
168+
})
169+
.on("finished", () => {
170+
console.log("FINISHED")
171+
}));
172+
}

0 commit comments

Comments
 (0)