From 693f069c48f3bd03096b6a4be6b6f29b3a7e323e Mon Sep 17 00:00:00 2001
From: tanghc <thc8775@163.com>
Date: Thu, 22 Aug 2019 20:11:08 +0800
Subject: [PATCH] 2.0

---
 .../sop/websiteserver/manager/SwaggerDocParser.java | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/sop-website/src/main/java/com/gitee/sop/websiteserver/manager/SwaggerDocParser.java b/sop-website/src/main/java/com/gitee/sop/websiteserver/manager/SwaggerDocParser.java
index 7435df42..c75e22bf 100644
--- a/sop-website/src/main/java/com/gitee/sop/websiteserver/manager/SwaggerDocParser.java
+++ b/sop-website/src/main/java/com/gitee/sop/websiteserver/manager/SwaggerDocParser.java
@@ -203,7 +203,18 @@ public class SwaggerDocParser implements DocParser {
         String ref = Optional.ofNullable(docInfo.getJSONObject("responses"))
                 .flatMap(jsonObject -> Optional.ofNullable(jsonObject.getJSONObject("200")))
                 .flatMap(jsonObject -> Optional.ofNullable(jsonObject.getJSONObject("schema")))
-                .flatMap(jsonObject -> Optional.ofNullable(jsonObject.getString("originalRef")))
+                .flatMap(jsonObject -> {
+                    // #/definitions/Category
+                    String $ref = jsonObject.getString("$ref");
+                    if ($ref == null) {
+                        return Optional.empty();
+                    }
+                    int index = $ref.lastIndexOf("/");
+                    if (index > -1) {
+                        $ref = $ref.substring(index + 1);
+                    }
+                    return Optional.of($ref);
+                })
                 .orElse("");
         return ref;
     }