53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"forge.redroom.link/yves/sqldb"
|
||
|
|
beego "github.com/beego/beego/v2/server/web"
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Operations about schema
|
||
|
|
type SchemaController struct {
|
||
|
|
beego.Controller
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title GetTable
|
||
|
|
// @Description get list table
|
||
|
|
// @Success 200 {string} success !
|
||
|
|
// @Failure 403 no table
|
||
|
|
// @router / [get]
|
||
|
|
func (s *SchemaController) GetTable() {
|
||
|
|
db := sqldb.Open(os.Getenv("driverdb"), os.Getenv("paramsdb"))
|
||
|
|
data, err := db.ListTables()
|
||
|
|
if err != nil {
|
||
|
|
log.Error().Msg(err.Error())
|
||
|
|
s.Data["json"] = map[string]string{"error": err.Error()}
|
||
|
|
}
|
||
|
|
s.Data["json"] = data
|
||
|
|
s.ServeJSON()
|
||
|
|
|
||
|
|
db.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title GetSchema
|
||
|
|
// @Description get table schema
|
||
|
|
// @Param table path string true "Name of the table"
|
||
|
|
// @Success 200 success !
|
||
|
|
// @Failure 403 no table
|
||
|
|
// @router /:table [get]
|
||
|
|
func (s *SchemaController) GetSchema() {
|
||
|
|
table := s.GetString(":table")
|
||
|
|
db := sqldb.Open(os.Getenv("driverdb"), os.Getenv("paramsdb"))
|
||
|
|
data, err := db.Table(table).GetSchema()
|
||
|
|
if err != nil {
|
||
|
|
log.Error().Msg(err.Error())
|
||
|
|
s.Data["json"] = map[string]string{"error": err.Error()}
|
||
|
|
}
|
||
|
|
s.Data["json"] = data
|
||
|
|
s.ServeJSON()
|
||
|
|
|
||
|
|
db.Close()
|
||
|
|
}
|