/
main.go
87 lines (71 loc) · 1.78 KB
/
main.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
package main
import (
"flag"
"fmt"
"os"
"github.com/Sirupsen/logrus"
"github.com/docker/docker/pkg/signal"
"github.com/docker/go-plugins-helpers/volume"
"github.com/docker/libkv/store/boltdb"
"github.com/docker/libkv/store/consul"
"github.com/docker/libkv/store/etcd"
"github.com/docker/libkv/store/zookeeper"
)
const (
dockerPluginSocketPath = "/var/run/docker/plugins"
)
func init() {
consul.Register()
boltdb.Register()
etcd.Register()
zookeeper.Register()
}
type stringsFlag []string
func (s *stringsFlag) String() string {
return fmt.Sprintf("%v", *s)
}
func (s *stringsFlag) Set(val string) error {
*s = append(*s, val)
return nil
}
func (s *stringsFlag) GetAll() []string {
var out []string
for _, i := range *s {
out = append(out, i)
}
return out
}
var (
flAddrs stringsFlag
flStore = flag.String("store", "", "Set the KV store type to use")
flHome = flag.String("home", "/var/run/kvfs", "home dir for volume storage")
flDebug = flag.Bool("debug", false, "enable debug logging")
flListen = flag.String("listen", dockerPluginSocketPath+"/kvfs.sock", "socket to listen for connections on")
)
func main() {
flag.Var(&flAddrs, "addr", "List of address to KV store")
flag.Parse()
if len(flAddrs) == 0 {
logrus.Fatal("need at least one addr to connect to kv store")
}
if *flDebug {
logrus.SetLevel(logrus.DebugLevel)
}
if _, err := os.Stat(*flHome); err != nil {
if !os.IsNotExist(err) {
logrus.Fatal(err)
}
logrus.Debugf("created home dir at %s", *flHome)
if err := os.MkdirAll(*flHome, 0700); err != nil {
logrus.Fatal(err)
}
}
kvfs := newDriver(*flStore, *flHome, flAddrs.GetAll())
signal.Trap(func() {
kvfs.cleanup()
})
h := volume.NewHandler(kvfs)
if err := h.ServeUnix("root", *flListen); err != nil {
logrus.Fatal(err)
}
}