feat: set read and write timeout (#29)

client.go

@@ -316,6 +316,12 @@ func (hc *HostClient) onConnectionDropped(c *clientConn) {
 
 func (hc *HostClient) createConn() (*clientConn, *list.Element, error) {
 	conn, err := hc.dialHostHard()
+	if hc.ReadTimeout != 0 {
+		conn.SetReadTimeout(hc.ReadTimeout)
+	}
+	if hc.WriteTimeout != 0 {
+		conn.SetWriteTimeout(hc.WriteTimeout)
+	}
 	if err != nil {
 		return nil, nil, err
 	}

config/client_option.go

@@ -107,6 +107,12 @@ type ClientConfig struct {
 
 	// If true, h2 client won't add default user-agent
 	NoDefaultUserAgent bool
+
+	// readtimeout sets timeout when read package
+	ReadTimeout time.Duration
+
+	// writetimeout sets timeout when write package
+	WriteTimeout time.Duration
 }
 
 func (o *ClientConfig) Apply(opts []ClientOption) {
@@ -226,6 +232,20 @@ func WithClientDisableKeepAlive(disable bool) ClientOption {
 	}}
 }
 
+// WithClientReadTimeout is used to set client readtimeout.
+func WithClientReadTimeout(t time.Duration) ClientOption {
+	return ClientOption{F: func(o *ClientConfig) {
+		o.ReadTimeout = t
+	}}
+}
+
+// WithClientWriteTimeout is used to set client writeTimeout.
+func WithClientWriteTimeout(t time.Duration) ClientOption {
+	return ClientOption{F: func(o *ClientConfig) {
+		o.WriteTimeout = t
+	}}
+}
+
 func NewClientConfig(opts ...ClientOption) *ClientConfig {
 	cfg := &ClientConfig{
 		PingTimeout:         consts.DefaultPingTimeout,