优化代码

cmd/edge-node/main.go

@@ -95,17 +95,12 @@ func main() {
 		if err != nil {
 			fmt.Println("[ERROR]" + err.Error())
 		} else {
-			instances, ok := reply.Params["result"]
-			if ok {
-				instancesJSON, err := json.MarshalIndent(instances, "", "  ")
+			instancesJSON, err := json.MarshalIndent(reply.Params, "", "  ")
 			if err != nil {
 				fmt.Println("[ERROR]" + err.Error())
 			} else {
 				fmt.Println(string(instancesJSON))
 			}
-			} else {
-				fmt.Println("no instances yet.")
-			}
 		}
 	})
 	app.On("conns", func() {

internal/nodes/http_writer_rate.go

@@ -88,3 +88,15 @@ func (this *HTTPRateWriter) Hijack() (conn net.Conn, buf *bufio.ReadWriter, err
 	}
 	return
 }
+
+// Flush Flush
+func (this *HTTPRateWriter) Flush() {
+	if this.parentWriter == nil {
+		return
+	}
+	flusher, ok := this.parentWriter.(http.Flusher)
+	if ok {
+		flusher.Flush()
+		return
+	}
+}

internal/nodes/node.go

@@ -170,9 +170,11 @@ func (this *Node) Start() {
 	})
 
 	// 统计
-	go stats.SharedTrafficStatManager.Start(func() *nodeconfigs.NodeConfig {
+	goman.New(func() {
+		stats.SharedTrafficStatManager.Start(func() *nodeconfigs.NodeConfig {
 			return sharedNodeConfig
 		})
+	})
 	goman.New(func() {
 		stats.SharedHTTPRequestStatManager.Start()
 	})
@@ -618,6 +620,7 @@ func (this *Node) listenSock() error {
 
 				_ = cmd.Reply(&gosock.Command{
 					Params: map[string]interface{}{
+						"total":  runtime.NumGoroutine(),
 						"result": result,
 					},
 				})