mirror of
https://gitcode.com/flipped-aurora/gin-vue-admin.git
synced 2025-12-30 11:52:26 +00:00
118 lines
3.6 KiB
Go
118 lines
3.6 KiB
Go
package system
|
|
|
|
import (
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/common"
|
|
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type AutoCodeApi struct{}
|
|
|
|
// GetDB
|
|
// @Tags AutoCode
|
|
// @Summary 获取当前所有数据库
|
|
// @Security ApiKeyAuth
|
|
// @accept application/json
|
|
// @Produce application/json
|
|
// @Success 200 {object} response.Response{data=map[string]interface{},msg=string} "获取当前所有数据库"
|
|
// @Router /autoCode/getDB [get]
|
|
func (autoApi *AutoCodeApi) GetDB(c *gin.Context) {
|
|
businessDB := c.Query("businessDB")
|
|
dbs, err := autoCodeService.Database(businessDB).GetDB(businessDB)
|
|
var dbList []map[string]interface{}
|
|
for _, db := range global.GVA_CONFIG.DBList {
|
|
var item = make(map[string]interface{})
|
|
item["aliasName"] = db.AliasName
|
|
item["dbName"] = db.Dbname
|
|
item["disable"] = db.Disable
|
|
item["dbtype"] = db.Type
|
|
dbList = append(dbList, item)
|
|
}
|
|
if err != nil {
|
|
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
|
response.FailWithMessage("获取失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"dbs": dbs, "dbList": dbList}, "获取成功", c)
|
|
}
|
|
}
|
|
|
|
// GetTables
|
|
// @Tags AutoCode
|
|
// @Summary 获取当前数据库所有表
|
|
// @Security ApiKeyAuth
|
|
// @accept application/json
|
|
// @Produce application/json
|
|
// @Success 200 {object} response.Response{data=map[string]interface{},msg=string} "获取当前数据库所有表"
|
|
// @Router /autoCode/getTables [get]
|
|
func (autoApi *AutoCodeApi) GetTables(c *gin.Context) {
|
|
dbName := c.Query("dbName")
|
|
businessDB := c.Query("businessDB")
|
|
if dbName == "" {
|
|
dbName = *global.GVA_ACTIVE_DBNAME
|
|
if businessDB != "" {
|
|
for _, db := range global.GVA_CONFIG.DBList {
|
|
if db.AliasName == businessDB {
|
|
dbName = db.Dbname
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
tables, err := autoCodeService.Database(businessDB).GetTables(businessDB, dbName)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("查询table失败!", zap.Error(err))
|
|
response.FailWithMessage("查询table失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"tables": tables}, "获取成功", c)
|
|
}
|
|
}
|
|
|
|
// GetColumn
|
|
// @Tags AutoCode
|
|
// @Summary 获取当前表所有字段
|
|
// @Security ApiKeyAuth
|
|
// @accept application/json
|
|
// @Produce application/json
|
|
// @Success 200 {object} response.Response{data=map[string]interface{},msg=string} "获取当前表所有字段"
|
|
// @Router /autoCode/getColumn [get]
|
|
func (autoApi *AutoCodeApi) GetColumn(c *gin.Context) {
|
|
businessDB := c.Query("businessDB")
|
|
dbName := c.Query("dbName")
|
|
if dbName == "" {
|
|
dbName = *global.GVA_ACTIVE_DBNAME
|
|
if businessDB != "" {
|
|
for _, db := range global.GVA_CONFIG.DBList {
|
|
if db.AliasName == businessDB {
|
|
dbName = db.Dbname
|
|
}
|
|
}
|
|
}
|
|
}
|
|
tableName := c.Query("tableName")
|
|
columns, err := autoCodeService.Database(businessDB).GetColumn(businessDB, tableName, dbName)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
|
response.FailWithMessage("获取失败", c)
|
|
} else {
|
|
response.OkWithDetailed(gin.H{"columns": columns}, "获取成功", c)
|
|
}
|
|
}
|
|
|
|
func (autoApi *AutoCodeApi) LLMAuto(c *gin.Context) {
|
|
var llm common.JSONMap
|
|
if err := c.ShouldBindJSON(&llm); err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
data, err := autoCodeService.LLMAuto(c.Request.Context(), llm)
|
|
if err != nil {
|
|
global.GVA_LOG.Error("大模型生成失败!", zap.Error(err))
|
|
response.FailWithMessage("大模型生成失败"+err.Error(), c)
|
|
return
|
|
}
|
|
response.OkWithData(data, c)
|
|
}
|