Added variable latency delay, normal and uniform based

Author: rikonor

thirdparty/delay/delay.go

@@ -1,6 +1,7 @@
 package delay
 
 import (
+	"math/rand"
 	"sync"
 	"time"
 )
@@ -23,8 +24,6 @@ type delay struct {
 	t time.Duration
 }
 
-// TODO func Variable(time.Duration) D returns a delay with probablistic latency
-
 func (d *delay) Set(t time.Duration) time.Duration {
 	d.l.Lock()
 	defer d.l.Unlock()
@@ -44,3 +43,53 @@ func (d *delay) Get() time.Duration {
 	defer d.l.Unlock()
 	return d.t
 }
+
+// VariableNormal is a delay following a normal distribution
+// Notice that to implement the D interface Set can only change the mean delay
+// the standard deviation is set only at initialization
+func VariableNormal(t, std time.Duration) D {
+	v := &variableNormal{
+		std: std,
+		rng: rand.New(rand.NewSource(int64(time.Now().Nanosecond()))),
+	}
+	v.t = t
+	return v
+}
+
+type variableNormal struct {
+	delay
+	std time.Duration
+	rng *rand.Rand
+}
+
+func (d *variableNormal) Wait() {
+	d.l.RLock()
+	defer d.l.RUnlock()
+	randomDelay := time.Duration(d.rng.NormFloat64() * float64(d.std))
+	time.Sleep(randomDelay + d.t)
+}
+
+// VariableUniform is a delay following a uniform distribution
+// Notice that to implement the D interface Set can only change the minimum delay
+// the delta is set only at initialization
+func VariableUniform(t, d time.Duration) D {
+	v := &variableUniform{
+		d:   d,
+		rng: rand.New(rand.NewSource(int64(time.Now().Nanosecond()))),
+	}
+	v.t = t
+	return v
+}
+
+type variableUniform struct {
+	delay
+	d   time.Duration // max delta
+	rng *rand.Rand
+}
+
+func (d *variableUniform) Wait() {
+	d.l.RLock()
+	defer d.l.RUnlock()
+	randomDelay := time.Duration(d.rng.Float64() * float64(d.d))
+	time.Sleep(randomDelay + d.t)
+}