add traffic sniffing for handler
This commit is contained in:
+15
-1
@@ -1,6 +1,9 @@
|
||||
package io
|
||||
|
||||
import "io"
|
||||
import (
|
||||
"io"
|
||||
"time"
|
||||
)
|
||||
|
||||
type readWriter struct {
|
||||
io.Reader
|
||||
@@ -27,3 +30,14 @@ func NewReadWriteCloser(r io.Reader, w io.Writer, c io.Closer) io.ReadWriteClose
|
||||
Closer: c,
|
||||
}
|
||||
}
|
||||
|
||||
type setReadDeadline interface {
|
||||
SetReadDeadline(t time.Time) error
|
||||
}
|
||||
|
||||
func SetReadDeadline(rw io.ReadWriter, t time.Time) error {
|
||||
if v, _ := rw.(setReadDeadline); v != nil {
|
||||
return v.SetReadDeadline(t)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user