add sub recv timeout support

nanomsg · Jul 30, 2019 · 12bbb93 · 12bbb93
1 parent 012b6c6
commit 12bbb93
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/protocol/sub/sub.go b/protocol/sub/sub.go
@@ -265,6 +265,15 @@ func (c *context) SetOption(name string, value interface{}) error {
 		}
 		return protocol.ErrBadValue
 
+	case protocol.OptionRecvDeadline:
+		if v, ok := value.(time.Duration); ok {
+			c.s.Lock()
+			c.recvExpire = v
+			c.s.Unlock()
+			return nil
+		}
+		return protocol.ErrBadValue
+
 	case protocol.OptionSubscribe:
 	case protocol.OptionUnsubscribe:
 	default: