/
spanneti.go
121 lines (104 loc) · 2.66 KB
/
spanneti.go
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package spanneti
import (
"context"
"fmt"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/events"
"github.com/docker/docker/client"
"os"
"os/signal"
"syscall"
"github.com/ciena/spanneti/spanneti/graph"
)
var (
GIT_BRANCH string
GIT_COMMIT_NUM string
GIT_COMMIT string
CHANGED string
)
func printLogo() {
displayBranch := ""
if GIT_BRANCH != "master" {
displayBranch = GIT_BRANCH + " "
}
fmt.Println(` _ _ `)
fmt.Println(` ___ _ __ __ _ _ __ _ __ ___| |_(_)`)
fmt.Println("/ __| '_ \\ / _` | '_ \\| '_ \\ / _ \\ __| |")
fmt.Println(`\__ \ |_) | (_| | | | | | | | __/ |_| |`)
fmt.Println(`|___/ .__/ \__,_|_| |_|_| |_|\___|\__|_|`)
if CHANGED == "true" {
fmt.Println(`DEV |_|`, displayBranch+"v0."+GIT_COMMIT_NUM+".x")
fmt.Println("Base:", GIT_COMMIT)
} else {
fmt.Println(` |_|`, displayBranch+"v0."+GIT_COMMIT_NUM)
fmt.Println(GIT_COMMIT)
}
fmt.Println()
//It's simple, we kill the PACketMAN
}
type Spanneti struct {
*spanneti
}
func New() Spanneti {
printLogo()
client, err := client.NewEnvClient()
if err != nil {
panic(err)
}
spanneti := &spanneti{
graph: graph.New(),
client: client,
plugins: make(map[string]*Plugin),
}
return Spanneti{spanneti}
}
func (s *spanneti) Start() {
if s.started {
return
}
s.started = true
//listen for shutdown signals
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
cli, err := client.NewEnvClient()
if err != nil {
panic(err)
}
//1. start listening for changes
eventChan, errChan := cli.Events(context.Background(), types.EventsOptions{})
//2. initialize the network
s.init()
//3. apply changes that happened while network was initializing,
// and continue listening for events
eventLoop(s, eventChan, errChan, sigChan)
}
func eventLoop(spanneti *spanneti, eventChan <-chan events.Message, errChan <-chan error, sigChan <-chan os.Signal) {
for {
select {
case event := <-eventChan:
containerEvent(spanneti, event)
case err := <-errChan:
panic(err)
case signal := <-sigChan:
fmt.Println("Received signal:", signal)
return
}
}
}
func containerEvent(spanneti *spanneti, event events.Message) {
if event.Type == events.ContainerEventType {
if len(event.Actor.ID) >= 12 {
fmt.Println("Container event:", event.Action, string(event.Actor.ID[0:12]))
} else {
fmt.Println("Container event:", event.Action, string(event.Actor.ID))
}
switch event.Action {
case "start":
if err := spanneti.UpdateContainer(event.Actor.ID); err != nil {
fmt.Println(err)
}
case "die":
spanneti.RemoveContainer(event.Actor.ID)
}
}
}