nezha/cmd/dashboard/controller/common_page.go

package controller

import (
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"

	"github.com/naiba/nezha/model"
	"github.com/naiba/nezha/pkg/mygin"
	"github.com/naiba/nezha/service/dao"
)

type commonPage struct {
	r *gin.Engine
}

func (cp *commonPage) serve() {
	cr := cp.r.Group("")
	cr.Use(mygin.Authorize(mygin.AuthorizeOption{}))
	cr.GET("/", cp.home)
	cr.GET("/service", cp.service)
	cr.GET("/ws", cp.ws)
}

type ServiceItem struct {
	Monitor     model.Monitor
	TotalUp     uint64
	TotalDown   uint64
	CurrentUp   uint64
	CurrentDown uint64
	Delay       *[30]float32
	Up          *[30]int
	Down        *[30]int
}

func (p *commonPage) service(c *gin.Context) {
	var ms []model.Monitor
	dao.DB.Find(&ms)
	year, month, day := time.Now().Date()
	today := time.Date(year, month, day, 0, 0, 0, 0, time.Local)
	var mhs []model.MonitorHistory
	dao.DB.Where("created_at >= ?", today.AddDate(0, 0, -29)).Find(&mhs)

	msm := make(map[uint64]*ServiceItem)
	for i := 0; i < len(ms); i++ {
		msm[ms[i].ID] = &ServiceItem{
			Monitor: ms[i],
			Delay:   &[30]float32{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
			Up:      &[30]int{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
			Down:    &[30]int{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
		}
	}

	// 整合数据
	todayStatus := make(map[uint64][]bool)
	for i := 0; i < len(mhs); i++ {
		dayIndex := 29
		if mhs[i].CreatedAt.Before(today) {
			dayIndex = 28 - (int(today.Sub(mhs[i].CreatedAt).Hours()) / 24)
		} else {
			todayStatus[mhs[i].MonitorID] = append(todayStatus[mhs[i].MonitorID], mhs[i].Successful)
		}
		if mhs[i].Successful {
			msm[mhs[i].MonitorID].TotalUp++
			msm[mhs[i].MonitorID].Delay[dayIndex] = (msm[mhs[i].MonitorID].Delay[dayIndex]*float32(msm[mhs[i].MonitorID].Up[dayIndex]) + mhs[i].Delay) / float32(msm[mhs[i].MonitorID].Up[dayIndex]+1)
			msm[mhs[i].MonitorID].Up[dayIndex]++
		} else {
			msm[mhs[i].MonitorID].TotalDown++
			msm[mhs[i].MonitorID].Down[dayIndex]++
		}
	}

	// 当日最后 20 个采样作为当前状态
	for _, m := range msm {
		for i := len(todayStatus[m.Monitor.ID]) - 1; i >= 0 && i >= (len(todayStatus[m.Monitor.ID])-1-20); i-- {
			if todayStatus[m.Monitor.ID][i] {
				m.CurrentUp++
			} else {
				m.CurrentDown++
			}
		}
	}

	c.HTML(http.StatusOK, "theme-"+dao.Conf.Site.Theme+"/service", mygin.CommonEnvironment(c, gin.H{
		"Title":    "服务状态",
		"Services": msm,
	}))
}

func (cp *commonPage) home(c *gin.Context) {
	dao.ServerLock.RLock()
	defer dao.ServerLock.RUnlock()

	c.HTML(http.StatusOK, "theme-"+dao.Conf.Site.Theme+"/home", mygin.CommonEnvironment(c, gin.H{
		"Servers":    dao.SortedServerList,
		"CustomCode": dao.Conf.Site.CustomCode,
	}))
}

var upgrader = websocket.Upgrader{}

func (cp *commonPage) ws(c *gin.Context) {
	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		mygin.ShowErrorPage(c, mygin.ErrInfo{
			Code:  http.StatusInternalServerError,
			Title: "网络错误",
			Msg:   "Websocket协议切换失败",
			Link:  "/",
			Btn:   "返回首页",
		}, true)
		return
	}
	defer conn.Close()
	for {
		dao.ServerLock.RLock()
		err = conn.WriteJSON(dao.SortedServerList)
		dao.ServerLock.RUnlock()
		if err != nil {
			break
		}
		time.Sleep(time.Second * 2)
	}
}