build: ✨ create customized metakg loading error

biothings · Feb 19, 2021 · caef894 · caef894
1 parent 45a3926
commit caef894
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 3 deletions.
diff --git a/src/middlewares/error.js b/src/middlewares/error.js
@@ -1,6 +1,7 @@
 const swaggerValidation = require('./validate');
 const InvalidQueryGraphError = require("../utils/errors/invalid_query_graph_error");
 const PredicatesLoadingError = require('../utils/errors/predicates_error');
+const MetaKGLoadingError = require("../utils/errors/metakg_error");
 
 class ErrorHandler {
     setRoutes(app) {
@@ -23,6 +24,13 @@ class ErrorHandler {
                     more_info: error.message
                 })
             }
+
+            if (error instanceof MetaKGLoadingError) {
+                return res.status(404).json({
+                    error: "Unable to load metakg",
+                    more_info: error.message
+                })
+            }
             if (!error.statusCode) error.statusCode = 500;
 
             if (error.statusCode === 301) {

diff --git a/src/routes/metakg.js b/src/routes/metakg.js
@@ -1,9 +1,10 @@
 const utils = require("../utils/common");
 const assoc = require("../controllers/association");
+const MetaKGLoadingError = require("../utils/errors/metakg_error");
 
 class RouteMetaKG {
     setRoutes(app) {
-        app.get('/metakg', async (req, res) => {
+        app.get('/metakg', async (req, res, next) => {
             try {
                 res.setHeader('Content-Type', 'application/json');
                 let api = undefined, source = undefined;
@@ -16,8 +17,7 @@ class RouteMetaKG {
                 let assocs = await assoc(req.query.subject, req.query.object, req.query.predicate, api, source);
                 res.end(JSON.stringify({ associations: assocs }));
             } catch (error) {
-                console.log(error);
-                res.end();
+                next(new MetaKGLoadingError());
             }
         })
     }

diff --git a/src/utils/errors/metakg_error.js b/src/utils/errors/metakg_error.js
@@ -0,0 +1,11 @@
+class MetaKGLoadingError extends Error {
+    constructor(message = "Failed to load metakg", ...params) {
+        super(...params);
+
+        this.name = 'MetaKGLoadingError';
+        this.message = message;
+        this.statusCode = 400;
+    }
+}
+
+module.exports = MetaKGLoadingError;