diff --git a/lib/generators.js b/lib/generators.js index 557d466..af6318c 100644 --- 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;