Fix ConnGauge nil crash

apernet · Apr 28, 2021 · 214f1cf · 214f1cf
1 parent ac1233a
commit 214f1cf
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/core/server_client.go b/pkg/core/server_client.go
@@ -81,11 +81,15 @@ func (c *serverClient) Run() {
 		if err != nil {
 			break
 		}
-		c.ConnGauge.Inc()
+		if c.ConnGauge != nil {
+			c.ConnGauge.Inc()
+		}
 		go func() {
 			c.handleStream(stream)
 			_ = stream.Close()
-			c.ConnGauge.Dec()
+			if c.ConnGauge != nil {
+				c.ConnGauge.Dec()
+			}
 		}()
 	}
 }