🐶 重构部分用户数据库操作代码

server/api/overview.go

@@ -0,0 +1,59 @@
+package api
+
+import (
+	"next-terminal/server/constant"
+	"next-terminal/server/model"
+
+	"github.com/labstack/echo/v4"
+)
+
+type Counter struct {
+	User          int64 `json:"user"`
+	Asset         int64 `json:"asset"`
+	Credential    int64 `json:"credential"`
+	OnlineSession int64 `json:"onlineSession"`
+}
+
+func OverviewCounterEndPoint(c echo.Context) error {
+	account, _ := GetCurrentAccount(c)
+
+	var (
+		countUser          int64
+		countOnlineSession int64
+		credential         int64
+		asset              int64
+	)
+	if constant.TypeUser == account.Type {
+		countUser, _ = userRepository.CountOnlineUser()
+		countOnlineSession, _ = model.CountOnlineSession()
+		credential, _ = model.CountCredentialByUserId(account.ID)
+		asset, _ = model.CountAssetByUserId(account.ID)
+	} else {
+		countUser, _ = userRepository.CountOnlineUser()
+		countOnlineSession, _ = model.CountOnlineSession()
+		credential, _ = model.CountCredential()
+		asset, _ = model.CountAsset()
+	}
+	counter := Counter{
+		User:          countUser,
+		OnlineSession: countOnlineSession,
+		Credential:    credential,
+		Asset:         asset,
+	}
+
+	return Success(c, counter)
+}
+
+func OverviewSessionPoint(c echo.Context) (err error) {
+	d := c.QueryParam("d")
+	var results []model.D
+	if d == "m" {
+		results, err = model.CountSessionByDay(30)
+	} else {
+		results, err = model.CountSessionByDay(7)
+	}
+	if err != nil {
+		return err
+	}
+	return Success(c, results)
+}