mirror of
https://github.com/skyle1995/NetworkAuth.git
synced 2026-05-25 02:24:05 +08:00
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package admin
|
||
|
||
import (
|
||
"NetworkAuth/constants"
|
||
"NetworkAuth/controllers"
|
||
"NetworkAuth/middleware"
|
||
"NetworkAuth/models"
|
||
"NetworkAuth/utils/timeutil"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// ============================================================================
|
||
// 全局变量
|
||
// ============================================================================
|
||
|
||
// 创建基础控制器实例
|
||
var handlersBaseController = controllers.NewBaseController()
|
||
|
||
// ============================================================================
|
||
// 辅助函数
|
||
// ============================================================================
|
||
|
||
// formatDBType 格式化数据库类型显示
|
||
// 将配置文件中的小写类型转换为友好的显示格式
|
||
func formatDBType(dbType string) string {
|
||
switch dbType {
|
||
case "mysql":
|
||
return "MySQL"
|
||
case "sqlite":
|
||
return "SQLite"
|
||
case "postgresql", "postgres":
|
||
return "PostgreSQL"
|
||
case "sqlserver":
|
||
return "SQL Server"
|
||
default:
|
||
return "SQLite" // 默认显示
|
||
}
|
||
}
|
||
|
||
// ============================================================================
|
||
// API处理器
|
||
// ============================================================================
|
||
// 返回系统运行状态的JSON数据，用于前端定时刷新
|
||
func SystemInfoHandler(c *gin.Context) {
|
||
version := constants.AppVersion
|
||
mode := middleware.IsDevModeFromContext(c)
|
||
dbType := viper.GetString("database.type")
|
||
if dbType == "" {
|
||
dbType = "sqlite"
|
||
}
|
||
uptime := timeutil.GetServerUptimeString()
|
||
uptimeSeconds := int64(timeutil.GetServerUptime().Seconds())
|
||
|
||
data := gin.H{
|
||
"version": version,
|
||
"mode": mode,
|
||
"db_type": formatDBType(dbType),
|
||
"uptime": uptime,
|
||
"uptime_seconds": uptimeSeconds,
|
||
}
|
||
|
||
handlersBaseController.HandleSuccess(c, "ok", data)
|
||
}
|
||
|
||
// DashboardStatsHandler 仪表盘统计数据API接口
|
||
// - 返回应用统计数据的JSON数据，包括全部/启用/变量数量
|
||
func DashboardStatsHandler(c *gin.Context) {
|
||
// 获取数据库连接
|
||
db, ok := handlersBaseController.GetDB(c)
|
||
if !ok {
|
||
return
|
||
}
|
||
|
||
// 统计应用数据
|
||
var totalApps int64
|
||
var totalFunctions int64
|
||
var totalVariables int64
|
||
|
||
// 统计全部应用数量
|
||
if err := db.Model(&models.App{}).Count(&totalApps).Error; err != nil {
|
||
handlersBaseController.HandleInternalError(c, "统计应用数量失败", err)
|
||
return
|
||
}
|
||
|
||
// 统计函数数量
|
||
if err := db.Model(&models.Function{}).Count(&totalFunctions).Error; err != nil {
|
||
handlersBaseController.HandleInternalError(c, "统计函数数量失败", err)
|
||
return
|
||
}
|
||
|
||
// 统计变量数量
|
||
if err := db.Model(&models.Variable{}).Count(&totalVariables).Error; err != nil {
|
||
handlersBaseController.HandleInternalError(c, "统计变量数量失败", err)
|
||
return
|
||
}
|
||
|
||
data := gin.H{
|
||
"total_apps": totalApps,
|
||
"total_functions": totalFunctions,
|
||
"total_variables": totalVariables,
|
||
}
|
||
|
||
handlersBaseController.HandleSuccess(c, "ok", data)
|
||
}
|