95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
// Copyright 2013 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package net
|
|
|
|
import (
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
type pollDesc struct {
|
|
fd *netFD
|
|
closing bool
|
|
}
|
|
|
|
func (pd *pollDesc) Init(fd *netFD) error { pd.fd = fd; return nil }
|
|
|
|
func (pd *pollDesc) Close() {}
|
|
|
|
func (pd *pollDesc) Lock() {}
|
|
|
|
func (pd *pollDesc) Unlock() {}
|
|
|
|
func (pd *pollDesc) Wakeup() {}
|
|
|
|
func (pd *pollDesc) Evict() bool {
|
|
pd.closing = true
|
|
if pd.fd != nil {
|
|
syscall.StopIO(pd.fd.sysfd)
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (pd *pollDesc) Prepare(mode int) error {
|
|
if pd.closing {
|
|
return errClosing
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (pd *pollDesc) PrepareRead() error { return pd.Prepare('r') }
|
|
|
|
func (pd *pollDesc) PrepareWrite() error { return pd.Prepare('w') }
|
|
|
|
func (pd *pollDesc) Wait(mode int) error {
|
|
if pd.closing {
|
|
return errClosing
|
|
}
|
|
return errTimeout
|
|
}
|
|
|
|
func (pd *pollDesc) WaitRead() error { return pd.Wait('r') }
|
|
|
|
func (pd *pollDesc) WaitWrite() error { return pd.Wait('w') }
|
|
|
|
func (pd *pollDesc) WaitCanceled(mode int) {}
|
|
|
|
func (pd *pollDesc) WaitCanceledRead() {}
|
|
|
|
func (pd *pollDesc) WaitCanceledWrite() {}
|
|
|
|
func (fd *netFD) setDeadline(t time.Time) error {
|
|
return setDeadlineImpl(fd, t, 'r'+'w')
|
|
}
|
|
|
|
func (fd *netFD) setReadDeadline(t time.Time) error {
|
|
return setDeadlineImpl(fd, t, 'r')
|
|
}
|
|
|
|
func (fd *netFD) setWriteDeadline(t time.Time) error {
|
|
return setDeadlineImpl(fd, t, 'w')
|
|
}
|
|
|
|
func setDeadlineImpl(fd *netFD, t time.Time, mode int) error {
|
|
d := t.UnixNano()
|
|
if t.IsZero() {
|
|
d = 0
|
|
}
|
|
if err := fd.incref(); err != nil {
|
|
return err
|
|
}
|
|
switch mode {
|
|
case 'r':
|
|
syscall.SetReadDeadline(fd.sysfd, d)
|
|
case 'w':
|
|
syscall.SetWriteDeadline(fd.sysfd, d)
|
|
case 'r' + 'w':
|
|
syscall.SetReadDeadline(fd.sysfd, d)
|
|
syscall.SetWriteDeadline(fd.sysfd, d)
|
|
}
|
|
fd.decref()
|
|
return nil
|
|
}
|