56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package controllers
|
|
|
|
import (
|
|
"net/http"
|
|
"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 GetTablesList
|
|
// @Description Get database tables list
|
|
// @Success 200 {string} success !
|
|
// @Failure 500 query error
|
|
// @router / [get]
|
|
func (s *SchemaController) GetTablesList() {
|
|
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.Ctx.Output.SetStatus(http.StatusInternalServerError)
|
|
}
|
|
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 204 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.Ctx.Output.SetStatus(http.StatusNoContent)
|
|
}
|
|
s.Data["json"] = data
|
|
s.ServeJSON()
|
|
|
|
db.Close()
|
|
}
|