feat: lock map while writing

metrics.go

@@ -3,12 +3,14 @@ package logstash
 import (
 	"encoding/json"
 	"errors"
+	"sync"
 )
 
 // Metrics represents a metric that will be sent to logstash
 type Metrics struct {
 	data   map[string]interface{}
 	metric string
+	sync.RWMutex
 }
 
 // NewMetrics Metric{} constructor
@@ -22,6 +24,9 @@ func NewMetrics(metric string) *Metrics {
 }
 
 func (m *Metrics) register(name string, value interface{}) error {
+	m.RLock()
+	defer m.RUnlock()
+
 	if name == "" {
 		return errors.New("Invalid metric name")
 	}
@@ -50,6 +55,5 @@ func (m *Metrics) Clear() {
 // ToJSON serializes data to json
 func (m *Metrics) ToJSON() []byte {
 	data, _ := json.Marshal(m.data)
-	data = append(data, "\n"...)
 	return data
 }

reporter.go

@@ -38,8 +38,6 @@ func NewReporter(r metrics.Registry, addr string, name string) (*Reporter, error
 		return nil, err
 	}
 
-	conn.SetWriteBuffer(2048)
-
 	return &Reporter{
 		Conn:     conn,
 		Registry: r,
@@ -67,12 +65,11 @@ func (r *Reporter) FlushEach(interval time.Duration) {
 	}
 }
 
-// FlushOnce submits a snapshot submission of the registry.
+// FlushOnce submits a snapshot of the registry.
 func (r *Reporter) FlushOnce() error {
 	m := NewMetrics(r.Name)
 
 	r.Registry.Each(func(name string, i interface{}) {
-
 		switch metric := i.(type) {
 		case metrics.Counter:
 			v := metric.Count()
@@ -128,6 +125,5 @@ func (r *Reporter) FlushOnce() error {
 	})
 	r.Conn.Write(m.ToJSON())
 	m.Clear()
-
 	return nil
 }