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()
}
// 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 {
n, err := pgConn.conn.Write(buf)
if err != nil && n > 0 {
pgConn.hardClose()
if err != nil {
if n > 0 {
pgConn.hardClose()
} else if ne, ok := err.(net.Error); ok && !ne.Temporary() {
pgConn.hardClose()
}
}
return err
}