mirror of https://github.com/jackc/pgx.git
Factor out sendSimpleQuery
parent
aff1eee892
commit
0cbf3b6f9e
24
conn.go
24
conn.go
|
@ -63,16 +63,8 @@ func (c *conn) Close() (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *conn) Query(sql string) (rows []map[string]string, err error) {
|
func (c *conn) Query(sql string) (rows []map[string]string, err error) {
|
||||||
bufSize := 5 + len(sql) + 1 // message identifier (1), message size (4), null string terminator (1)
|
if err = c.sendSimpleQuery(sql); err != nil {
|
||||||
buf := c.getBuf(bufSize)
|
return
|
||||||
buf[0] = 'Q'
|
|
||||||
binary.BigEndian.PutUint32(buf[1:5], uint32(bufSize-1))
|
|
||||||
copy(buf[5:], sql)
|
|
||||||
buf[bufSize-1] = 0
|
|
||||||
|
|
||||||
_, err = c.conn.Write(buf)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var response interface{}
|
var response interface{}
|
||||||
|
@ -91,6 +83,18 @@ func (c *conn) Query(sql string) (rows []map[string]string, err error) {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *conn) sendSimpleQuery(sql string) (err error) {
|
||||||
|
bufSize := 5 + len(sql) + 1 // message identifier (1), message size (4), null string terminator (1)
|
||||||
|
buf := c.getBuf(bufSize)
|
||||||
|
buf[0] = 'Q'
|
||||||
|
binary.BigEndian.PutUint32(buf[1:5], uint32(bufSize-1))
|
||||||
|
copy(buf[5:], sql)
|
||||||
|
buf[bufSize-1] = 0
|
||||||
|
|
||||||
|
_, err = c.conn.Write(buf)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
func (c *conn) rxMsg() (msg interface{}, err error) {
|
func (c *conn) rxMsg() (msg interface{}, err error) {
|
||||||
var t byte
|
var t byte
|
||||||
var bodySize int32
|
var bodySize int32
|
||||||
|
|
Loading…
Reference in New Issue