Update.

Author: xOS

model/server.go

@@ -12,37 +12,49 @@ import (
 	"gorm.io/gorm"
 )
 
+// State 服务器状态
+type State struct {
+	CPU            float64   `json:"cpu"`
+	MemUsed        uint64    `json:"mem_used"`
+	SwapUsed       uint64    `json:"swap_used"`
+	DiskUsed       uint64    `json:"disk_used"`
+	NetInTransfer  uint64    `json:"net_in_transfer"`
+	NetOutTransfer uint64    `json:"net_out_transfer"`
+	NetInSpeed     uint64    `json:"net_in_speed"`
+	NetOutSpeed    uint64    `json:"net_out_speed"`
+	Uptime         uint64    `json:"uptime"`
+	Load1          float64   `json:"load1"`
+	Load5          float64   `json:"load5"`
+	Load15         float64   `json:"load15"`
+	TcpConnCount   int64     `json:"tcp_conn_count"`
+	UdpConnCount   int64     `json:"udp_conn_count"`
+	ProcessCount   int64     `json:"process_count"`
+	Temperatures   []float64 `json:"temperatures"`
+}
+
 type Server struct {
-	ID              uint64    `json:"id" gorm:"primaryKey"`
-	Name            string    `json:"name"`
-	Secret          string    `json:"secret"`
-	Tag             string    `json:"tag"`
-	DisplayIndex    int       `json:"display_index"`
-	Host            *Host     `json:"host" gorm:"-"`
-	State           *State    `json:"state" gorm:"-"`
-	LastActive      time.Time `json:"last_active"`
-	Note            string    `json:"note"`
-	PublicNote      string    `json:"public_note"`
-	EnableDDNS      bool      `json:"enable_ddns"`
-	DDNSProfilesRaw string    `json:"ddns_profiles_raw"`
-	HideForGuest    bool      `json:"hide_for_guest"`
+	ID                     uint64     `json:"id" gorm:"primaryKey"`
+	Name                   string     `json:"name"`
+	Secret                 string     `json:"secret"`
+	Tag                    string     `json:"tag"`           // 分组名
+	DisplayIndex           int        `json:"display_index"` // 展示排序，越大越靠前
+	Host                   *Host      `json:"host" gorm:"-"`
+	State                  *State     `json:"state" gorm:"-"`
+	LastActive             time.Time  `json:"last_active"`
+	Note                   string     `json:"note"`                  // 管理员可见备注
+	PublicNote             string     `json:"public_note,omitempty"` // 公开备注
+	EnableDDNS             bool       `json:"enable_ddns"`           // 启用DDNS
+	DDNSProfilesRaw        string     `json:"ddns_profiles_raw" gorm:"default:'[]'"`
+	HideForGuest           bool       `json:"hide_for_guest"`
+	DDNSProfiles           []uint64   `gorm:"-" json:"-"`         // DDNS配置
+	LastStateBeforeOffline *HostState `gorm:"-" json:"-"`         // 离线前最后一次状态
+	IsOnline               bool       `gorm:"-" json:"is_online"` // 是否在线
+
 	// 累计流量统计
 	CumulativeNetInTransfer  uint64    `json:"cumulative_net_in_transfer" gorm:"default:0"`
 	CumulativeNetOutTransfer uint64    `json:"cumulative_net_out_transfer" gorm:"default:0"`
 	LastTrafficReset         time.Time `json:"last_traffic_reset"`
 
-	Tag          string   // 分组名
-	Note         string   `json:"-"`                    // 管理员可见备注
-	PublicNote   string   `json:"PublicNote,omitempty"` // 公开备注
-	DisplayIndex int      // 展示排序，越大越靠前
-	EnableDDNS   bool     // 启用DDNS
-	DDNSProfiles []uint64 `gorm:"-" json:"-"` // DDNS配置
-
-	DDNSProfilesRaw string `gorm:"default:'[]';column:ddns_profiles_raw" json:"-"`
-
-	LastStateBeforeOffline *HostState `gorm:"-" json:"-"`         // 离线前最后一次状态
-	IsOnline               bool       `gorm:"-" json:"is_online"` // 是否在线
-
 	// 持久化保存的最后状态
 	LastStateJSON string    `gorm:"type:text" json:"-"` // 最后一次状态的JSON格式
 	LastOnline    time.Time // 最后一次在线时间

service/singleton/traffic_manager.go

@@ -23,6 +23,8 @@ type TrafficManager struct {
 	batchBuffer []*model.Transfer
 	// 上次批量写入时间
 	lastBatchWrite time.Time
+	// 数据库连接
+	db *gorm.DB
 }
 
 // TrafficStats 流量统计数据
@@ -72,6 +74,11 @@ func GetTrafficManager() *TrafficManager {
 	return trafficManager
 }
 
+// SetDB 设置数据库连接
+func (tm *TrafficManager) SetDB(db *gorm.DB) {
+	tm.db = db
+}
+
 // UpdateTraffic 更新服务器流量统计
 func (tm *TrafficManager) UpdateTraffic(serverID uint64, inBytes, outBytes uint64) {
 	tm.Lock()
@@ -182,7 +189,7 @@ func (tm *TrafficManager) writeBatchToDatabase() {
 	}
 
 	// 创建数据库事务
-	tx := model.DB.Begin()
+	tx := tm.db.Begin()
 	if tx.Error != nil {
 		log.Printf("创建事务失败: %v", tx.Error)
 		return
@@ -220,7 +227,7 @@ func (tm *TrafficManager) CleanupOldData(days int) error {
 	// 分批删除数据以减少数据库压力
 	batchSize := 1000
 	for {
-		result := DB.Where("created_at < ?", deadline).Limit(batchSize).Delete(&model.Transfer{})
+		result := tm.db.Where("created_at < ?", deadline).Limit(batchSize).Delete(&model.Transfer{})
 		if result.Error != nil {
 			return fmt.Errorf("failed to cleanup old traffic data: %v", result.Error)
 		}