fix: allow ref for query param fields

theoomoregbee · Aug 29, 2019 · 3dfc2f1 · 3dfc2f1
1 parent 78c482f
commit 3dfc2f1
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/lib/generators.js b/lib/generators.js
@@ -367,6 +367,7 @@ function _getPathParamsFromKeys(keys) {
     return params;
 }
 
+
 /**
  * this is used to create param for custom controllers
  * @param parameters
@@ -405,13 +406,20 @@ function _getQueryParamsFromRouteQuery(queries) {
     var params = [];
 
     _.forEach(queries, function (query) {
-        params.push({
-            in: 'query',
-            name: query,
-            required: false,
-            type: 'string',
-            description: 'This is a query param for ' + query
-        });
+      let param
+      if(typeof query !== 'string' && query.$ref){
+        param = query
+      }else{
+        param = {
+          in: 'query',
+          name: query,
+          required: false,
+          type: 'string',
+          description: 'This is a query param for ' + query
+        }
+      }
+
+        params.push(param);
     });
 
     return params;