geonix 1.10.8 → 1.10.10

package/index.d.ts

@@ -61,7 +61,8 @@ declare class Connection {
 }
 export {};
 export class Gateway {
-    static start(): Gateway;
+    static start(opts: any): Gateway;
+    constructor(opts: any);
     #private;
 }
 export const registry: Registry;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "geonix",
-    "version": "1.10.8",
+    "version": "1.10.10",
     "type": "module",
     "description": "",
     "bin": {

package/src/Gateway.js

@@ -27,12 +27,18 @@ const stats = {
     debug_requests: 0
 }
 
+const defaultOpts = {
+    afterRequest: (req, res) => { },
+}
+
+
 export class Gateway {
 
-    static start() {
-        return new Gateway()
+    static start(opts) {
+        return new Gateway(opts)
     }
 
+    #opts = defaultOpts
     #api = express()
     #router = (req, res, next) => next()
     #activeServers = []
@@ -44,9 +50,11 @@ export class Gateway {
 
     #registry = {}
 
-    constructor() {
+    constructor(opts) {
         expressWs(this.#api)
 
+        this.#opts = { ...this.#opts, ...opts }
+
         this.#start()
     }
 
@@ -92,6 +100,13 @@ export class Gateway {
                 next()
         })
 
+        this.#api.use((req, res, next) => {
+            if (this.#opts.afterRequest) {
+                this.#opts.afterRequest(req, res)
+            }
+            next()
+        })
+
         // config
         this.#api.disable('x-powered-by')
         this.#api.disable('etag')

package/test/package.json

@@ -0,0 +1,16 @@
+{
+    "name": "test",
+    "version": "1.0.0",
+    "description": "",
+    "type": "module",
+    "main": "stream.js",
+    "scripts": {
+        "test": "echo \"Error: no test specified\" && exit 1"
+    },
+    "keywords": [],
+    "author": "",
+    "license": "ISC",
+    "dependencies": {
+        "geonix": "file:.."
+    }
+}

package/test/stream.js

@@ -0,0 +1,38 @@
+import { randomBytes } from 'node:crypto'
+import { Stream, getReadable, connection } from 'geonix'
+import { createWriteStream, readFileSync } from 'node:fs'
+import { createHash } from 'node:crypto'
+import { pipeline } from 'node:stream/promises'
+
+await connection.waitUntilReady()
+
+const hash = data => createHash('sha512').update(data).digest('base64')
+
+const PAYLOAD_SIZE = 1024 * 1024 * 1024
+const TEMP_FILE = '/tmp/geonix.stream_test'
+
+console.time('test')
+try {
+    const payload = randomBytes(PAYLOAD_SIZE)
+    const sourceHash = hash(payload)
+
+    const source = await getReadable(Stream(payload))
+    const dest = createWriteStream(TEMP_FILE)
+
+    await pipeline(source, dest)
+
+    const check = readFileSync(TEMP_FILE)
+    const destHash = hash(check)
+
+    if (sourceHash == destHash) {
+        console.log('MATCH')
+    } else {
+        console.error('Destination does not match the source!')
+        console.log(payload)
+        console.log(check)
+    }
+} catch (e) {
+    console.error(e)
+} finally {
+    console.timeEnd('test')
+}