| 122 | } |
| 123 | |
| 124 | func (w *connWriter) Write(b []byte) (int, error) { |
| 125 | if timeout := w.WriterTimeout; timeout != 0 { |
| 126 | if err := w.Sock.SetWriteDeadline(time.Now().Add(timeout)); err != nil { |
| 127 | return 0, errors.Trace(err) |
| 128 | } |
| 129 | w.hasDeadline = true |
| 130 | } else if w.hasDeadline { |
| 131 | if err := w.Sock.SetWriteDeadline(time.Time{}); err != nil { |
| 132 | return 0, errors.Trace(err) |
| 133 | } |
| 134 | w.hasDeadline = false |
| 135 | } |
| 136 | n, err := w.Sock.Write(b) |
| 137 | if err != nil { |
| 138 | err = errors.Trace(err) |
| 139 | } |
| 140 | w.LastWrite = time.Now() |
| 141 | return n, err |
| 142 | } |
| 143 | |
| 144 | func IsTimeout(err error) bool { |
| 145 | if err := errors.Cause(err); err != nil { |