func Start(site, pool string) *server.Server { datastore, err := resolver.NewDatastore("dir:./json") if err != nil { logging.Errorp(err.Error()) os.Exit(1) } configstore, err := config_resolver.NewConfigstore("stub:") if err != nil { logging.Errorp("Could not connect to configstore", logging.Pair{"error", err}, ) } acctstore, err := acct_resolver.NewAcctstore("stub:") if err != nil { logging.Errorp("Could not connect to acctstore", logging.Pair{"error", err}, ) } channel := make(server.RequestChannel, 10) server, err := server.NewServer(datastore, configstore, acctstore, "json", false, channel, 4, 0, false, false, server.KEEP_ALIVE_DEFAULT) if err != nil { logging.Errorp(err.Error()) os.Exit(1) } go server.Serve() return server }
func TestStub(t *testing.T) { logger := NewLogger(os.Stdout, logging.Debug, false) logging.SetLogger(logger) logger.Infof("This is a message from %s", "test") logging.Infof("This is a message from %s", "test") logger.Infop("This is a message from ", logging.Pair{"name", "test"}, logging.Pair{"Queue Size", 10}, logging.Pair{"Debug Mode", false}) logging.Infop("This is a message from ", logging.Pair{"name", "test"}) logger.Infom("This is a message from ", logging.Map{"name": "test", "Queue Size": 10, "Debug Mode": false}) logging.Infom("This is a message from ", logging.Map{"name": "test"}) logger.Requestf(logging.Warn, "This is a Request from %s", "test") logging.Requestf(logging.Info, "This is a Request from %s", "test") logger.Requestp(logging.Debug, "This is a Request from ", logging.Pair{"name", "test"}) logging.Requestp(logging.Error, "This is a Request from ", logging.Pair{"name", "test"}) logger.SetLevel(logging.Warn) fmt.Printf("Log level is %s\n", logger.Level()) logger.Requestf(logging.Warn, "This is a Request from %s", "test") logging.Requestf(logging.Info, "This is a Request from %s", "test") logger.Requestp(logging.Debug, "This is a Request from ", logging.Pair{"name", "test"}) logging.Requestp(logging.Error, "This is a Request from ", logging.Pair{"name", "test"}) logger.Warnf("This is a message from %s", "test") logging.Infof("This is a message from %s", "test") logger.Debugp("This is a message from ", logging.Pair{"name", "test"}) logging.Errorp("This is a message from ", logging.Pair{"name", "test"}) fmt.Printf("Changing to json formatter\n") logger.entryFormatter = &jsonFormatter{} logger.SetLevel(logging.Debug) logger.Infof("This is a message from %s", "test") logging.Infof("This is a message from %s", "test") logger.Infop("This is a message from ", logging.Pair{"name", "test"}, logging.Pair{"Queue Size", 10}, logging.Pair{"Debug Mode", false}) logging.Infop("This is a message from ", logging.Pair{"name", "test"}) logger.Infom("This is a message from ", logging.Map{"name": "test", "Queue Size": 10, "Debug Mode": false}) logging.Infom("This is a message from ", logging.Map{"name": "test"}) logger.Requestf(logging.Warn, "This is a Request from %s", "test") logging.Requestf(logging.Info, "This is a Request from %s", "test") logger.Requestp(logging.Debug, "This is a Request from ", logging.Pair{"name", "test"}) logging.Requestp(logging.Error, "This is a Request from ", logging.Pair{"name", "test"}) }
func pollStdin() { reader := bufio.NewReader(os.Stdin) logging.Infop("pollEOF: About to start stdin polling") for { ch, err := reader.ReadByte() if err == io.EOF { logging.Infop("Received EOF; Exiting...") os.Exit(0) } if err != nil { logging.Errorp("Unexpected error polling stdin", logging.Pair{"error", err}) os.Exit(1) } if ch == '\n' || ch == '\r' { logging.Infop("Received EOL; Exiting...") // TODO: "graceful" shutdown should be placed here os.Exit(0) } } }
func main() { flag.Parse() var f *os.File if *MEM_PROFILE != "" { var err error f, err = os.Create(*MEM_PROFILE) if err != nil { fmt.Printf("Cannot start mem profiler %v\n", err) } } if *CPU_PROFILE != "" { f, err := os.Create(*CPU_PROFILE) if err != nil { fmt.Printf("Cannot start cpu profiler %v\n", err) } else { pprof.StartCPUProfile(f) } } if *LOGGER != "" { logger, _ := log_resolver.NewLogger(*LOGGER) if logger == nil { fmt.Printf("Invalid logger: %s\n", *LOGGER) os.Exit(1) } logging.SetLogger(logger) } if *DEBUG { logging.SetLevel(logging.Debug) logging.Debugp("Debug mode enabled") } else { logging.SetLevel(logging.Info) } datastore, err := resolver.NewDatastore(*DATASTORE) if err != nil { logging.Errorp(err.Error()) os.Exit(1) } datastore_package.SetDatastore(datastore) configstore, err := config_resolver.NewConfigstore(*CONFIGSTORE) if err != nil { logging.Errorp("Could not connect to configstore", logging.Pair{"error", err}, ) } acctstore, err := acct_resolver.NewAcctstore(*ACCTSTORE) if err != nil { logging.Errorp("Could not connect to acctstore", logging.Pair{"error", err}, ) } else { // Create the metrics we are interested in accounting.RegisterMetrics(acctstore) // Make metrics available acctstore.MetricReporter().Start(1, 1) } keep_alive_length, e := util.ParseQuantity(*KEEP_ALIVE_LENGTH) if e != nil { logging.Errorp("Error parsing keep alive length; reverting to default", logging.Pair{"keep alive length", *KEEP_ALIVE_LENGTH}, logging.Pair{"error", e}, logging.Pair{"default", server.KEEP_ALIVE_DEFAULT}, ) } if e == nil && keep_alive_length < 1 { logging.Infop("Negative or zero keep alive length; reverting to default", logging.Pair{"keep alive length", *KEEP_ALIVE_LENGTH}, logging.Pair{"default", server.KEEP_ALIVE_DEFAULT}, ) } channel := make(server.RequestChannel, *REQUEST_CAP) server, err := server.NewServer(datastore, configstore, acctstore, *NAMESPACE, *READONLY, channel, *THREAD_COUNT, *TIMEOUT, *SIGNATURE, *METRICS, keep_alive_length) if err != nil { logging.Errorp(err.Error()) os.Exit(1) } go server.Serve() logging.Infop("cbq-engine started", logging.Pair{"version", VERSION}, logging.Pair{"datastore", *DATASTORE}, ) // Create http endpoint endpoint := http.NewServiceEndpoint(server, *STATIC_PATH, *METRICS) er := endpoint.Listen(*HTTP_ADDR) if er != nil { logging.Errorp("cbq-engine exiting with error", logging.Pair{"error", er}, logging.Pair{"HTTP_ADDR", *HTTP_ADDR}, ) os.Exit(1) } if *CERT_FILE != "" && *KEY_FILE != "" { er := endpoint.ListenTLS(*HTTPS_ADDR, *CERT_FILE, *KEY_FILE) if er != nil { logging.Errorp("cbq-engine exiting with error", logging.Pair{"error", er}, logging.Pair{"HTTP_ADDR", *HTTP_ADDR}, ) os.Exit(1) } } signalCatcher(server, endpoint, *CPU_PROFILE != "", f) }