完善生成模版，调整目录

2025-12-24 08:12:55 +08:00 · 2025-06-25 01:31:34 +08:00
parent cf4e879fe2
commit 915bcfe22c
35 changed files with 57 additions and 67 deletions
@@ -0,0 +1,138 @@
+package generatorController
+
+import (
+	"x_admin/core/request"
+	"x_admin/core/response"
+	"x_admin/schema/generatorSchema"
+	"x_admin/service/generatorService"
+
+	"net/http"
+	"strings"
+	"x_admin/middleware"
+	"x_admin/util"
+
+	"github.com/gin-gonic/gin"
+)
+
+func GenRoute(rg *gin.RouterGroup) {
+	handle := genHandler{}
+
+	rg = rg.Group("/gen", middleware.TokenAuth())
+	rg.GET("/db", handle.dbTables)
+	rg.GET("/list", handle.List)
+	rg.GET("/detail", handle.Detail)
+	rg.POST("/importTable", handle.importTable)
+	rg.POST("/syncTable", handle.syncTable)
+	rg.POST("/editTable", handle.editTable)
+	rg.POST("/delTable", handle.delTable)
+	rg.GET("/previewCode", handle.previewCode)
+	rg.GET("/downloadCode", handle.downloadCode)
+}
+
+type genHandler struct {
+	// Service IGenerateService
+}
+
+// dbTables 数据表列表
+func (gh genHandler) dbTables(c *gin.Context) {
+	var page request.PageReq
+	var tbReq generatorSchema.DbTablesReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &page)) {
+		return
+	}
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &tbReq)) {
+		return
+	}
+	res, err := generatorService.GenerateService.DbTables(page, tbReq)
+	response.CheckAndRespWithData(c, res, err)
+}
+
+// list 生成列表
+func (gh genHandler) List(c *gin.Context) {
+	var page request.PageReq
+	var listReq generatorSchema.ListTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &page)) {
+		return
+	}
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &listReq)) {
+		return
+	}
+	res, err := generatorService.GenerateService.List(page, listReq)
+	response.CheckAndRespWithData(c, res, err)
+}
+
+// detail 生成详情
+func (gh genHandler) Detail(c *gin.Context) {
+	var detailReq generatorSchema.DetailTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &detailReq)) {
+		return
+	}
+	res, err := generatorService.GenerateService.Detail(detailReq.ID)
+	response.CheckAndRespWithData(c, res, err)
+}
+
+// importTable 导入表结构
+func (gh genHandler) importTable(c *gin.Context) {
+	var importReq generatorSchema.ImportTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &importReq)) {
+		return
+	}
+	err := generatorService.GenerateService.ImportTable(strings.Split(importReq.Tables, ","))
+	response.CheckAndResp(c, err)
+}
+
+// syncTable 同步表结构
+func (gh genHandler) syncTable(c *gin.Context) {
+	var syncReq generatorSchema.SyncTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &syncReq)) {
+		return
+	}
+	err := generatorService.GenerateService.SyncTable(syncReq.ID)
+	response.CheckAndResp(c, err)
+}
+
+// editTable 编辑表结构
+func (gh genHandler) editTable(c *gin.Context) {
+	var editReq generatorSchema.EditTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyJSON(c, &editReq)) {
+		return
+	}
+	err := generatorService.GenerateService.EditTable(editReq)
+	response.CheckAndResp(c, err)
+}
+
+// delTable 删除表结构
+func (gh genHandler) delTable(c *gin.Context) {
+	var delReq generatorSchema.DelTableReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyJSON(c, &delReq)) {
+		return
+	}
+	err := generatorService.GenerateService.DelTable(delReq.Ids)
+	response.CheckAndResp(c, err)
+}
+
+// previewCode 预览代码
+func (gh genHandler) previewCode(c *gin.Context) {
+	var previewReq generatorSchema.PreviewCodeReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &previewReq)) {
+		return
+	}
+	res, err := generatorService.GenerateService.PreviewCode(previewReq.ID)
+	response.CheckAndRespWithData(c, res, err)
+}
+
+// downloadCode 下载代码
+func (gh genHandler) downloadCode(c *gin.Context) {
+	var downloadReq generatorSchema.DownloadReq
+	if response.IsFailWithResp(c, util.VerifyUtil.VerifyQuery(c, &downloadReq)) {
+		return
+	}
+	zipBytes, err := generatorService.GenerateService.DownloadCode(strings.Split(downloadReq.Tables, ","))
+	if response.IsFailWithResp(c, err) {
+		return
+	}
+	contentType := "application/zip"
+	c.Header("Content-Type", contentType)
+	c.Header("Content-Disposition", "attachment; filename=gen-"+downloadReq.Tables+".zip")
+	c.Data(http.StatusOK, contentType, zipBytes)
+}