GOMODULES/pgx
1
0
Fork 0
mirror of https://github.com/jackc/pgx.git synced 2025-05-31 11:42:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

writeAll dies on permanent net errors

Browse Source
This commit is contained in:
Jack Christensen 2019-01-26 16:45:06 -06:00
parent 5b9108a20c
commit e3d431d0df
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pgconn/pgconn.go
View File

@ -398,11 +398,15 @@ func (pgConn *PgConn) hardClose() error {
return pgConn.conn.Close() return pgConn.conn.Close()
} }
// writeAll writes the entire buffer successfully or it hard closes the connection. // writeAll writes the entire buffer. The connection is hard closed on a partial write or a non-temporary error.
func (pgConn *PgConn) writeAll(buf []byte) error { func (pgConn *PgConn) writeAll(buf []byte) error {
n, err := pgConn.conn.Write(buf) n, err := pgConn.conn.Write(buf)
if err != nil && n > 0 { if err != nil {
pgConn.hardClose() if n > 0 {
pgConn.hardClose()
} else if ne, ok := err.(net.Error); ok && !ne.Temporary() {
pgConn.hardClose()
}
} }
return err return err
} }