You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.6 KiB
66 lines
1.6 KiB
3 years ago
|
package pgproto3
|
||
|
|
||
|
import (
|
||
|
"encoding/hex"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
)
|
||
|
|
||
|
// Message is the interface implemented by an object that can decode and encode
|
||
|
// a particular PostgreSQL message.
|
||
|
type Message interface {
|
||
|
// Decode is allowed and expected to retain a reference to data after
|
||
|
// returning (unlike encoding.BinaryUnmarshaler).
|
||
|
Decode(data []byte) error
|
||
|
|
||
|
// Encode appends itself to dst and returns the new buffer.
|
||
|
Encode(dst []byte) []byte
|
||
|
}
|
||
|
|
||
|
type FrontendMessage interface {
|
||
|
Message
|
||
|
Frontend() // no-op method to distinguish frontend from backend methods
|
||
|
}
|
||
|
|
||
|
type BackendMessage interface {
|
||
|
Message
|
||
|
Backend() // no-op method to distinguish frontend from backend methods
|
||
|
}
|
||
|
|
||
|
type AuthenticationResponseMessage interface {
|
||
|
BackendMessage
|
||
|
AuthenticationResponse() // no-op method to distinguish authentication responses
|
||
|
}
|
||
|
|
||
|
type invalidMessageLenErr struct {
|
||
|
messageType string
|
||
|
expectedLen int
|
||
|
actualLen int
|
||
|
}
|
||
|
|
||
|
func (e *invalidMessageLenErr) Error() string {
|
||
|
return fmt.Sprintf("%s body must have length of %d, but it is %d", e.messageType, e.expectedLen, e.actualLen)
|
||
|
}
|
||
|
|
||
|
type invalidMessageFormatErr struct {
|
||
|
messageType string
|
||
|
}
|
||
|
|
||
|
func (e *invalidMessageFormatErr) Error() string {
|
||
|
return fmt.Sprintf("%s body is invalid", e.messageType)
|
||
|
}
|
||
|
|
||
|
// getValueFromJSON gets the value from a protocol message representation in JSON.
|
||
|
func getValueFromJSON(v map[string]string) ([]byte, error) {
|
||
|
if v == nil {
|
||
|
return nil, nil
|
||
|
}
|
||
|
if text, ok := v["text"]; ok {
|
||
|
return []byte(text), nil
|
||
|
}
|
||
|
if binary, ok := v["binary"]; ok {
|
||
|
return hex.DecodeString(binary)
|
||
|
}
|
||
|
return nil, errors.New("unknown protocol representation")
|
||
|
}
|