oss/FrankerFaceZ
oss/FrankerFaceZ
1
0
Fork 0
mirror of https://github.com/FrankerFaceZ/FrankerFaceZ.git synced 2025-06-27 21:05:53 +00:00
Code Wiki Activity

Optimize MarshalClientMessage to avoid copies

Browse source
This commit is contained in:
Kane York 2017-10-23 15:39:00 -07:00
parent e0bad1dc10
commit edb728f7bb
1 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

19
socketserver/server/handlecore.go
View file

@ -1,6 +1,7 @@
package server // import "github.com/FrankerFaceZ/FrankerFaceZ/socketserver/server"
import (
"bytes"
"encoding/json"
"errors"
"fmt"
@ -619,7 +620,6 @@ func MarshalClientMessage(clientMessage interface{}) (int, []byte, error) {
}
msg = *pMsg
}
var dataStr string
if msg.Command == "" && msg.MessageID == 0 {
panic("MarshalClientMessage: attempt to send an empty ClientMessage")
@ -632,20 +632,25 @@ func MarshalClientMessage(clientMessage interface{}) (int, []byte, error) {
msg.MessageID = -1
}
// optimized from fmt.Sprintf("%d %s %s", msg.MessageID, msg.Command, ...)
var buf bytes.Buffer
fmt.Fprint(&buf, msg.MessageID)
buf.WriteByte(' ')
buf.WriteString(string(msg.Command))
if msg.origArguments != "" {
dataStr = fmt.Sprintf("%d %s %s", msg.MessageID, msg.Command, msg.origArguments)
buf.WriteByte(' ')
buf.WriteString(msg.origArguments)
} else if msg.Arguments != nil {
argBytes, err := json.Marshal(msg.Arguments)
if err != nil {
return 0, nil, err
}
dataStr = fmt.Sprintf("%d %s %s", msg.MessageID, msg.Command, string(argBytes))
} else {
dataStr = fmt.Sprintf("%d %s", msg.MessageID, msg.Command)
buf.WriteByte(' ')
buf.Write(argBytes)
}
return websocket.TextMessage, []byte(dataStr), nil
return websocket.TextMessage, buf.Bytes(), nil
}
// ArgumentsAsString parses the arguments of the ClientMessage as a single string.