fix #69 , /api/stats updates

boypt · Mar 31, 2020 · 01a32ba · 01a32ba
1 parent 4efbd6b
commit 01a32ba
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/server/server_api.go b/server/server_api.go
@@ -79,6 +79,8 @@ func (s *Server) apiGET(w http.ResponseWriter, r *http.Request) error {
 		}
 	case "stat":
 		s.state.Lock()
+		c := s.engine.Config()
+		s.state.Stats.System.loadStats(c.DownloadDirectory)
 		json.NewEncoder(w).Encode(s.state.Stats)
 		s.state.Unlock()
 	case "enginedebug":

diff --git a/server/server_bg.go b/server/server_bg.go
@@ -44,6 +44,7 @@ func (s *Server) backgroundRoutines() {
 			if s.state.NumConnections() > 0 {
 				c := s.engine.Config()
 				s.state.Stats.System.loadStats(c.DownloadDirectory)
+				s.state.Push()
 			}
 			time.Sleep(5 * time.Second)
 		}

diff --git a/server/server_stats.go b/server/server_stats.go
@@ -22,6 +22,10 @@ type stats struct {
 	pusher velox.Pusher
 }
 
+func (s *stats) Push() {
+	s.pusher.Push()
+}
+
 func (s *stats) loadStats(diskDir string) {
 	//count cpu cycles between last count
 	//count disk usage
@@ -44,7 +48,6 @@ func (s *stats) loadStats(diskDir string) {
 	s.GoRoutines = runtime.NumGoroutine()
 	//done
 	s.Set = true
-	s.pusher.Push()
 }
 
 func detectDiskStat(dir string) error {