WIP: refactoring

Author: jscriptcoder

package-lock.json

@@ -34,10 +34,12 @@
   "homepage": "https://github.com/jscriptcoder/HTTP-vs-WebSockets-performance#readme",
   "dependencies": {
     "axios": "^0.20.0",
+    "chalk": "^4.1.0",
     "node-fetch": "^2.6.1",
     "socket.io-client": "^2.3.0",
     "websocket": "^1.0.32",
-    "ws": "^7.3.1"
+    "ws": "^7.3.1",
+    "yargs": "^16.0.3"
   },
   "devDependencies": {
     "@babel/core": "^7.11.6",

package.json

@@ -0,0 +1,27 @@
+const { PerformanceObserver, performance } = require('perf_hooks')
+
+function performanceObserverCallback(items) {
+    console.log('Timer stopped. Measuring...')
+    console.log(`Duration: ${items.getEntries()[0].duration}`)
+    performance.clearMarks()
+}
+
+export default class PerformanceTimer {
+    startMark = 'START'
+    endMark = 'END'
+
+    constructor() {
+        this.obs = new PerformanceObserver(performanceObserverCallback)
+        this.obs.observe({ entryTypes: ['measure'] })
+    }
+
+    start() {
+        console.log('Timer started...')
+        performance.mark(this.startMark)
+    }
+
+    end() {
+        performance.mark(this.endMark)
+        performance.measure(`${this.startMark} to ${this.endMark}`, this.startMark, this.endMark)
+    }
+}