mirror of
https://github.com/tiennm99/gomoku.git
synced 2026-05-24 12:26:09 +00:00
tiennm99
1b9eec5f7d
- Module: github.com/ratel-online/server → github.com/tiennm99/gomoku/server
- Copied core/{log,util/async,util/json,util/strings,model,network,protocol,consts}
into server/pkg/* (temporary shims until phase-05 replaces protocol/network)
- Rewrote all ratel-online import paths across server/**/*.go
- Trimmed main.go: single -p flag, WS-only on :1999, no TCP/bot/static
- Trimmed network/wss.go: endpoint /gomoku, no static file serving
- Updated Makefile: removed TCP/bot references, port 1999
97 lines
1.7 KiB
Go
97 lines
1.7 KiB
Go
package protocol
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"github.com/tiennm99/gomoku/server/pkg/consts"
|
|
"github.com/tiennm99/gomoku/server/pkg/json"
|
|
"io"
|
|
"strconv"
|
|
)
|
|
|
|
var (
|
|
lenSize = 4
|
|
)
|
|
|
|
type Packet struct {
|
|
Body []byte `json:"data"`
|
|
}
|
|
|
|
func (p Packet) Int() (int, error) {
|
|
v, err := strconv.ParseInt(p.String(), 10, 64)
|
|
return int(v), err
|
|
}
|
|
|
|
func (p Packet) Int64() (int64, error) {
|
|
v, _ := strconv.ParseInt(p.String(), 10, 64)
|
|
return v, nil
|
|
}
|
|
|
|
func (p Packet) String() string {
|
|
return string(p.Body)
|
|
}
|
|
|
|
func (p Packet) Unmarshal(v interface{}) error {
|
|
return json.Unmarshal(p.Body, v)
|
|
}
|
|
|
|
func StringPacket(msg string) Packet {
|
|
return Packet{
|
|
Body: []byte(msg),
|
|
}
|
|
}
|
|
|
|
func ErrorPacket(err error) Packet {
|
|
return Packet{
|
|
Body: []byte(err.Error()),
|
|
}
|
|
}
|
|
|
|
func ObjectPacket(obj interface{}) Packet {
|
|
return Packet{
|
|
Body: json.Marshal(obj),
|
|
}
|
|
}
|
|
|
|
type ReadWriteCloser interface {
|
|
Read() (*Packet, error)
|
|
Write(msg Packet) error
|
|
Close() error
|
|
IP() string
|
|
}
|
|
|
|
func readUint32(reader io.Reader) (uint32, error) {
|
|
data := make([]byte, 4)
|
|
_, err := io.ReadFull(reader, data)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return binary.BigEndian.Uint32(data), nil
|
|
}
|
|
|
|
func encode(msg Packet) []byte {
|
|
lenBytes := make([]byte, lenSize)
|
|
binary.BigEndian.PutUint32(lenBytes, uint32(len(msg.Body)))
|
|
data := make([]byte, 0)
|
|
data = append(data, lenBytes...)
|
|
return append(data, msg.Body...)
|
|
}
|
|
|
|
func decode(r io.Reader) (*Packet, error) {
|
|
l, err := readUint32(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if l > consts.MaxPacketSize {
|
|
return nil, errors.New("Overflow max packet size " + strconv.Itoa(consts.MaxPacketSize))
|
|
}
|
|
dataBytes := make([]byte, l)
|
|
_, err = io.ReadFull(r, dataBytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Packet{
|
|
Body: dataBytes,
|
|
}, nil
|
|
}
|