fix: pass flow field in CreateUser and prevent NaN JSON crash (Ehco1996#427)

Author: Ehco1996

pkg/metric_reader/node.go

@@ -130,9 +130,15 @@ func (b *readerImpl) processLoadMetrics(metricMap map[string]*dto.MetricFamily,
 
 func (b *readerImpl) calculateFinalMetrics(nm *NodeMetrics, cpu *cpuStats) {
 	nm.CpuCoreCount = cpu.cores
-	nm.CpuUsagePercent = 100 * (cpu.totalTime - cpu.idleTime) / cpu.totalTime
-	nm.MemoryUsagePercent = 100 * float64(nm.MemoryUsageBytes) / float64(nm.MemoryTotalBytes)
-	nm.DiskUsagePercent = 100 * float64(nm.DiskUsageBytes) / float64(nm.DiskTotalBytes)
+	if cpu.totalTime > 0 {
+		nm.CpuUsagePercent = 100 * (cpu.totalTime - cpu.idleTime) / cpu.totalTime
+	}
+	if nm.MemoryTotalBytes > 0 {
+		nm.MemoryUsagePercent = 100 * float64(nm.MemoryUsageBytes) / float64(nm.MemoryTotalBytes)
+	}
+	if nm.DiskTotalBytes > 0 {
+		nm.DiskUsagePercent = 100 * float64(nm.DiskUsageBytes) / float64(nm.DiskTotalBytes)
+	}
 
 	nm.CpuUsagePercent = math.Round(nm.CpuUsagePercent*100) / 100
 	nm.MemoryUsagePercent = math.Round(nm.MemoryUsagePercent*100) / 100

pkg/xray/bandwidth_recorder.go

@@ -75,8 +75,10 @@ func (b *bandwidthRecorder) RecordOnce(ctx context.Context) (uploadIncr float64,
 		elapsed := now.Sub(b.lastRecordTime).Seconds()
 		uploadIncr = (send - b.currentSendBytes)
 		downloadIncr = (recv - b.currentRecvBytes)
-		b.uploadBandwidthBytes = uploadIncr / elapsed
-		b.downloadBandwidthBytes = downloadIncr / elapsed
+		if elapsed > 0 {
+			b.uploadBandwidthBytes = uploadIncr / elapsed
+			b.downloadBandwidthBytes = downloadIncr / elapsed
+		}
 	}
 	b.lastRecordTime = now
 	b.currentRecvBytes = recv

pkg/xray/user.go

@@ -146,7 +146,7 @@ func NewUserPool(grpcEndPoint, remoteConfigURL, metricURL string, proxyTags []st
 	return up
 }
 
-func (up *UserPool) CreateUser(userId, level int, password, method, protocol string, enable bool) *User {
+func (up *UserPool) CreateUser(userId, level int, password, method, protocol, flow string, enable bool) *User {
 	up.Lock()
 	defer up.Unlock()
 	u := &User{
@@ -157,6 +157,7 @@ func (up *UserPool) CreateUser(userId, level int, password, method, protocol str
 		Enable:   enable,
 		Method:   method,
 		Protocol: protocol,
+		Flow:     flow,
 	}
 	up.users[u.ID] = u
 	return u
@@ -269,7 +270,7 @@ func (up *UserPool) syncUserConfigsFromServer(ctx context.Context, proxyTag stri
 		oldUser, found := up.GetUser(newUser.ID)
 		if !found {
 			newUser := up.CreateUser(
-				newUser.ID, newUser.Level, newUser.Password, newUser.Method, newUser.Protocol, newUser.Enable)
+				newUser.ID, newUser.Level, newUser.Password, newUser.Method, newUser.Protocol, newUser.Flow, newUser.Enable)
 			if newUser.Enable {
 				if err := AddInboundUser(ctx, up.proxyClient, proxyTag, newUser); err != nil {
 					return err