blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-04 09:09:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

gzip msg

Browse Source
This commit is contained in:
wangchuxiao 2022-11-16 19:26:49 +08:00
parent 05d83ebe8b
commit 4a3f168720
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/msg_gateway/gate/ws_server.go
View File

@ -135,15 +135,15 @@ func (ws *WServer) writeMsg(conn *UserConn, a int, msg []byte) error {
conn.w.Lock() conn.w.Lock()
defer conn.w.Unlock() defer conn.w.Unlock()
if conn.IsCompress { if conn.IsCompress {
buff := bytes.NewBuffer(msg) var buffer bytes.Buffer
gz := gzip.NewWriter(buff) gz := gzip.NewWriter(&buffer)
if _, err := gz.Write(buff.Bytes()); err != nil { if _, err := gz.Write(msg); err != nil {
return utils.Wrap(err, "") return utils.Wrap(err, "")
} }
if err := gz.Close(); err != nil { if err := gz.Close(); err != nil {
return utils.Wrap(err, "") return utils.Wrap(err, "")
} }
msg = buff.Bytes() msg = buffer.Bytes()
} }
conn.SetWriteDeadline(time.Now().Add(time.Duration(60) * time.Second)) conn.SetWriteDeadline(time.Now().Add(time.Duration(60) * time.Second))
return conn.WriteMessage(a, msg) return conn.WriteMessage(a, msg)