func (c *ChgPass) Get() error {
engines, err := models.FindEngines()
if err != nil {
return err
}
return c.Render("chgpass.html", renders.T{
"XsrfFormHtml": c.XsrfFormHtml(),
"engines": engines,
"flash": c.Flash.Data(),
"IsLogin": c.IsLogin(),
})
}
func (c *Addb) Get() error {
engines, err := models.FindEngines()
if err != nil {
return err
}
return c.Render("add.html", renders.T{
"dbs": SupportDBs,
"flash": c.Flash.Data(),
"engines": engines,
"XsrfFormHtml": c.XsrfFormHtml(),
"IsLogin": c.IsLogin(),
"isAdd": true,
})
}
func (c *Home) Get() error {
engines, err := models.FindEngines()
if err != nil {
return err
}
return c.Render("root.html", renders.T{
"engines": engines,
"tables": []core.Table{},
"records": [][]string{},
"columns": []string{},
"id": 0,
"ishome": true,
"IsLogin": c.IsLogin(),
})
}
func (c *Del) Get() error {
id, err := strconv.ParseInt(c.Req().FormValue("id"), 10, 64)
if err != nil {
return err
}
if err := models.DelEngineById(id); err != nil {
return err
}
engines, err := models.FindEngines()
if err != nil {
return err
}
return c.Render("delsuccess.html", renders.T{
"engines": engines,
"IsLogin": c.IsLogin(),
})
}
func (c *View) Get() error {
id, err := strconv.ParseInt(c.Req().FormValue("id"), 10, 64)
if err != nil {
return err
}
engine, err := models.GetEngineById(id)
if err != nil {
return err
}
o := GetOrm(engine)
if o == nil {
return fmt.Errorf("get engine %s failed", engine.Name)
}
tables, err := o.DBMetas()
if err != nil {
return err
}
var records = make([][]*string, 0)
var columns = make([]*core.Column, 0)
tb := c.Req().FormValue("tb")
tb = strings.Replace(tb, `"`, "", -1)
tb = strings.Replace(tb, `'`, "", -1)
tb = strings.Replace(tb, "`", "", -1)
var isTableView = len(tb) > 0
sql := c.Req().FormValue("sql")
var table *core.Table
var pkIdx int
var isExecute bool
var affected int64
var total int
var countSql string
var args = make([]interface{}, 0)
start, _ := strconv.Atoi(c.Req().FormValue("start"))
limit, _ := strconv.Atoi(c.Req().FormValue("limit"))
if limit == 0 {
limit = 20
}
if sql != "" || tb != "" {
if sql != "" {
isExecute = !strings.HasPrefix(strings.ToLower(sql), "select")
} else if tb != "" {
countSql = "select count(*) from `" + tb + "`"
sql = fmt.Sprintf("select * from `"+tb+"` LIMIT %d OFFSET %d", limit, start)
//args = append(args, []interface{}{limit, start}...)
} else {
return errors.New("unknow operation")
}
if isExecute {
res, err := o.Exec(sql)
if err != nil {
return err
}
affected, _ = res.RowsAffected()
} else {
if len(countSql) > 0 {
err = o.DB().QueryRow(countSql).Scan(&total)
if err != nil {
return err
}
fmt.Println("total records:", total)
}
rows, err := o.DB().Query(sql, args...)
if err != nil {
return err
}
defer rows.Close()
cols, err := rows.Columns()
if err != nil {
return err
}
if len(tb) > 0 {
for _, tt := range tables {
if tb == tt.Name {
table = tt
break
}
}
if table != nil {
for i, col := range cols {
c := table.GetColumn(col)
if len(table.PKColumns()) == 1 && c.IsPrimaryKey {
pkIdx = i
}
columns = append(columns, c)
}
}
} else {
for _, col := range cols {
columns = append(columns, &core.Column{
Name: col,
})
}
}
for rows.Next() {
datas := make([]*string, len(columns))
err = rows.ScanSlice(&datas)
if err != nil {
return err
}
records = append(records, datas)
}
}
}
engines, err := models.FindEngines()
if err != nil {
return err
}
return c.Render("root.html", renders.T{
"engines": engines,
"tables": tables,
"table": table,
"records": records,
"columns": columns,
"id": id,
"sql": sql,
"tb": tb,
"isExecute": isExecute,
"isTableView": isTableView,
"limit": limit,
"curPage": start / limit,
"totalPage": pager(total, limit),
"affected": affected,
"pkIdx": pkIdx,
"curEngine": engine.Name,
"IsLogin": c.IsLogin(),
})
}