package timeout

import (
	"net"
	"orly.dev/pkg/utils/chk"
	"time"
)

// Conn extends deadline after successful read or write operations
type Conn struct {
	time.Duration
	*net.TCPConn
}

func (c Conn) Read(b []byte) (n int, e error) {
	if n, e = c.TCPConn.Read(b); !chk.E(e) {
		if e = c.SetDeadline(c.getTimeout()); chk.E(e) {
		}
	}
	return
}

func (c Conn) Write(b []byte) (n int, e error) {
	if n, e = c.TCPConn.Write(b); !chk.E(e) {
		if e = c.SetDeadline(c.getTimeout()); chk.E(e) {
		}
	}
	return
}

func (c Conn) getTimeout() (t time.Time) { return time.Now().Add(c.Duration) }