blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-03 16:26:36 +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()
defer conn.w.Unlock()
if conn.IsCompress {
buff := bytes.NewBuffer(msg)
gz := gzip.NewWriter(buff)
if _, err := gz.Write(buff.Bytes()); err != nil {
var buffer bytes.Buffer
gz := gzip.NewWriter(&buffer)
if _, err := gz.Write(msg); err != nil {
return utils.Wrap(err, "")
}
if err := gz.Close(); err != nil {
return utils.Wrap(err, "")
}
msg = buff.Bytes()
msg = buffer.Bytes()
}
conn.SetWriteDeadline(time.Now().Add(time.Duration(60) * time.Second))
return conn.WriteMessage(a, msg)