From a20c98e61957faa93b4014fbd902f20ab9317a6a Mon Sep 17 00:00:00 2001
From: Jae Kwon <jae@tendermint.com>
Date: Fri, 4 Nov 2016 06:14:49 -0700
Subject: [PATCH] Add Monitor.SetREMA()

---
 flowrate/flowrate.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/flowrate/flowrate.go b/flowrate/flowrate.go
index 1b727721e..e233eae0f 100644
--- a/flowrate/flowrate.go
+++ b/flowrate/flowrate.go
@@ -72,6 +72,14 @@ func (m *Monitor) Update(n int) int {
 	return n
 }
 
+// Hack to set the current rEMA.
+func (m *Monitor) SetREMA(rEMA float64) {
+	m.mu.Lock()
+	m.rEMA = rEMA
+	m.samples++
+	m.mu.Unlock()
+}
+
 // IO is a convenience method intended to wrap io.Reader and io.Writer method
 // execution. It calls m.Update(n) and then returns (n, err) unmodified.
 func (m *Monitor) IO(n int, err error) (int, error) {