You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.9 KiB
99 lines
2.9 KiB
package sys
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
"os"
|
|
"pure-admin/global"
|
|
"pure-admin/model"
|
|
"pure-admin/model/response"
|
|
"pure-admin/service"
|
|
"pure-admin/utils"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func PreviewTemp(c *gin.Context) {
|
|
var a model.AutoCodeStruct
|
|
_ = c.ShouldBindJSON(&a)
|
|
if err := utils.Verify(a, utils.AutoCodeVerify); err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
autoCode, err := service.PreviewTemp(a)
|
|
if err != nil {
|
|
global.MG_LOG.Error("预览失败!", zap.Any("err", err))
|
|
response.FailWithMessage("预览失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"autoCode": autoCode}, "预览成功", c)
|
|
}
|
|
}
|
|
|
|
func CreateTemp(c *gin.Context) {
|
|
var a model.AutoCodeStruct
|
|
_ = c.ShouldBindJSON(&a)
|
|
if err := utils.Verify(a, utils.AutoCodeVerify); err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
if a.AutoCreateApiToSql {
|
|
if err := service.AutoCreateApi(&a); err != nil {
|
|
global.MG_LOG.Error("自动化创建失败!请自行清空垃圾数据!", zap.Any("err", err))
|
|
c.Writer.Header().Add("success", "false")
|
|
c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据!"))
|
|
return
|
|
}
|
|
}
|
|
err := service.CreateTemp(a)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
if errors.Is(err, model.AutoMoveErr) {
|
|
c.Writer.Header().Add("success", "false")
|
|
c.Writer.Header().Add("msgtype", "success")
|
|
c.Writer.Header().Add("msg", url.QueryEscape(err.Error()))
|
|
} else {
|
|
c.Writer.Header().Add("success", "false")
|
|
c.Writer.Header().Add("msg", url.QueryEscape(err.Error()))
|
|
_ = os.Remove("./ginvueadmin.zip")
|
|
}
|
|
} else {
|
|
c.Writer.Header().Add("Content-Disposition", fmt.Sprintf("attachment; filename=%s", "ginvueadmin.zip")) // fmt.Sprintf("attachment; filename=%s", filename)对下载的文件重命名
|
|
c.Writer.Header().Add("Content-Type", "application/json")
|
|
c.Writer.Header().Add("success", "true")
|
|
c.File("./ginvueadmin.zip")
|
|
_ = os.Remove("./ginvueadmin.zip")
|
|
}
|
|
}
|
|
|
|
func GetTables(c *gin.Context) {
|
|
dbName := c.DefaultQuery("dbName", "")
|
|
err, tables := service.GetTables(dbName)
|
|
if err != nil {
|
|
global.MG_LOG.Error("查询table失败!", zap.Any("err", err))
|
|
response.FailWithMessage("查询table失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"tables": tables}, "获取成功", c)
|
|
}
|
|
}
|
|
|
|
func GetDB(c *gin.Context) {
|
|
if err, dbs := service.GetDB(); err != nil {
|
|
global.MG_LOG.Error("获取失败!", zap.Any("err", err))
|
|
response.FailWithMessage("获取失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"dbs": dbs}, "获取成功", c)
|
|
}
|
|
}
|
|
|
|
func GetColumn(c *gin.Context) {
|
|
dbName := c.DefaultQuery("dbName", "")
|
|
tableName := c.Query("tableName")
|
|
if err, columns := service.GetColumn(tableName, dbName); err != nil {
|
|
global.MG_LOG.Error("获取失败!", zap.Any("err", err))
|
|
response.FailWithMessage("获取失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"columns": columns}, "获取成功", c)
|
|
}
|
|
}
|
|
|