From dfae737da3763e76b34d1ee203944fe53c68fc2c Mon Sep 17 00:00:00 2001
From: Martin Geno <geno+dev@fireorbit.de>
Date: Tue, 4 Oct 2016 14:54:19 +0200
Subject: [PATCH] fix bugs after refactoring

---
 database/database.go | 2 +-
 respond/collector.go | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/database/database.go b/database/database.go
index 5e88dc9..12a8034 100644
--- a/database/database.go
+++ b/database/database.go
@@ -105,7 +105,7 @@ func (db *DB) worker() {
 		}
 
 		// write batch now?
-		if bp != nil && (writeNow || closed) || len(bp.Points()) >= batchMaxSize {
+		if bp != nil && (writeNow || closed || len(bp.Points()) >= batchMaxSize) {
 			log.Println("saving", len(bp.Points()), "points")
 
 			if err = db.client.Write(bp); err != nil {
diff --git a/respond/collector.go b/respond/collector.go
index d971b4d..3b6284b 100644
--- a/respond/collector.go
+++ b/respond/collector.go
@@ -45,6 +45,7 @@ func NewCollector(db *database.DB, nodes *models.Nodes, interval time.Duration,
 
 	collector := &Collector{
 		connection:    conn,
+		db:            db,
 		nodes:         nodes,
 		multicastAddr: net.JoinHostPort(multiCastGroup+"%"+iface, port),
 		queue:         make(chan *Response, 400),