Make MsgReader private
This commit is contained in:
+33
-33
@@ -8,26 +8,26 @@ import (
|
||||
"io/ioutil"
|
||||
)
|
||||
|
||||
// MsgReader is a helper that reads values from a PostgreSQL message.
|
||||
type MsgReader struct {
|
||||
// msgReader is a helper that reads values from a PostgreSQL message.
|
||||
type msgReader struct {
|
||||
reader *bufio.Reader
|
||||
buf [128]byte
|
||||
msgBytesRemaining int32
|
||||
err error
|
||||
}
|
||||
|
||||
// Err returns any error that the MsgReader has experienced
|
||||
func (r *MsgReader) Err() error {
|
||||
// Err returns any error that the msgReader has experienced
|
||||
func (r *msgReader) Err() error {
|
||||
return r.err
|
||||
}
|
||||
|
||||
// Fatal tells r that a Fatal error has occurred
|
||||
func (r *MsgReader) Fatal(err error) {
|
||||
// fatal tells r that a Fatal error has occurred
|
||||
func (r *msgReader) fatal(err error) {
|
||||
r.err = err
|
||||
}
|
||||
|
||||
// rxMsg reads the type and size of the next message.
|
||||
func (r *MsgReader) rxMsg() (t byte, err error) {
|
||||
func (r *msgReader) rxMsg() (t byte, err error) {
|
||||
if r.err != nil {
|
||||
return 0, err
|
||||
}
|
||||
@@ -43,123 +43,123 @@ func (r *MsgReader) rxMsg() (t byte, err error) {
|
||||
return t, err
|
||||
}
|
||||
|
||||
func (r *MsgReader) ReadByte() byte {
|
||||
func (r *msgReader) readByte() byte {
|
||||
if r.err != nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= 1
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return 0
|
||||
}
|
||||
|
||||
b, err := r.reader.ReadByte()
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return 0
|
||||
}
|
||||
|
||||
return b
|
||||
}
|
||||
|
||||
func (r *MsgReader) ReadInt16() int16 {
|
||||
func (r *msgReader) readInt16() int16 {
|
||||
if r.err != nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= 2
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return 0
|
||||
}
|
||||
|
||||
b := r.buf[0:2]
|
||||
_, err := io.ReadFull(r.reader, b)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return 0
|
||||
}
|
||||
|
||||
return int16(binary.BigEndian.Uint16(b))
|
||||
}
|
||||
|
||||
func (r *MsgReader) ReadInt32() int32 {
|
||||
func (r *msgReader) readInt32() int32 {
|
||||
if r.err != nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= 4
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return 0
|
||||
}
|
||||
|
||||
b := r.buf[0:4]
|
||||
_, err := io.ReadFull(r.reader, b)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return 0
|
||||
}
|
||||
|
||||
return int32(binary.BigEndian.Uint32(b))
|
||||
}
|
||||
|
||||
func (r *MsgReader) ReadInt64() int64 {
|
||||
func (r *msgReader) readInt64() int64 {
|
||||
if r.err != nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= 8
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return 0
|
||||
}
|
||||
|
||||
b := r.buf[0:8]
|
||||
_, err := io.ReadFull(r.reader, b)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return 0
|
||||
}
|
||||
|
||||
return int64(binary.BigEndian.Uint64(b))
|
||||
}
|
||||
|
||||
func (r *MsgReader) ReadOid() Oid {
|
||||
return Oid(r.ReadInt32())
|
||||
func (r *msgReader) readOid() Oid {
|
||||
return Oid(r.readInt32())
|
||||
}
|
||||
|
||||
// ReadCString reads a null terminated string
|
||||
func (r *MsgReader) ReadCString() string {
|
||||
// readCString reads a null terminated string
|
||||
func (r *msgReader) readCString() string {
|
||||
if r.err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
b, err := r.reader.ReadBytes(0)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return ""
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= int32(len(b))
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return ""
|
||||
}
|
||||
|
||||
return string(b[0 : len(b)-1])
|
||||
}
|
||||
|
||||
// ReadString reads count bytes and returns as string
|
||||
func (r *MsgReader) ReadString(count int32) string {
|
||||
// readString reads count bytes and returns as string
|
||||
func (r *msgReader) readString(count int32) string {
|
||||
if r.err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= count
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return ""
|
||||
}
|
||||
|
||||
@@ -172,22 +172,22 @@ func (r *MsgReader) ReadString(count int32) string {
|
||||
|
||||
_, err := io.ReadFull(r.reader, b)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return ""
|
||||
}
|
||||
|
||||
return string(b)
|
||||
}
|
||||
|
||||
// ReadBytes reads count bytes and returns as []byte
|
||||
func (r *MsgReader) ReadBytes(count int32) []byte {
|
||||
// readBytes reads count bytes and returns as []byte
|
||||
func (r *msgReader) readBytes(count int32) []byte {
|
||||
if r.err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
r.msgBytesRemaining -= count
|
||||
if r.msgBytesRemaining < 0 {
|
||||
r.Fatal(errors.New("read past end of message"))
|
||||
r.fatal(errors.New("read past end of message"))
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -195,7 +195,7 @@ func (r *MsgReader) ReadBytes(count int32) []byte {
|
||||
|
||||
_, err := io.ReadFull(r.reader, b)
|
||||
if err != nil {
|
||||
r.Fatal(err)
|
||||
r.fatal(err)
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user