From bb840779512c8b505bfffa4aaf13317832502c77 Mon Sep 17 00:00:00 2001
From: Hellojack <106379370+h1jk@users.noreply.github.com>
Date: Sun, 28 Aug 2022 01:42:35 +0800
Subject: [PATCH] Cleanup code

---
 transport/v2raygrpclite/conn.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/transport/v2raygrpclite/conn.go b/transport/v2raygrpclite/conn.go
index b0607176..22803a98 100644
--- a/transport/v2raygrpclite/conn.go
+++ b/transport/v2raygrpclite/conn.go
@@ -102,8 +102,8 @@ func (c *GunConn) Write(b []byte) (n int, err error) {
 	binary.BigEndian.PutUint32(grpcHeader[1:5], grpcPayloadLen)
 	_, err = bufio.Copy(c.writer, io.MultiReader(bytes.NewReader(grpcHeader), bytes.NewReader(protobufHeader[:varuintLen+1]), bytes.NewReader(b)))
 	buf.Put(grpcHeader)
-	if f, ok := c.writer.(http.Flusher); ok {
-		f.Flush()
+	if c.flusher != nil {
+		c.flusher.Flush()
 	}
 	return len(b), wrapError(err)
 }