v2ray · xiaokangwang · Jul 14, 2019 · Jul 8, 2019 · Jul 14, 2019
diff --git a/app/stats/command/command.go b/app/stats/command/command.go
@@ -6,6 +6,8 @@ package command
 
 import (
 	"context"
+	"runtime"
+	"time"
 
 	grpc "google.golang.org/grpc"
 
@@ -18,11 +20,15 @@ import (
 
 // statsServer is an implementation of StatsService.
 type statsServer struct {
-	stats feature_stats.Manager
+	stats     feature_stats.Manager
+	startTime time.Time
 }
 
 func NewStatsServer(manager feature_stats.Manager) StatsServiceServer {
-	return &statsServer{stats: manager}
+	return &statsServer{
+		stats:     manager,
+		startTime: time.Now(),
+	}
 }
 
 func (s *statsServer) GetStats(ctx context.Context, request *GetStatsRequest) (*GetStatsResponse, error) {
@@ -76,6 +82,28 @@ func (s *statsServer) QueryStats(ctx context.Context, request *QueryStatsRequest
 	return response, nil
 }
 
+func (s *statsServer) GetSysStats(ctx context.Context, request *SysStatsRequest) (*SysStatsResponse, error) {
+	var rtm runtime.MemStats
+	runtime.ReadMemStats(&rtm)
+
+	uptime := time.Since(s.startTime)
+
+	response := &SysStatsResponse{
+		Uptime:       uint32(uptime.Seconds()),
+		NumGoroutine: uint32(runtime.NumGoroutine()),
+		Alloc:        rtm.Alloc,
+		TotalAlloc:   rtm.TotalAlloc,
+		Sys:          rtm.Sys,
+		Mallocs:      rtm.Mallocs,
+		Frees:        rtm.Frees,
+		LiveObjects:  rtm.Mallocs - rtm.Frees,
+		NumGC:        rtm.NumGC,
+		PauseTotalNs: rtm.PauseTotalNs,
+	}
+
+	return response, nil
+}
+
 type service struct {
 	statsManager feature_stats.Manager
 }

diff --git a/app/stats/command/command.pb.go b/app/stats/command/command.pb.go