Skip to content

Commit

Permalink
conncheck: move var declarations into closure (#997)
Browse files Browse the repository at this point in the history
  • Loading branch information
julienschmidt committed Sep 7, 2019
1 parent d9aa6d3 commit b2c03bc
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 22 deletions.
43 changes: 22 additions & 21 deletions conncheck.go
Expand Up @@ -19,35 +19,36 @@ import (

var errUnexpectedRead = errors.New("unexpected read from socket")

func connCheck(c net.Conn) error {
var (
n int
err error
buff [1]byte
)

sconn, ok := c.(syscall.Conn)
func connCheck(conn net.Conn) error {
var sysErr error

sysConn, ok := conn.(syscall.Conn)
if !ok {
return nil
}
rc, err := sconn.SyscallConn()
rawConn, err := sysConn.SyscallConn()
if err != nil {
return err
}
rerr := rc.Read(func(fd uintptr) bool {
n, err = syscall.Read(int(fd), buff[:])

err = rawConn.Read(func(fd uintptr) bool {
var buf [1]byte
n, err := syscall.Read(int(fd), buf[:])
switch {
case n == 0 && err == nil:
sysErr = io.EOF
case n > 0:
sysErr = errUnexpectedRead
case err == syscall.EAGAIN || err == syscall.EWOULDBLOCK:
sysErr = nil
default:
sysErr = err
}
return true
})
switch {
case rerr != nil:
return rerr
case n == 0 && err == nil:
return io.EOF
case n > 0:
return errUnexpectedRead
case err == syscall.EAGAIN || err == syscall.EWOULDBLOCK:
return nil
default:
if err != nil {
return err
}

return sysErr
}
2 changes: 1 addition & 1 deletion conncheck_dummy.go
Expand Up @@ -12,6 +12,6 @@ package mysql

import "net"

func connCheck(c net.Conn) error {
func connCheck(conn net.Conn) error {
return nil
}

0 comments on commit b2c03bc

Please sign in to comment.