Skip to content

Instantly share code, notes, and snippets.

@snooc

snooc/nettimeout

Forked from jbardin/nettimeout
Created May 3, 2014 11:19
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save snooc/1aec4dd23c64e570e01d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save snooc/1aec4dd23c64e570e01d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. James Bardin revised this gist Mar 20, 2014. 1 changed file with 3 additions and 3 deletions.
    6 changes: 3 additions & 3 deletions nettimeout
    View file
    Original file line number Diff line number Diff line change
    @@ -5,8 +5,8 @@ import (
    "time"
    )

    // ReadWriteTimeout wraps our net.Listener, and gives a place to store the rwTimeout
    // parameter.
    // Listener wraps a net.Listener, and gives a place to store the timeout
    // parameters. On Accept, it will wrap the net.Conn with our own Conn for us.
    type Listener struct {
    net.Listener
    ReadTimeout time.Duration
    @@ -26,7 +26,7 @@ func (l *Listener) Accept() (net.Conn, error) {
    return tc, nil
    }

    // Conn wraps a net.Conn, nad sets a deadline for every read
    // Conn wraps a net.Conn, and sets a deadline for every read
    // and write operation.
    type Conn struct {
    net.Conn
  2. James Bardin created this gist Mar 20, 2014.
    65 changes: 65 additions & 0 deletions nettimeout
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,65 @@
    package nettimeout

    import (
    "net"
    "time"
    )

    // ReadWriteTimeout wraps our net.Listener, and gives a place to store the rwTimeout
    // parameter.
    type Listener struct {
    net.Listener
    ReadTimeout time.Duration
    WriteTimeout time.Duration
    }

    func (l *Listener) Accept() (net.Conn, error) {
    c, err := l.Listener.Accept()
    if err != nil {
    return nil, err
    }
    tc := &Conn{
    Conn: c,
    ReadTimeout: l.ReadTimeout,
    WriteTimeout: l.WriteTimeout,
    }
    return tc, nil
    }

    // Conn wraps a net.Conn, nad sets a deadline for every read
    // and write operation.
    type Conn struct {
    net.Conn
    ReadTimeout time.Duration
    WriteTimeout time.Duration
    }

    func (c *Conn) Read(b []byte) (int, error) {
    err := c.Conn.SetReadDeadline(time.Now().Add(c.ReadTimeout))
    if err != nil {
    return 0, err
    }
    return c.Conn.Read(b)
    }

    func (c *Conn) Write(b []byte) (int, error) {
    err := c.Conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout))
    if err != nil {
    return 0, err
    }
    return c.Conn.Write(b)
    }

    func NewListener(addr string, readTimeout, writeTimeout time.Duration) (net.Listener, error) {
    l, err := net.Listen("tcp", addr)
    if err != nil {
    return nil, err
    }

    tl := &Listener{
    Listener: l,
    ReadTimeout: readTimeout,
    WriteTimeout: writeTimeout,
    }
    return tl, nil
    }