feat: init for open api

unit-mesh · Aug 15, 2023 · 314db4b · 314db4b
1 parent 1cec4b3
commit 314db4b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -80,7 +80,6 @@ ArchGuard APIs:
 
 GET http://127.0.0.1:8765/api/query?q=create%20arch%20system&repo_ref=archguard&query_type=HttpApi
 
-
 ### 3. TEXT EMBEDDING
 
 GET http://127.0.0.1:8765/api/text-embedding?q=create%20arch%20system

diff --git a/counit-server/src/repository/payload.rs b/counit-server/src/repository/payload.rs
@@ -42,6 +42,7 @@ pub enum PayloadType {
     Comment,
     Doc,
     HttpApi,
+    OpenApi,
 }
 
 impl Default for PayloadType {
@@ -57,6 +58,7 @@ impl PayloadType {
             "comment" => PayloadType::Comment,
             "doc" => PayloadType::Doc,
             "http_api" => PayloadType::HttpApi,
+            "open_api" => PayloadType::OpenApi,
             _ => PayloadType::Code,
         }
     }
@@ -69,6 +71,7 @@ impl std::fmt::Display for PayloadType {
             PayloadType::Comment => write!(f, "comment"),
             PayloadType::Doc => write!(f, "doc"),
             PayloadType::HttpApi => write!(f, "http_api"),
+            PayloadType::OpenApi => write!(f, "open_api"),
         }
     }
 }
@@ -80,6 +83,7 @@ impl Into<Value> for PayloadType {
             PayloadType::Comment => Value::from("comment"),
             PayloadType::Doc => Value::from("doc"),
             PayloadType::HttpApi => Value::from("http_api"),
+            PayloadType::OpenApi => Value::from("open_api"),
         }
     }
 }

diff --git a/counit-server/src/server/archguard_api.rs b/counit-server/src/server/archguard_api.rs
@@ -53,7 +53,7 @@ pub async fn save_openapi(
                                 params.path.as_str(),
                                 item.display_text.as_str(),
                                 params.language.as_str(),
-                                PayloadType::HttpApi,
+                                PayloadType::OpenApi,
                                 item.display_text.as_str(),
                             ).await;
                         });