Merge pull request #3094 from Countly/hotfix/populator-push

[push] Missing files

Author: iartem

plugins/push/api/jobs/util/do_finish.js

@@ -0,0 +1,45 @@
+const { DoFlush } = require('./do_flush');
+
+/**
+ * Transform which only closes once syn frame is received 
+ */
+class DoFinish extends DoFlush {
+
+    /**
+     * Constructor
+     * 
+     * @param {object} opts transform options
+     */
+    constructor(opts) {
+        super(opts);
+    }
+
+    /**
+     * A method to be overridden
+     * 
+     * @param {function} callback finish callback
+     */
+    do_final(/*callback*/) {
+        throw new Error('Must be overridden');
+    }
+
+    /**
+     * Finish the stream by flushing it and then delegating the finish job to do_final() method
+     * 
+     * @param {function} callback finish callback
+     */
+    _final(callback) {
+        this.log.d('finishing');
+        this.do_final(finishErr => {
+            if (finishErr) {
+                this.log.w('finish error', finishErr);
+            }
+            else {
+                this.log.d('finished successfully');
+            }
+            callback(finishErr);
+        });
+    }
+}
+
+module.exports = { DoFinish };

plugins/push/api/jobs/util/do_flush.js

@@ -0,0 +1,44 @@
+const { Transform } = require('stream');
+
+/**
+ * Transform which only closes once syn frame is received 
+ */
+class DoFlush extends Transform {
+
+    /**
+     * Constructor
+     * 
+     * @param {object} opts transform options
+     */
+    constructor(opts) {
+        super(opts);
+    }
+
+    /**
+     * A method to be overridden
+     * @param {function} callback flush callback
+     */
+    do_flush(/*callback*/) {
+        throw new Error('Must be overridden');
+    }
+
+    /**
+     * Flush the stream delegating the job to do_flush() method
+     * 
+     * @param {function} callback flush callback
+     */
+    _flush(callback) {
+        this.log.d('flushing');
+        this.do_flush(err => {
+            if (err) {
+                this.log.w('flush error', err);
+            }
+            else {
+                this.log.d('flushed successfully');
+            }
+            callback(err);
+        });
+    }
+}
+
+module.exports = { DoFlush };