mirror of
https://github.com/cuigh/swirl
synced 2024-12-28 14:51:57 +00:00
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/cuigh/auxo/data"
|
|
"github.com/cuigh/auxo/errors"
|
|
"github.com/cuigh/auxo/ext/times"
|
|
"github.com/cuigh/auxo/net/web"
|
|
"github.com/cuigh/swirl/biz"
|
|
"github.com/cuigh/swirl/dao"
|
|
"github.com/cuigh/swirl/misc"
|
|
)
|
|
|
|
// DashboardHandler encapsulates dashboard related handlers.
|
|
type DashboardHandler struct {
|
|
Find web.HandlerFunc `path:"/find" auth:"?" desc:"find dashboard by name and key"`
|
|
Save web.HandlerFunc `path:"/save" method:"post" auth:"dashboard.edit" desc:"save dashboard"`
|
|
FetchData web.HandlerFunc `path:"/fetch-data" auth:"?" desc:"fetch data of dashboard charts"`
|
|
}
|
|
|
|
// NewDashboard creates an instance of DashboardHandler
|
|
func NewDashboard(b biz.DashboardBiz) *DashboardHandler {
|
|
return &DashboardHandler{
|
|
Find: dashboardFind(b),
|
|
Save: dashboardSave(b),
|
|
FetchData: dashboardFetchData(b),
|
|
}
|
|
}
|
|
|
|
func dashboardFind(b biz.DashboardBiz) web.HandlerFunc {
|
|
return func(c web.Context) (err error) {
|
|
var (
|
|
d *dao.Dashboard
|
|
name = c.Query("name")
|
|
key = c.Query("key")
|
|
)
|
|
|
|
ctx, cancel := misc.Context(defaultTimeout)
|
|
defer cancel()
|
|
|
|
d, err = b.FindDashboard(ctx, name, key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return success(c, d)
|
|
}
|
|
}
|
|
|
|
func dashboardSave(b biz.DashboardBiz) web.HandlerFunc {
|
|
return func(c web.Context) error {
|
|
dashboard := &dao.Dashboard{}
|
|
err := c.Bind(dashboard)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch dashboard.Name {
|
|
case "home", "service":
|
|
ctx, cancel := misc.Context(defaultTimeout)
|
|
defer cancel()
|
|
|
|
err = b.UpdateDashboard(ctx, dashboard, c.User())
|
|
default:
|
|
err = errors.New("unknown dashboard: " + dashboard.Name)
|
|
}
|
|
return ajax(c, err)
|
|
}
|
|
}
|
|
|
|
func dashboardFetchData(b biz.DashboardBiz) web.HandlerFunc {
|
|
type Args struct {
|
|
Key string `json:"key" bind:"key"`
|
|
Dashboards string `json:"charts" bind:"charts"`
|
|
Period int32 `json:"period" bind:"period"`
|
|
}
|
|
return func(c web.Context) (err error) {
|
|
var (
|
|
args = &Args{}
|
|
d data.Map
|
|
)
|
|
if err = c.Bind(args); err == nil {
|
|
ctx, cancel := misc.Context(defaultTimeout)
|
|
defer cancel()
|
|
d, err = b.FetchData(ctx, args.Key, strings.Split(args.Dashboards, ","), times.Minutes(args.Period))
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return success(c, d)
|
|
}
|
|
}
|