From 573c0627041cb2703a7de6a74ae206b2b70f66aa Mon Sep 17 00:00:00 2001
From: Maksadbek <a.maksadbek@gmail.com>
Date: Tue, 26 Jul 2016 23:46:55 +0300
Subject: [PATCH] fixed data races

---
 plugins/outputs/influxdb/influxdb.go      |  2 +-
 plugins/outputs/influxdb/influxdb_test.go | 24 ++++++++++-------------
 2 files changed, 11 insertions(+), 15 deletions(-)

diff --git a/plugins/outputs/influxdb/influxdb.go b/plugins/outputs/influxdb/influxdb.go
index 18d2074fd..3729a1724 100644
--- a/plugins/outputs/influxdb/influxdb.go
+++ b/plugins/outputs/influxdb/influxdb.go
@@ -425,10 +425,10 @@ func (d *Downsampling) Mean(fields ...Aggregation) (telegraf.Metric, error) {
 	var (
 		aggrMetric telegraf.Metric
 		sums       = make(map[string]interface{})
-		size       = len(d.Metrics)
 	)
 
 	d.RLock()
+	var size = len(d.Metrics)
 	for _, metric := range d.Metrics {
 		for _, field := range fields {
 			value, ok := metric.Fields()[field.FieldName]
diff --git a/plugins/outputs/influxdb/influxdb_test.go b/plugins/outputs/influxdb/influxdb_test.go
index e8e411b50..669d947d9 100644
--- a/plugins/outputs/influxdb/influxdb_test.go
+++ b/plugins/outputs/influxdb/influxdb_test.go
@@ -6,7 +6,7 @@ import (
 	"math/rand"
 	"net/http"
 	"net/http/httptest"
-	"sync"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -115,11 +115,9 @@ func TestDownsampling_aggregate(t *testing.T) {
 }
 
 func TestDownsampling_run(t *testing.T) {
-	testCase := struct {
-		sum   int
-		count int
-		sync.Mutex
-	}{}
+	var (
+		sum, count int32
+	)
 
 	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		var createDatabaseQuery = "CREATE DATABASE IF NOT EXISTS \"\""
@@ -151,11 +149,9 @@ func TestDownsampling_run(t *testing.T) {
 			return
 		}
 
-		testCase.Lock()
-		want := testCase.sum / testCase.count
-		testCase.sum = 0
-		testCase.count = 0
-		defer testCase.Unlock()
+		want := atomic.LoadInt32(&sum) / atomic.LoadInt32(&count)
+		atomic.StoreInt32(&sum, 0)
+		atomic.StoreInt32(&count, 0)
 
 		require.EqualValues(t, want, mean)
 
@@ -188,9 +184,9 @@ func TestDownsampling_run(t *testing.T) {
 	for {
 		select {
 		case <-tick:
-			testCase.count += 1
-			val := rand.Intn(120)
-			testCase.sum += val
+			atomic.AddInt32(&count, 1)
+			val := rand.Int31n(120)
+			atomic.AddInt32(&sum, val)
 			err := influxdb.Write([]telegraf.Metric{testutil.TestMetric(val)})
 			require.NoError(t, err)
 		case <-after: