From 5394698277ada218482568806adab3cc8454bcf5 Mon Sep 17 00:00:00 2001
From: heedaeshin <heedae.shin@namutech.co.kr>
Date: Thu, 22 Aug 2024 18:02:01 +0900
Subject: [PATCH] Add  : MySql Data gen

---
 websrc/controllers/gendata.go          | 10 ++++++++++
 websrc/controllers/generateHandlers.go |  2 +-
 websrc/docs/docs.go                    | 25 ++++++++++++++++++++++++-
 websrc/docs/swagger.json               | 25 ++++++++++++++++++++++++-
 websrc/docs/swagger.yaml               | 17 ++++++++++++++++-
 5 files changed, 75 insertions(+), 4 deletions(-)

diff --git a/websrc/controllers/gendata.go b/websrc/controllers/gendata.go
index dcf61eb..6f77971 100644
--- a/websrc/controllers/gendata.go
+++ b/websrc/controllers/gendata.go
@@ -65,12 +65,22 @@ type GenDataParams struct {
 	GCPCredential *multipart.FileHeader `form:"gcpCredential" swaggerignore:"true"`
 	ProjectID     string                `json:"projectid" form:"projectid"`
 }
+
 type GenFirestoreParams struct {
 	Region        string                `json:"region" form:"region"`
 	GCPCredential *multipart.FileHeader `form:"gcpCredential" swaggerignore:"true"`
 	ProjectID     string                `json:"projectid" form:"projectid"`
 }
 
+type GenMySQLParams struct {
+	DBProvider   string `json:"provider" form:"provider"`
+	DBHost       string `json:"host" form:"host"`
+	DBPort       string `json:"port" form:"port"`
+	DBUser       string `json:"username" form:"username"`
+	DBPassword   string `json:"password" form:"password"`
+	DatabaseName string `json:"databaseName" form:"databaseName"`
+}
+
 func genData(params GenDataParams, logger *logrus.Logger) error {
 	if params.CheckSQL == "on" {
 		logger.Info("Start creating sql dummy")
diff --git a/websrc/controllers/generateHandlers.go b/websrc/controllers/generateHandlers.go
index 7898fb5..5c05db9 100644
--- a/websrc/controllers/generateHandlers.go
+++ b/websrc/controllers/generateHandlers.go
@@ -322,7 +322,7 @@ func GenerateNCPPostHandler(ctx echo.Context) error {
 //	@Tags			[Test Data Generation]
 //	@Accept			json
 //	@Produce		json
-//	@Param			RequestBody	body		GenDataParams			true	"Parameters required to generate test data"
+//	@Param			RequestBody	body		GenMySQLParams			true	"Parameters required to generate test data"
 //	@Success		200			{object}	models.BasicResponse	"Successfully generated test data"
 //	@Failure		500			{object}	models.BasicResponse	"Internal Server Error"
 //	@Router			/generate/mysql [post]
diff --git a/websrc/docs/docs.go b/websrc/docs/docs.go
index 267b7e5..4e9299a 100644
--- a/websrc/docs/docs.go
+++ b/websrc/docs/docs.go
@@ -420,7 +420,7 @@ const docTemplate = `{
                         "in": "body",
                         "required": true,
                         "schema": {
-                            "$ref": "#/definitions/controllers.GenDataParams"
+                            "$ref": "#/definitions/controllers.GenMySQLParams"
                         }
                     }
                 ],
@@ -2671,6 +2671,29 @@ const docTemplate = `{
                 }
             }
         },
+        "controllers.GenMySQLParams": {
+            "type": "object",
+            "properties": {
+                "databaseName": {
+                    "type": "string"
+                },
+                "host": {
+                    "type": "string"
+                },
+                "password": {
+                    "type": "string"
+                },
+                "port": {
+                    "type": "string"
+                },
+                "provider": {
+                    "type": "string"
+                },
+                "username": {
+                    "type": "string"
+                }
+            }
+        },
         "controllers.LinuxMigrationParams": {
             "type": "object",
             "properties": {
diff --git a/websrc/docs/swagger.json b/websrc/docs/swagger.json
index d4d80e1..419d14a 100644
--- a/websrc/docs/swagger.json
+++ b/websrc/docs/swagger.json
@@ -413,7 +413,7 @@
                         "in": "body",
                         "required": true,
                         "schema": {
-                            "$ref": "#/definitions/controllers.GenDataParams"
+                            "$ref": "#/definitions/controllers.GenMySQLParams"
                         }
                     }
                 ],
@@ -2664,6 +2664,29 @@
                 }
             }
         },
+        "controllers.GenMySQLParams": {
+            "type": "object",
+            "properties": {
+                "databaseName": {
+                    "type": "string"
+                },
+                "host": {
+                    "type": "string"
+                },
+                "password": {
+                    "type": "string"
+                },
+                "port": {
+                    "type": "string"
+                },
+                "provider": {
+                    "type": "string"
+                },
+                "username": {
+                    "type": "string"
+                }
+            }
+        },
         "controllers.LinuxMigrationParams": {
             "type": "object",
             "properties": {
diff --git a/websrc/docs/swagger.yaml b/websrc/docs/swagger.yaml
index dc34483..60b053a 100644
--- a/websrc/docs/swagger.yaml
+++ b/websrc/docs/swagger.yaml
@@ -69,6 +69,21 @@ definitions:
       username:
         type: string
     type: object
+  controllers.GenMySQLParams:
+    properties:
+      databaseName:
+        type: string
+      host:
+        type: string
+      password:
+        type: string
+      port:
+        type: string
+      provider:
+        type: string
+      username:
+        type: string
+    type: object
   controllers.LinuxMigrationParams:
     properties:
       path:
@@ -422,7 +437,7 @@ paths:
         name: RequestBody
         required: true
         schema:
-          $ref: '#/definitions/controllers.GenDataParams'
+          $ref: '#/definitions/controllers.GenMySQLParams'
       produces:
       - application/json
       responses: