use deadline within http transport

transport/http_transport.go

@@ -42,6 +42,7 @@ type httpTransportClient struct {
 }
 
 type httpTransportSocket struct {
+	ht   *httpTransport
 	r    chan *http.Request
 	conn net.Conn
 	once sync.Once
@@ -51,6 +52,7 @@ type httpTransportSocket struct {
 }
 
 type httpTransportListener struct {
+	ht       *httpTransport
 	listener net.Listener
 }
 
@@ -144,6 +146,11 @@ func (h *httpTransportClient) Send(m *Message) error {
 	}
 	h.Unlock()
 
+	// set deadline if its greater than 0
+	if h.ht.opts.Deadline > time.Duration(0) {
+		h.conn.SetDeadline(time.Now().Add(h.ht.opts.Deadline))
+	}
+
 	return req.Write(h.conn)
 }
 
@@ -163,6 +170,11 @@ func (h *httpTransportClient) Recv(m *Message) error {
 		return io.EOF
 	}
 
+	// set deadline if its greater than 0
+	if h.ht.opts.Deadline > time.Duration(0) {
+		h.conn.SetDeadline(time.Now().Add(h.ht.opts.Deadline))
+	}
+
 	rsp, err := http.ReadResponse(h.buff, r)
 	if err != nil {
 		return err
@@ -212,6 +224,11 @@ func (h *httpTransportSocket) Recv(m *Message) error {
 		return errors.New("message passed in is nil")
 	}
 
+	// set deadline if its greater than 0
+	if h.ht.opts.Deadline > time.Duration(0) {
+		h.conn.SetDeadline(time.Now().Add(h.ht.opts.Deadline))
+	}
+
 	r, err := http.ReadRequest(h.buff)
 	if err != nil {
 		return err
@@ -271,6 +288,11 @@ func (h *httpTransportSocket) Send(m *Message) error {
 	default:
 	}
 
+	// set deadline if its greater than 0
+	if h.ht.opts.Deadline > time.Duration(0) {
+		h.conn.SetDeadline(time.Now().Add(h.ht.opts.Deadline))
+	}
+
 	return rsp.Write(h.conn)
 }
 
@@ -337,6 +359,7 @@ func (h *httpTransportListener) Accept(fn func(Socket)) error {
 		}
 
 		sock := &httpTransportSocket{
+			ht:   h.ht,
 			conn: c,
 			buff: bufio.NewReader(c),
 			r:    make(chan *http.Request, 1),
@@ -444,6 +467,7 @@ func (h *httpTransport) Listen(addr string, opts ...ListenOption) (Listener, err
 	}
 
 	return &httpTransportListener{
+		ht:       h,
 		listener: l,
 	}, nil
 }