Jack Christensen
4e2900b774
pgproto3 will wrap the message encoding and decoding for the PostgreSQL frontend/backend protocol version 3.
42 lines
715 B
Go
42 lines
715 B
Go
package pgproto3
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
)
|
|
|
|
type CopyData struct {
|
|
Data []byte
|
|
}
|
|
|
|
func (*CopyData) Backend() {}
|
|
func (*CopyData) Frontend() {}
|
|
|
|
func (dst *CopyData) UnmarshalBinary(src []byte) error {
|
|
dst.Data = make([]byte, len(src))
|
|
copy(dst.Data, src)
|
|
return nil
|
|
}
|
|
|
|
func (src *CopyData) MarshalBinary() ([]byte, error) {
|
|
var bigEndian BigEndianBuf
|
|
buf := &bytes.Buffer{}
|
|
|
|
buf.WriteByte('d')
|
|
buf.Write(bigEndian.Uint32(uint32(4 + len(src.Data))))
|
|
buf.Write(src.Data)
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
func (src *CopyData) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(struct {
|
|
Type string
|
|
Data string
|
|
}{
|
|
Type: "CopyData",
|
|
Data: hex.EncodeToString(src.Data),
|
|
})
|
|
}
|