Example #1
0
File: main.go Project: ndlib/bendo
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
		}
	}
}
Example #2
0
File: main.go Project: ndlib/bendo
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()
}