From 30442ee69239786b171b44f75262b135f2f098ce Mon Sep 17 00:00:00 2001 From: Matt Reiferson Date: Fri, 24 Oct 2014 00:46:32 -0700 Subject: [PATCH] config: allow heartbeat_interval and output_buffer_timeout to be disabled --- conn.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/conn.go b/conn.go index 13528119..e4fc235f 100644 --- a/conn.go +++ b/conn.go @@ -280,10 +280,18 @@ func (c *Conn) identify() (*IdentifyResponse, error) { ci["deflate_level"] = c.config.DeflateLevel ci["snappy"] = c.config.Snappy ci["feature_negotiation"] = true - ci["heartbeat_interval"] = int64(c.config.HeartbeatInterval / time.Millisecond) + if c.config.HeartbeatInterval == -1 { + ci["heartbeat_interval"] = -1 + } else { + ci["heartbeat_interval"] = int64(c.config.HeartbeatInterval / time.Millisecond) + } ci["sample_rate"] = c.config.SampleRate ci["output_buffer_size"] = c.config.OutputBufferSize - ci["output_buffer_timeout"] = int64(c.config.OutputBufferTimeout / time.Millisecond) + if c.config.OutputBufferTimeout == -1 { + ci["output_buffer_timeout"] = -1 + } else { + ci["output_buffer_timeout"] = int64(c.config.OutputBufferTimeout / time.Millisecond) + } ci["msg_timeout"] = int64(c.config.MsgTimeout / time.Millisecond) cmd, err := Identify(ci) if err != nil {