/
main.go
109 lines (91 loc) · 2.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package main
import (
"flag"
"fmt"
"log"
"net/http"
"os"
"os/signal"
"strings"
"time"
"github.com/dadgar/onecache/onecache"
_ "net/http/pprof"
)
var (
defaultConfig = onecache.DefaultNodeConfig()
join = flag.String("join", "", "A comma seperated list of peers (ip:port/dns) to join")
retry_join = flag.String("retry_join", "", "A comma seperated list of peers (ip:port/dns) to join with retries enabled")
retry_interval = flag.Duration("retry_interval", 30*time.Second, "The interval to try retrying a join if there was an error")
port = flag.Int("port", defaultConfig.RpcPort, "Port to bind to for memcache connections")
gossipPort = flag.Int("gossip_port", defaultConfig.SerfConfig.MemberlistConfig.BindPort, "Port to use for membership gossiping")
addr = flag.String("addr", "", "Address to bind to")
replicas = flag.Int("replicas", defaultConfig.Replicas, "Number of replicas")
maxMemory = flag.Int64("max_memory", defaultConfig.MaxMemoryUsage, "Maximum memory to use (in bytes)")
pprofEnabled = flag.Bool("pprof", false, "Enable pprof for debugging")
pprofPort = flag.Int("pprof_port", 6060, "Pprof http port")
)
func main() {
flag.Parse()
config := buildConfig()
logger := log.New(config.LogOutput, "", log.LstdFlags)
node, err := onecache.Create(config)
if err != nil {
logger.Printf("[ERROR] onecache: could not create node: %v\n", err)
os.Exit(1)
}
// Handle interupts.
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
node.Start()
if *join != "" {
peerNodes := strings.Split(*join, ",")
if err := node.Join(peerNodes); err != nil {
logger.Printf("[ERROR] onecache: could not join peers %v: %v", peerNodes, err)
node.Exit()
return
}
}
if *retry_join != "" {
var retry *time.Timer
retryNodes := strings.Split(*retry_join, ",")
retry = time.AfterFunc(0, func() {
if err := node.Join(retryNodes); err != nil {
logger.Printf("[DEBUG] onecache: could not join peers %v: %v\n", retryNodes, err)
logger.Printf("[DEBUG] onecache: retrying in %v", *retry_interval)
retry.Reset(*retry_interval)
} else {
retry.Stop()
}
})
}
if *pprofEnabled {
go func() {
logger.Println(http.ListenAndServe(fmt.Sprintf(":%d", *pprofPort), nil))
}()
}
// Exit the node upon an interupt.
for range c {
node.Exit()
return
}
}
func buildConfig() *onecache.NodeConfig {
config := onecache.DefaultNodeConfig()
config.LogOutput = os.Stderr
if *port != 0 {
config.RpcPort = *port
}
if *gossipPort != 0 {
config.SerfConfig.MemberlistConfig.BindPort = *gossipPort
}
if *addr != "" {
config.SerfConfig.MemberlistConfig.BindAddr = *addr
}
if *replicas != 0 {
config.Replicas = *replicas
}
if *maxMemory != -1 {
config.MaxMemoryUsage = *maxMemory
}
return config
}