From 9cd652ef0b783b8ee547d6573d4d81a8041dc903 Mon Sep 17 00:00:00 2001
From: Matthias Radestock <matthias.radestock@gmail.com>
Date: Wed, 24 Dec 2014 13:29:13 +0000
Subject: [PATCH] refactor: introduce small helper

---
 router/connection.go | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/router/connection.go b/router/connection.go
index a02bd1b1..35c1d2b5 100644
--- a/router/connection.go
+++ b/router/connection.go
@@ -179,15 +179,9 @@ func (conn *LocalConnection) queryLoop(queryChan <-chan *ConnectionInteraction,
 			conn.log("error:", err)
 			break
 		}
-		if conn.heartbeat != nil {
-			heartbeat = conn.heartbeat.C
-		}
-		if conn.fetchAll != nil {
-			fetchAll = conn.fetchAll.C
-		}
-		if conn.fragTest != nil {
-			fragTest = conn.fragTest.C
-		}
+		heartbeat = tickerChan(conn.heartbeat)
+		fetchAll = tickerChan(conn.fetchAll)
+		fragTest = tickerChan(conn.fragTest)
 		select {
 		case query, ok := <-queryChan:
 			if !ok {
@@ -543,6 +537,13 @@ func (conn *LocalConnection) forwardHeartbeatFrame() {
 	conn.Forward(true, heartbeatFrame, nil)
 }
 
+func tickerChan(ticker *time.Ticker) <-chan time.Time {
+	if ticker != nil {
+		return ticker.C
+	}
+	return nil
+}
+
 func stopTicker(ticker *time.Ticker) {
 	if ticker != nil {
 		ticker.Stop()