0.0.8 release

goyounha11 · Jul 23, 2024 · 7773ce0 · 7773ce0
1 parent ec2a78f
commit 7773ce0
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/main/kotlin/com/goyounha11/docs/DocsUtil.kt b/src/main/kotlin/com/goyounha11/docs/DocsUtil.kt
@@ -38,7 +38,9 @@ object DocsUtil {
         description: String,
         resultActions: ResultActions,
         requestClazz: Class<*>? = null,
-        responseClazz: Class<*>? = null
+        responseClazz: Class<*>? = null,
+        requestSchema: String? = null,
+        responseSchema: String? = null
     ): RestDocumentationResultHandler {
         val resourceSnippetParametersBuilder =
             ResourceSnippetParametersBuilder().tags(tag).description(description)
@@ -55,6 +57,16 @@ object DocsUtil {
             response.contentAsString.let { jacksonObjectMapper().readTree(it) }
         val responseFieldDescriptors = createFieldDescriptors(responseNode, responseClazz?.kotlin)
 
+        resourceSnippetParametersBuilder.apply {
+            requestSchema?.let { schema ->
+                requestSchema(Schema(schema))
+            }
+
+            responseSchema?.let { schema ->
+                responseSchema(Schema(schema))
+            }
+        }
+
         val requestParameter = createParameters(request, ParameterType.QUERY)
         val requestPathParameter = createParameters(request, ParameterType.PATH)