aboutsummaryrefslogtreecommitdiffstats
path: root/simulation/network-model.go
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
}