presidium-db-client-server 0.6.11 → 0.7.1

package/DatabaseLoadBalancer.js

@@ -10,7 +10,7 @@ const notFoundHandler = require("./notFoundHandler")
 const errorHandler = require("./errorHandler")
 
 /**
- * @name DatabaseLoadbalancer
+ * @name DatabaseLoadBalancer
  *
  * @docs
  * ```specscript
@@ -18,16 +18,16 @@ const errorHandler = require("./errorHandler")
  *
  * HTTP.Handler(HTTP.Request, HTTP.Response) -> ()
  *
- * new DatabaseLoadbalancer(
+ * new DatabaseLoadBalancer(
  *   hash(tableName string, key string, length number) -> Promise|number,
  *   options {
- *     serverIps: Array<string>,
+ *     serverIps: Array<string>|Array<Array<string>>,
  *   },
  *   ...handlers Arguments<HTTP.Handler>
  * )
  * ```
  */
-class DatabaseLoadbalancer {
+class DatabaseLoadBalancer {
   constructor(hash, options, ...handlers) {
     this.hash = hash
     this.serverIps = options.serverIps
@@ -94,4 +94,4 @@ class DatabaseLoadbalancer {
 
 }
 
-module.exports = DatabaseLoadbalancer
+module.exports = DatabaseLoadBalancer

package/loadBalancerHandler.js

@@ -1,4 +1,27 @@
 const HTTP = require("presidium/HTTP")
+const Readable = require("presidium/Readable")
+const stream = require("stream")
+
+/**
+ * @name getClient
+ *
+ * @docs
+ * ```specscript
+ * module HTTP "https://presidium.services/docs/HTTP"
+ *
+ * getClient(serverIp string) -> client HTTP
+ * ```
+ */
+function getClient(serverIp) {
+  let client
+  if (this.clients.has(serverIp)) {
+    client = this.clients.get(serverIp)
+  } else {
+    client = new HTTP(serverIp)
+    this.clients.set(serverIp, client)
+  }
+  return client
+}
 
 /**
  * @name loadBalancerHandler
@@ -19,22 +42,54 @@ async function loadBalancerHandler(request, response) {
     index = await index
   }
 
-  const serverIp = this.serverIps[index]
-  let client
-  if (this.clients.has(serverIp)) {
-    client = this.clients.get(serverIp)
+  const serverIpOrIps = this.serverIps[index]
+
+  if (Array.isArray(serverIpOrIps)) {
+    const passthroughStreams = []
+    for (let i = 0; i < serverIpOrIps.length; i++) {
+      passthroughStreams.push(new stream.PassThrough())
+    }
+
+    serverIpOrIps.forEach(async (serverIp, index) => {
+      const client = getClient.call(this, serverIp)
+
+      const serverResponse = await client[request.method](request.url, {
+        headers: request.headers,
+        body: passthroughStreams[index],
+      })
+
+      if (index == 0) {
+        response.writeHead(serverResponse.status, serverResponse.headers)
+        serverResponse.pipe(response)
+      } else {
+        await Readable.Text(serverResponse)
+      }
+    })
+
+    request.on("data", chunk => {
+      passthroughStreams.forEach(stream => {
+        stream.write(chunk)
+      })
+    })
+
+    request.on("end", () => {
+      passthroughStreams.forEach(stream => {
+        stream.end()
+      })
+    })
+
   } else {
-    client = new HTTP(serverIp)
-    this.clients.set(serverIp, client)
-  }
+    const client = getClient.call(this, serverIpOrIps)
 
-  const serverResponse = await client[request.method](request.url, {
-    headers: request.headers,
-    body: request,
-  })
+    const serverResponse = await client[request.method](request.url, {
+      headers: request.headers,
+      body: request,
+    })
+
+    response.writeHead(serverResponse.status, serverResponse.headers)
+    serverResponse.pipe(response)
+  }
 
-  response.writeHead(serverResponse.status, serverResponse.headers)
-  serverResponse.pipe(response)
 }
 
 module.exports = loadBalancerHandler

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "presidium-db-client-server",
-  "version": "0.6.11",
+  "version": "0.7.1",
   "description": "Presidium DB client and server library",
   "author": "Richard Tong",
   "license": "CFOSS",