pacer: make sleep get-able and set-able

pacer/pacer.go

@@ -76,6 +76,21 @@ func New() *Pacer {
 	return p
 }
 
+// SetSleep sets the current sleep time
+func (p *Pacer) SetSleep(t time.Duration) *Pacer {
+	p.mu.Lock()
+	defer p.mu.Unlock()
+	p.sleepTime = t
+	return p
+}
+
+// GetSleep gets the current sleep time
+func (p *Pacer) GetSleep() time.Duration {
+	p.mu.Lock()
+	defer p.mu.Unlock()
+	return p.sleepTime
+}
+
 // SetMinSleep sets the minimum sleep time for the pacer
 func (p *Pacer) SetMinSleep(t time.Duration) *Pacer {
 	p.mu.Lock()

pacer/pacer_test.go

@@ -51,6 +51,20 @@ func TestNew(t *testing.T) {
 	}
 }
 
+func TestSetSleep(t *testing.T) {
+	p := New().SetSleep(2 * time.Millisecond)
+	if p.sleepTime != 2*time.Millisecond {
+		t.Errorf("didn't set")
+	}
+}
+
+func TestGetSleep(t *testing.T) {
+	p := New().SetSleep(2 * time.Millisecond)
+	if p.GetSleep() != 2*time.Millisecond {
+		t.Errorf("didn't get")
+	}
+}
+
 func TestSetMinSleep(t *testing.T) {
 	p := New().SetMinSleep(1 * time.Millisecond)
 	if p.minSleep != 1*time.Millisecond {