Minor fixes (#45)

* Cleanup code
* Fix documentation typo
This commit is contained in:
Hellojack authored and GitHub committed 2022-08-28 12:40:44 +08:00
1 parent c6ef276811
commit e0de96eb4c
2 files changed
+3 -3

No files matched your search

+2 -2
View File
@@ -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)
}