From a1e562e2482c49952006c99168b1494851003f45 Mon Sep 17 00:00:00 2001
From: Radu Ioan Fericean <radu@fericean.ro>
Date: Fri, 1 Aug 2014 20:13:53 +0300
Subject: [PATCH] Added ResetQueue and cleaned cdrs on queue conf update

---
 engine/stats.go       | 35 +++++++++++++++++++++++++++++++++++
 engine/stats_queue.go |  6 ++++--
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/engine/stats.go b/engine/stats.go
index 3a163d19c..3b11780a1 100644
--- a/engine/stats.go
+++ b/engine/stats.go
@@ -33,6 +33,7 @@ type StatsInterface interface {
 	AppendCDR(*utils.StoredCdr, *int) error
 	AddQueue(*CdrStats, *int) error
 	ReloadQueues([]string, *int) error
+	ResetQueues([]string, *int) error
 }
 
 type Stats struct {
@@ -104,6 +105,36 @@ func (s *Stats) ReloadQueues(ids []string, out *int) error {
 	return nil
 }
 
+func (s *Stats) ResetQueues(ids []string, out *int) error {
+	if ids == nil {
+		for _, sq := range s.queues {
+			sq.cdrs = make([]*QCdr, sq.conf.QueueLength)
+			sq.metrics = make(map[string]Metric, len(sq.conf.Metrics))
+			for _, m := range sq.conf.Metrics {
+				if metric := CreateMetric(m); metric != nil {
+					sq.metrics[m] = metric
+				}
+			}
+		}
+	} else {
+		for _, id := range ids {
+			sq, exists := s.queues[id]
+			if !exists {
+				Logger.Err(fmt.Sprintf("Cannot reset queue id %v: Not Fund", id))
+				continue
+			}
+			sq.cdrs = make([]*QCdr, sq.conf.QueueLength)
+			sq.metrics = make(map[string]Metric, len(sq.conf.Metrics))
+			for _, m := range sq.conf.Metrics {
+				if metric := CreateMetric(m); metric != nil {
+					sq.metrics[m] = metric
+				}
+			}
+		}
+	}
+	return nil
+}
+
 // change the xisting ones
 // add new ones
 // delete the ones missing from the new list
@@ -169,3 +200,7 @@ func (ps *ProxyStats) AddQueue(cs *CdrStats, out *int) error {
 func (ps *ProxyStats) ReloadQueues(ids []string, out *int) error {
 	return ps.Client.Call("Stats.ReloadQueues", ids, out)
 }
+
+func (ps *ProxyStats) ResetQueues(ids []string, out *int) error {
+	return ps.Client.Call("Stats.ReserQueues", ids, out)
+}
diff --git a/engine/stats_queue.go b/engine/stats_queue.go
index 9907c397e..fce96c1fd 100644
--- a/engine/stats_queue.go
+++ b/engine/stats_queue.go
@@ -51,11 +51,13 @@ func NewStatsQueue(conf *CdrStats) *StatsQueue {
 }
 
 func (sq *StatsQueue) UpdateConf(conf *CdrStats) {
+	sq.mux.Lock()
+	defer sq.mux.Unlock()
 	sq.conf = conf
+	sq.cdrs = make([]*QCdr, conf.QueueLength)
 	sq.metrics = make(map[string]Metric, len(conf.Metrics))
 	for _, m := range conf.Metrics {
-		metric := CreateMetric(m)
-		if metric != nil {
+		if metric := CreateMetric(m); metric != nil {
 			sq.metrics[m] = metric
 		}
 	}