diff --git a/server/server-service/src/main/java/cn/torna/service/DocImportService.java b/server/server-service/src/main/java/cn/torna/service/DocImportService.java index 680a712f2507d2a788002c718c217899440fb210..1bee124b79f2b5a011089d576647446c19afb185 100644 --- a/server/server-service/src/main/java/cn/torna/service/DocImportService.java +++ b/server/server-service/src/main/java/cn/torna/service/DocImportService.java @@ -288,23 +288,10 @@ public class DocImportService { DocInfo docItem = docInfoService.createDocItem(docItemCreateDTO); // query参数 List queryParameters = item.getQueryParameters(); - this.saveParams(queryParameters, docItem, docParameter -> ParamStyleEnum.QUERY, user); + this.saveParams(queryParameters, docItem, this::buildStyleEnum, user); // body参数 List requestParameters = item.getRequestParameters(); - this.saveParams(requestParameters, docItem, docParameter -> { - String in = ((DocParameter) docParameter).getIn(); - if (in == null) { - in = "request"; - } - switch (in) { - case "path": - return ParamStyleEnum.PATH; - case "header": - return ParamStyleEnum.HEADER; - default: - return ParamStyleEnum.REQUEST; - } - }, user); + this.saveParams(requestParameters, docItem, this::buildStyleEnum, user); List responseParameters = item.getResponseParameters(); this.saveParams(responseParameters, docItem, p -> ParamStyleEnum.RESPONSE, user); } @@ -435,4 +422,21 @@ public class DocImportService { } } + private ParamStyleEnum buildStyleEnum(IParam docParameter) { + String in = ((DocParameter)docParameter).getIn(); + if (in == null) { + in = "request"; + } + switch (in) { + case "path": + return ParamStyleEnum.PATH; + case "query": + return ParamStyleEnum.QUERY; + case "header": + return ParamStyleEnum.HEADER; + default: + return ParamStyleEnum.REQUEST; + } + } + }