add registry service

discovery/src/controller/index.ts

@@ -1,3 +1,4 @@
 export { default as system } from './system';
 export { default as package } from './package';
-export { default as pretrained } from './pretrained';
+export { default as pretrained } from './pretrained';
+export { default as registry } from './registry';

discovery/src/controller/package.ts

@@ -4,7 +4,7 @@ import { ImportPackageRequest } from './entity';
 export default class PackageController {
     public static async importPackage (ctx: Context) {
         const toImportPackage: Package = new Package();
-        
+
         console.log(toImportPackage);
     }
 }

discovery/src/controller/registry.ts

@@ -0,0 +1,15 @@
+import { Context } from 'koa';
+import { addRegistry } from '../dynamo/action';
+
+export default class RegistryController {
+    public static async putRegistry (ctx: Context) {
+        const body = ctx.request.body;
+        addRegistry(body.name, body.urlPrefix);
+        ctx.status = 200;
+        ctx.body = {
+            'code': 200,
+            'desc': 'success',
+            'req': body
+        };
+    }
+}

discovery/src/dynamo/action.ts

@@ -1,5 +1,5 @@
 import mapper from './dynamo';
-import { Package, Pretrained } from './entity';
+import { Package, Pretrained, Registry } from './entity';
 import { Guid } from 'guid-typescript';
 import { isSymbol } from 'util';
 
@@ -54,8 +54,20 @@ async function getPretrained () {
     return results;
 }
 
+async function addRegistry (name: string, urlPrefix: string) {
+    const toAddRegistry = Object.assign(new Registry, {
+        id: Guid.create(),
+        name: name,
+        urlPrefix: urlPrefix,
+    });
+    mapper.put(toAddRegistry).then(objectSaved => {
+        console.log(objectSaved);
+    });
+}
+
 export {
     putPretrained,
     putPackage,
-    getPretrained
+    getPretrained,
+    addRegistry
 };

discovery/src/dynamo/entity.ts

@@ -38,7 +38,23 @@ class Pretrained {
     name: string;
 }
 
+@table('cvpm-registry')
+class Registry {
+    @hashKey()
+    id: string;
+
+    @rangeKey({defaultProvider: () => new Date()})
+    createdAt: Date;
+
+    @attribute()
+    name: string;
+
+    @attribute()
+    urlPrefix: string;
+}
+
 export {
     Package,
-    Pretrained
+    Pretrained,
+    Registry
 };

discovery/src/logging.ts

@@ -43,7 +43,7 @@ export function logger(winstonInstance: any) {
                 new Logsene({
                     token: process.env.LOGSENE_TOKEN,
                     type: 'discovery',
-                    level:'error',
+                    level: 'error',
                 })
             ]
         });

discovery/src/routes.ts

@@ -5,8 +5,12 @@ import controller = require('./controller');
 const router = new Router();
 
 router.get('/system/status', controller.system.getSystemStatus);
+
 router.post('/packages', controller.package.importPackage);
+
 router.put('/pretrained', controller.pretrained.importPretrained);
 router.get('/pretrained', controller.pretrained.getAllPretrained);
 
+router.put('/registry', controller.registry.putRegistry);
+
 export { router };

discovery/src/server.ts

@@ -1,6 +1,6 @@
 import * as cors from '@koa/cors';
 import * as Koa from 'koa';
-import * as bodyParser from 'koa-bodyparser';
+import bodyParser from 'koa-bodyparser-ts';
 import * as helmet from 'koa-helmet';
 import * as winston from 'winston';
 

discovery/tsconfig.json

@@ -10,7 +10,7 @@
         "emitDecoratorMetadata": true,
         "moduleResolution": "node",
         "typeRoots": [
-            "node_modules/koa-bodyparser",
+            "node_modules/koa-bodyparser-ts",
             "node_modules/**/*",
             "node_modules/@types"
         ],