From 17ea6975f9e4d9a25db634f39391e4eef3267709 Mon Sep 17 00:00:00 2001
From: hrq <f00lish@qq.com>
Date: Fri, 30 Jul 2021 17:03:04 +0800
Subject: [PATCH] =?UTF-8?q?fix=20swagger=20=E5=85=A8=E5=B1=80header?=
 =?UTF-8?q?=E5=8F=82=E6=95=B0=E6=97=B6=E5=8F=AF=E8=83=BD=E4=BC=9A=E5=A4=9A?=
 =?UTF-8?q?=E8=A7=A3=E6=9E=90=E6=88=90query=E5=8F=82=E6=95=B0=E4=B8=80?=
 =?UTF-8?q?=E4=BB=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cn/torna/service/DocImportService.java    | 34 +++++++++++--------
 1 file changed, 19 insertions(+), 15 deletions(-)

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 680a712f..1bee124b 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<DocParameter> queryParameters = item.getQueryParameters();
-                this.saveParams(queryParameters, docItem, docParameter -> ParamStyleEnum.QUERY, user);
+                this.saveParams(queryParameters, docItem, this::buildStyleEnum, user);
                 // body参数
                 List<DocParameter> 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<DocParameter> 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;
+        }
+    }
+
 }
-- 
Gitee