added custom counters

counter/counter.go

@@ -4,7 +4,8 @@ import "sync/atomic"
 
 //Counter represents a counter
 type Counter struct {
-	Count int64 `json:",omitempty"`
+	Count  int64 `json:",omitempty"`
+	Custom CustomCounter
 }
 
 //CountValue returns count

counter/multi.go

@@ -43,7 +43,12 @@ func (c *MultiCounter) incrementValueBy(value interface{}, count, i int64) int64
 	}
 	if _, ok := value.(string); !ok {
 		c.locker.Lock()
-		c.Counters[idx].Value = toolbox.AsString(value)
+		if custom, ok := value.(CustomCounter); ok {
+			c.Counters[idx].Custom.Aggregate(custom)
+			value = c.Counters[idx].Custom
+		} else {
+			c.Counters[idx].Value = toolbox.AsString(value)
+		}
 		c.locker.Unlock()
 	}
 

counter/operation.go

@@ -65,11 +65,20 @@ func NewOperation(timeUnit time.Duration, provider Provider) *Operation {
 			locker:   &sync.Mutex{},
 		},
 	}
+	customProvider, _ := provider.(CustomProvider)
 	if provider != nil {
 		op.Counters = make([]*Value, len(provider.Keys()))
 		for i, val := range provider.Keys() {
+			var customCounter CustomCounter
+			if customProvider != nil {
+				customCounter = customProvider.NewCounter()
+			}
 			op.Counters[i] = &Value{
 				Value: val,
+				Counter: Counter{
+					Count:  0,
+					Custom: customCounter,
+				},
 			}
 		}
 	}

counter/provider.go

@@ -7,3 +7,11 @@ type Provider interface {
 	//Keys  returns mapped keys
 	Keys() []string
 }
+
+type CustomCounter interface {
+	Aggregate(value interface{})
+}
+
+type CustomProvider interface {
+	NewCounter() CustomCounter
+}