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.
65 lines
1.6 KiB
65 lines
1.6 KiB
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") |
|
}
|
|
|