func signalHandler(sig <-chan os.Signal, svr *server.RESTServer) {
for s := range sig {
log.Println("---Received signal", s)
switch s {
case syscall.SIGINT, syscall.SIGTERM:
log.Println("Exiting")
svr.Stop() // this will cause Run to exit
}
}
}
func main() {
// Start with the Default values
config := bendoConfig{
StoreDir: ".",
Tokenfile: "",
CacheDir: "",
CacheSize: 100,
PortNumber: "14000",
PProfPort: "14001",
Mysql: "",
CowHost: "",
CowToken: "",
}
var configFile = flag.String("config-file", "", "Configuration File")
flag.Parse()
// If config file arg provided, try to open & decode it
if *configFile != "" {
log.Printf("Using config file %s\n", *configFile)
if _, err := toml.DecodeFile(*configFile, &config); err != nil {
log.Println(err)
return
}
}
log.Printf("Using storage dir %s\n", config.StoreDir)
log.Printf("Using cache dir %s\n", config.CacheDir)
var validator server.TokenValidator
if config.Tokenfile != "" {
var err error
log.Printf("Using user token file %s\n", config.Tokenfile)
validator, err = server.NewListValidatorFile(config.Tokenfile)
if err != nil {
log.Println(err)
return
}
} else {
log.Printf("No user token file specified")
validator = server.NobodyValidator{}
}
if config.CacheDir != "" {
os.MkdirAll(config.CacheDir, 0755)
}
var itemstore store.Store = store.NewFileSystem(config.StoreDir)
if config.CowHost != "" {
log.Printf("Using COW with target %s", config.CowHost)
itemstore = store.NewCOW(itemstore, config.CowHost, config.CowToken)
}
var s = server.RESTServer{
Items: items.New(itemstore),
Validator: validator,
MySQL: config.Mysql,
CacheDir: config.CacheDir,
CacheSize: config.CacheSize * 1000000,
PortNumber: config.PortNumber,
PProfPort: config.PProfPort,
}
if config.CowHost != "" {
// don't run fixity if we are using a copy-on-write.
// (doing so will cause us to download ALL the data from
// the target bendo over time)
s.DisableFixity = true
}
// set up signal handlers
sig := make(chan os.Signal, 5)
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
go signalHandler(sig, &s)
s.Run()
}