59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/Difrex/zoorest/rest"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
zk string
|
|
listen string
|
|
path string
|
|
mc bool
|
|
mcHosts string
|
|
mcPrefix string
|
|
)
|
|
|
|
// init ...
|
|
func init() {
|
|
flag.StringVar(&zk, "zk", "127.0.0.1:2181", "Zk servers. Comma separated")
|
|
flag.StringVar(&listen, "listen", "127.0.0.1:8889", "Address to listen")
|
|
flag.StringVar(&path, "path", "/", "Zk root path")
|
|
flag.BoolVar(&mc, "mc", false, "Enable memcached support")
|
|
flag.StringVar(&mcHosts, "mchosts", "127.0.0.1:11211", "Memcached servers. Comma separated")
|
|
flag.StringVar(&mcPrefix, "mcprefix", "zoorest", "Memcached key prefix")
|
|
flag.Parse()
|
|
}
|
|
|
|
// main ...
|
|
func main() {
|
|
var z rest.Zk
|
|
hosts := getSlice(zk, ",")
|
|
z.Hosts = hosts
|
|
conn, err := z.InitConnection()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var zoo rest.ZooNode
|
|
zoo.Path = path
|
|
zoo.Conn = conn
|
|
zoo.Zoo = z
|
|
|
|
var MC rest.MC
|
|
MC.Hosts = getSlice(mcHosts, ",")
|
|
MC.Prefix = mcPrefix
|
|
MC.Enabled = mc
|
|
MC.Client = MC.InitConnection()
|
|
|
|
zoo.MC = MC
|
|
|
|
rest.Serve(listen, zoo)
|
|
}
|
|
|
|
// getSlice returm slice
|
|
func getSlice(s string, delimeter string) []string {
|
|
return strings.Split(s, ",")
|
|
}
|