rethinkDB

counter.go

@@ -8,26 +8,28 @@ import (
 
 // Counter structure
 type Counter struct {
-	mu   sync.Mutex
-	user int
-	log  int
-	rem  int
-	err  int
-	dup  int
-	time time.Duration
-	wg   int
+	mu     sync.Mutex
+	user   int
+	log    int
+	insert int
+	rem    int
+	err    int
+	dup    int
+	time   time.Duration
+	wg     int
 }
 
 // NewCounter iniitialized Counter structure
 func NewCounter() *Counter {
 	return &Counter{
-		user: 0,
-		log:  0,
-		err:  0,
-		rem:  0,
-		dup:  0,
-		time: 0,
-		wg:   0,
+		user:   0,
+		log:    0,
+		insert: 0,
+		err:    0,
+		rem:    0,
+		dup:    0,
+		time:   0,
+		wg:     0,
 	}
 }
 
@@ -45,6 +47,13 @@ func (c *Counter) AddDuplicate(add int) {
 	c.dup += add
 }
 
+// AddInsert increment number of inserted rows
+func (c *Counter) AddInsert(add int) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	c.insert += add
+}
+
 // AddLog increment number of log's rows managed
 func (c *Counter) AddLog(add int) {
 	c.mu.Lock()
@@ -104,6 +113,14 @@ func (c *Counter) GetLog() (ret int) {
 	return
 }
 
+// GetInsert return total inserted rows
+func (c *Counter) GetInsert() (ret int) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	ret = c.insert
+	return
+}
+
 // GetErr return total errors
 func (c *Counter) GetErr() (ret int) {
 	c.mu.Lock()