func readExtQuery(cn *pool.Conn) (res types.Result, e error) {
for {
c, msgLen, err := readMessageType(cn)
if err != nil {
return nil, err
}
switch c {
case bindCompleteMsg:
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
case dataRowMsg:
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
case commandCompleteMsg: // Response to the EXECUTE message.
b, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
res = types.ParseResult(b)
case readyForQueryMsg: // Response to the SYNC message.
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
return
case errorResponseMsg:
var err error
e, err = readError(cn)
if err != nil {
return nil, err
}
case noticeResponseMsg:
if err := logNotice(cn, msgLen); err != nil {
return nil, err
}
case parameterStatusMsg:
if err := logParameterStatus(cn, msgLen); err != nil {
return nil, err
}
default:
if e != nil {
return nil, e
}
return nil, fmt.Errorf("pg: readExtQuery: unexpected message %#x", c)
}
}
}
func readCopyData(cn *pool.Conn, w io.WriteCloser) (types.Result, error) {
defer w.Close()
for {
c, msgLen, err := readMessageType(cn)
if err != nil {
return nil, err
}
switch c {
case copyDataMsg:
b, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
_, err = w.Write(b)
if err != nil {
return nil, err
}
case copyDoneMsg:
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
case commandCompleteMsg:
b, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
return types.ParseResult(b), nil
case errorResponseMsg:
e, err := readError(cn)
if err != nil {
return nil, err
}
return nil, e
case noticeResponseMsg:
if err := logNotice(cn, msgLen); err != nil {
return nil, err
}
case parameterStatusMsg:
if err := logParameterStatus(cn, msgLen); err != nil {
return nil, err
}
default:
return nil, fmt.Errorf("pg: readCopyData: unexpected message %#x", c)
}
}
}
func readExtQueryData(cn *pool.Conn, mod interface{}, columns []string) (res types.Result, e error) {
coll, ok := mod.(orm.Collection)
if !ok {
coll, e = orm.NewModel(mod)
if e != nil {
coll = Discard
}
}
var model orm.ColumnScanner
for {
c, msgLen, err := readMessageType(cn)
if err != nil {
return nil, err
}
switch c {
case bindCompleteMsg:
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
case dataRowMsg:
model = coll.NewModel()
if err := readDataRow(cn, model, columns); err != nil {
e = err
}
if err := coll.AddModel(model); err != nil {
return nil, err
}
case commandCompleteMsg: // Response to the EXECUTE message.
b, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
res = types.ParseResult(b)
case readyForQueryMsg: // Response to the SYNC message.
_, err := cn.ReadN(msgLen)
if err != nil {
return nil, err
}
return
case errorResponseMsg:
var err error
e, err = readError(cn)
if err != nil {
return nil, err
}
case noticeResponseMsg:
if err := logNotice(cn, msgLen); err != nil {
return nil, err
}
case parameterStatusMsg:
if err := logParameterStatus(cn, msgLen); err != nil {
return nil, err
}
default:
if e != nil {
return nil, e
}
return nil, fmt.Errorf("pg: readExtQueryData: unexpected message %#x", c)
}
}
}
