@miso.ai/doggoganger 0.9.0-beta.15 → 0.9.0-beta.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@miso.ai/doggoganger",
-  "version": "0.9.0-beta.15",
+  "version": "0.9.0-beta.16",
   "description": "A dummy miso endpoint for demo and testing",
   "type": "module",
   "bin": {

package/src/index.js

@@ -5,13 +5,12 @@ import serveStatic from 'koa-static';
 import { koaBody } from 'koa-body';
 import _route from './route/index.js';
 import Api from './api/index.js';
-import { exclusion, delay } from './utils.js';
-import { gaussRandom } from './data/utils.js';
+import { exclusion } from './utils.js';
 
 export default function doggoganger({ port = 9901, serve = false, ...options } = {}) {
   const app = new Koa();
   const router = new Router();
-  const api = _route(new Api(options));
+  const api = _route(new Api(options), options);
   
   router.use('/api', api.routes(), api.allowedMethods());
   router.use('/v1', api.routes(), api.allowedMethods());
@@ -34,7 +33,6 @@ export default function doggoganger({ port = 9901, serve = false, ...options } =
       },
     }))
     .use(handleAllPath(options))
-    .use(simulateLatency(options))
     .use(router.routes())
     .use(router.allowedMethods())
     .use(handleUnrecognizedPath(options))
@@ -48,16 +46,6 @@ function handleAllPath({ verbose } = {}) {
   };
 }
 
-function simulateLatency({ min = 200, max = 2000, verbose } = {}) {
-  // TODO: only apply to API calls
-  return async (_, next) => {
-    const time = ((min + max) + (gaussRandom() * (max - min))) / 2;
-    verbose && console.log(`Add latency: ${time}ms`);
-    await delay(time);
-    await next();
-  };
-}
-
 function handleUnrecognizedPath({ verbose } = {}) {
   return async (ctx, next) => {
     verbose && console.log(`Unrecognized path: ${ctx.method} ${ctx.url}`);

package/src/middleware/error.js

@@ -0,0 +1,29 @@
+import { trimObj } from '../utils.js';
+
+const DEFAULT_OPTIONS = {
+  rate: 0,
+  verbose: false,
+};
+
+export default function error(options) {
+  const globalOptions = {
+    ...DEFAULT_OPTIONS,
+    ...trimObj(options),
+  };
+  return async (ctx, next) => {
+    const { rate, verbose } = {
+      ...globalOptions, 
+      ...getOptionsFromCtx(ctx),
+    };
+    if (rate && Math.random() < rate) {
+      verbose && console.log(`Simulate error`);
+      throw new Error();
+    }
+    await next();
+  };
+}
+
+function getOptionsFromCtx(ctx) {
+  const rate = Number(ctx.get('x-error-rate')) || undefined;
+  return trimObj({ rate });
+}

package/src/middleware/index.js

@@ -0,0 +1,2 @@
+export { default as latency } from './latency.js';
+export { default as error } from './error.js';

package/src/middleware/latency.js

@@ -0,0 +1,39 @@
+import { delay, trimObj } from '../utils.js';
+import { gaussRandom } from '../data/utils.js';
+
+const DEFAULT_OPTIONS = {
+  enabled: true,
+  verbose: false,
+  min: 200,
+  max: 2000,
+};
+
+export default function latency(options) {
+  const globalOptions = {
+    ...DEFAULT_OPTIONS,
+    ...trimObj(options),
+  };
+  return async (ctx, next) => {
+    const { enabled, min, max, verbose } = {
+      ...globalOptions, 
+      ...getOptionsFromCtx(ctx),
+    };
+    if (enabled) {
+      const time = (min + max) / 2 + gaussRandom() * (max - min) / 6;
+      verbose && console.log(`Add latency: ${time}ms`);
+      await delay(time);
+    }
+    await next();
+  };
+}
+
+function getOptionsFromCtx(ctx) {
+  const latencyStr = ctx.get('x-latency') || undefined;
+  if (latencyStr === '0' || latencyStr === 'false') {
+    return { enabled: false };
+  }
+  const value = Number(latencyStr) || undefined;
+  const min = value || Number(ctx.get('x-latency-min')) || undefined;
+  const max = value || Number(ctx.get('x-latency-max')) || undefined;
+  return trimObj({ min, max });
+}

package/src/route/index.js

@@ -4,14 +4,18 @@ import recommendation from './recommendation.js';
 import search from './search.js';
 import interactions from './interactions.js';
 import products from './products.js';
+import { latency, error } from '../middleware/index.js';
 
 function use(router, path, middleware) {
   router.use(path, middleware.routes(), middleware.allowedMethods());
 }
 
-export default function(api) {
+export default function(api, options) {
   const router = new Router();
 
+  router.use(latency(options.latency));
+  //router.use(error(options.error));
+
   use(router, '/ask', ask(api));
   use(router, '/recommendation', recommendation(api));
   use(router, '/search', search(api));

package/src/route/products.js

@@ -20,5 +20,14 @@ export default function(api) {
     };
   });
 
+  router.get('/_status/:taskId', (ctx) => {
+    ctx.body = {
+      took: 100,
+      errors: false,
+      data: [],
+      code: 200,
+    };
+  });
+
   return router;
 }

package/src/utils.js

@@ -28,3 +28,16 @@ export async function exclusion(ctx, next) {
 export async function delay(time) {
   return new Promise(resolve => setTimeout(resolve, time));
 }
+
+export function trimObj(obj) {
+  if (typeof obj !== 'object') {
+    return obj;
+  }
+  const trimmed = {};
+  for (const k in obj) {
+    if (Object.prototype.hasOwnProperty.call(obj, k) && obj[k] !== undefined) {
+      trimmed[k] = obj[k];
+    }
+  }
+  return trimmed;
+}