sashakoshka/hopp
sashakoshka/hopp
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Break out encodeMessageB, decodeMessageB

Browse Source
This commit is contained in:
Sasha Koshka 2025-01-19 15:54:13 -05:00
parent 256f0202a2
commit 31930f871b
1 changed files with 27 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
metadaptb.go
View File

@ -48,28 +48,11 @@ func (trans transB) ID() int64 {
}
func (trans transB) Send(method uint16, data []byte) error {
buffer := make([]byte, 4 + len(data))
tape.EncodeI16(buffer[:2], method)
length, ok := tape.U16CastSafe(len(data))
if !ok { return ErrPayloadTooLarge }
tape.EncodeI16(data[2:4], length)
copy(buffer[4:], data)
_, err := trans.underlying.Write(buffer)
return err
return encodeMessageB(trans.underlying, method, data)
}
func (trans transB) Receive() (uint16, []byte, error) {
headerBuffer := [4]byte { }
_, err := io.ReadFull(trans.underlying, headerBuffer[:])
if err != nil { return 0, nil, err }
method, err := tape.DecodeI16[uint16](headerBuffer[:2])
if err != nil { return 0, nil, err }
length, err := tape.DecodeI16[uint16](headerBuffer[2:4])
if err != nil { return 0, nil, err }
payloadBuffer := make([]byte, int(length))
_, err = io.ReadFull(trans.underlying, payloadBuffer)
if err != nil { return 0, nil, err }
return method, payloadBuffer, nil
return decodeMessageB(trans.underlying)
}
// MultiConn represens a multiplexed stream-oriented transport for use in [B].
@ -91,3 +74,28 @@ type Stream interface {
// ID returns the stream ID
ID() int64
}
func encodeMessageB(writer io.Writer, method uint16, data []byte) error {
buffer := make([]byte, 4 + len(data))
tape.EncodeI16(buffer[:2], method)
length, ok := tape.U16CastSafe(len(data))
if !ok { return ErrPayloadTooLarge }
tape.EncodeI16(data[2:4], length)
copy(buffer[4:], data)
_, err := writer.Write(buffer)
return err
}
func decodeMessageB(reader io.Reader) (uint16, []byte, error) {
headerBuffer := [4]byte { }
_, err := io.ReadFull(reader, headerBuffer[:])
if err != nil { return 0, nil, err }
method, err := tape.DecodeI16[uint16](headerBuffer[:2])
if err != nil { return 0, nil, err }
length, err := tape.DecodeI16[uint16](headerBuffer[2:4])
if err != nil { return 0, nil, err }
payloadBuffer := make([]byte, int(length))
_, err = io.ReadFull(reader, payloadBuffer)
if err != nil { return 0, nil, err }
return method, payloadBuffer, nil
}