blob: 041bd12028a592bb9d085d095e5e680d49c6aee3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
package simulation
import (
"math/rand"
"time"
)
// Model is the interface for define a given network environment.
type Model interface {
// LossRate returns the message lost ratio between [0, 1)
LossRate() float64
// Delay returns the send delay of the message. This function is called each
// time before the message is sent, so one can return different number each
// time.
Delay() time.Duration
}
// LosslessNetwork is a lossless network model.
type LosslessNetwork struct {
}
// LossRate returns lossrate for the model.
func (l *LosslessNetwork) LossRate() float64 {
return 0.0
}
// Delay returns the send delay of a given message.
func (l *LosslessNetwork) Delay() time.Duration {
return time.Duration(0)
}
// FixedLostNoDelayModel is a network with no delay and a fixed lost
// ratio.
type FixedLostNoDelayModel struct {
LossRateValue float64
}
// LossRate returns lossrate for the model.
func (f *FixedLostNoDelayModel) LossRate() float64 {
return f.LossRateValue
}
// Delay returns the send delay of a given message.
func (f *FixedLostNoDelayModel) Delay() time.Duration {
return time.Duration(0)
}
// NormalNetwork is a model where it's delay is a normal distribution.
type NormalNetwork struct {
Sigma float64
Mean float64
LossRateValue float64
}
// LossRate returns lossrate for the model.
func (n *NormalNetwork) LossRate() float64 {
return n.LossRateValue
}
// Delay returns the send delay of a given message.
func (n *NormalNetwork) Delay() time.Duration {
delay := rand.NormFloat64()*n.Sigma + n.Mean
if delay < 0 {
delay = n.Sigma / 2
}
return time.Duration(delay) * time.Millisecond
}
|