feat: add system instance info panel (#5716)

* feat: add system instance reporting

* feat: show system instance resources

* fix: update translations for heartbeat messages in Russian and Vietnamese

model/main.go

@@ -294,6 +294,7 @@ func migrateDB() error {
 		&CustomOAuthProvider{},
 		&UserOAuthBinding{},
 		&PerfMetric{},
+		&SystemInstance{},
 		&SystemTask{},
 		&SystemTaskLock{},
 	)
@@ -345,6 +346,7 @@ func migrateDBFast() error {
 		{&CustomOAuthProvider{}, "CustomOAuthProvider"},
 		{&UserOAuthBinding{}, "UserOAuthBinding"},
 		{&PerfMetric{}, "PerfMetric"},
+		{&SystemInstance{}, "SystemInstance"},
 		{&SystemTask{}, "SystemTask"},
 		{&SystemTaskLock{}, "SystemTaskLock"},
 	}

model/system_instance.go

@@ -0,0 +1,113 @@
+package model
+
+import (
+	"github.com/QuantumNous/new-api/common"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+)
+
+const (
+	SystemInstanceStatusOnline = "online"
+	SystemInstanceStatusStale  = "stale"
+
+	SystemInstanceStaleAfterSeconds int64 = 90
+)
+
+type SystemInstance struct {
+	NodeName   string `json:"node_name" gorm:"type:varchar(128);primaryKey"`
+	Info       string `json:"info" gorm:"type:text"`
+	StartedAt  int64  `json:"started_at" gorm:"bigint;index"`
+	LastSeenAt int64  `json:"last_seen_at" gorm:"bigint;index"`
+	CreatedAt  int64  `json:"created_at" gorm:"bigint;index"`
+	UpdatedAt  int64  `json:"updated_at" gorm:"bigint;index"`
+}
+
+type SystemInstanceResponse struct {
+	NodeName          string `json:"node_name"`
+	Status            string `json:"status"`
+	StaleAfterSeconds int64  `json:"stale_after_seconds"`
+	StartedAt         int64  `json:"started_at"`
+	LastSeenAt        int64  `json:"last_seen_at"`
+	Info              any    `json:"info"`
+}
+
+func (instance *SystemInstance) BeforeCreate(_ *gorm.DB) error {
+	now := common.GetTimestamp()
+	if instance.CreatedAt == 0 {
+		instance.CreatedAt = now
+	}
+	if instance.UpdatedAt == 0 {
+		instance.UpdatedAt = now
+	}
+	return nil
+}
+
+func UpsertSystemInstance(nodeName string, info any, startedAt int64, lastSeenAt int64) error {
+	infoText, err := marshalSystemInstanceInfo(info)
+	if err != nil {
+		return err
+	}
+	if lastSeenAt == 0 {
+		lastSeenAt = common.GetTimestamp()
+	}
+	instance := &SystemInstance{
+		NodeName:   nodeName,
+		Info:       infoText,
+		StartedAt:  startedAt,
+		LastSeenAt: lastSeenAt,
+		UpdatedAt:  lastSeenAt,
+	}
+	return DB.Clauses(clause.OnConflict{
+		Columns: []clause.Column{{Name: "node_name"}},
+		DoUpdates: clause.AssignmentColumns([]string{
+			"info",
+			"started_at",
+			"last_seen_at",
+			"updated_at",
+		}),
+	}).Create(instance).Error
+}
+
+func ListSystemInstances() ([]*SystemInstance, error) {
+	var instances []*SystemInstance
+	err := DB.Order("last_seen_at desc").Find(&instances).Error
+	return instances, err
+}
+
+func (instance *SystemInstance) ToResponse(now int64) SystemInstanceResponse {
+	status := SystemInstanceStatusOnline
+	if now-instance.LastSeenAt > SystemInstanceStaleAfterSeconds {
+		status = SystemInstanceStatusStale
+	}
+	return SystemInstanceResponse{
+		NodeName:          instance.NodeName,
+		Status:            status,
+		StaleAfterSeconds: SystemInstanceStaleAfterSeconds,
+		StartedAt:         instance.StartedAt,
+		LastSeenAt:        instance.LastSeenAt,
+		Info:              decodeSystemInstanceInfo(instance.Info),
+	}
+}
+
+func marshalSystemInstanceInfo(v any) (string, error) {
+	if v == nil {
+		return "", nil
+	}
+	data, err := common.Marshal(v)
+	if err != nil {
+		return "", err
+	}
+	return string(data), nil
+}
+
+func decodeSystemInstanceInfo(data string) any {
+	if data == "" {
+		return nil
+	}
+	var value any
+	if err := common.UnmarshalJsonStr(data, &value); err != nil {
+		return data
+	}
+	return value
+}

model/task_cas_test.go

@@ -48,6 +48,7 @@ func TestMain(m *testing.M) {
 		&UserSubscription{},
 		&UserOAuthBinding{},
 		&PerfMetric{},
+		&SystemInstance{},
 		&SystemTask{},
 		&SystemTaskLock{},
 	); err != nil {
@@ -73,6 +74,7 @@ func truncateTables(t *testing.T) {
 		DB.Exec("DELETE FROM user_subscriptions")
 		DB.Exec("DELETE FROM user_oauth_bindings")
 		DB.Exec("DELETE FROM perf_metrics")
+		DB.Exec("DELETE FROM system_instances")
 		DB.Exec("DELETE FROM system_task_locks")
 		DB.Exec("DELETE FROM system_tasks")
 	})