aboutsummaryrefslogtreecommitdiffstats
path: root/golang/cmd
diff options
context:
space:
mode:
Diffstat (limited to 'golang/cmd')
-rw-r--r--golang/cmd/fatcatd/root.go14
-rw-r--r--golang/cmd/fatcatd/serve.go50
2 files changed, 21 insertions, 43 deletions
diff --git a/golang/cmd/fatcatd/root.go b/golang/cmd/fatcatd/root.go
index d4c7f161..b5263b06 100644
--- a/golang/cmd/fatcatd/root.go
+++ b/golang/cmd/fatcatd/root.go
@@ -6,7 +6,6 @@ import (
"fmt"
log "github.com/sirupsen/logrus"
- //middleware "github.com/go-openapi/runtime/middleware"
"github.com/spf13/viper"
"github.com/spf13/cobra"
"github.com/getsentry/raven-go"
@@ -25,18 +24,20 @@ func init() {
cobra.OnInitialize(initConfig)
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is ./fatcatd.toml)")
rootCmd.PersistentFlags().BoolP("verbose", "v", false, "increase logging volume")
+ viper.BindPFlag("verbose", rootCmd.PersistentFlags().Lookup("verbose"))
+
+ serveCmd.Flags().String("db-url", "postgres://", "database connecion string")
rootCmd.AddCommand(serveCmd)
}
func initConfig() {
- viper.SetDefault("port", 9411)
- viper.SetDefault("verbose", true)
+ viper.SetDefault("port", 9411)
+ viper.SetDefault("db_url", "postgres://bnewbold@localhost/fatcat")
viper.SetEnvPrefix("FATCAT")
viper.AutomaticEnv()
-
if cfgFile != "" {
// Use config file from the flag.
viper.SetConfigFile(cfgFile)
@@ -54,6 +55,11 @@ func initConfig() {
// not default of stderr
log.SetOutput(os.Stdout);
+ if viper.GetBool("verbose") == true {
+ log.SetLevel(log.DebugLevel)
+ }
+ log.Debug("It's verbose!")
+
raven.SetDSN(viper.GetString("sentry_dsn"));
}
diff --git a/golang/cmd/fatcatd/serve.go b/golang/cmd/fatcatd/serve.go
index dcee4f4f..f1eeb995 100644
--- a/golang/cmd/fatcatd/serve.go
+++ b/golang/cmd/fatcatd/serve.go
@@ -2,17 +2,15 @@
package main
import (
- //"os"
"net/http"
"github.com/spf13/viper"
log "github.com/sirupsen/logrus"
loads "github.com/go-openapi/loads"
- //flags "github.com/jessevdk/go-flags"
- //"github.com/getsentry/raven-go"
"github.com/carbocation/interpose"
"github.com/carbocation/interpose/adaptors"
"github.com/meatballhat/negroni-logrus"
+ "github.com/bradleyg/go-sentroni"
"github.com/go-pg/pg"
"github.com/spf13/cobra"
@@ -31,8 +29,6 @@ var serveCmd = &cobra.Command{
func main_serve() {
- log.Warn("Starting up...");
-
// load embedded swagger file
swaggerSpec, err := loads.Analyzed(restapi.SwaggerJSON, "")
if err != nil {
@@ -49,36 +45,13 @@ func main_serve() {
server := restapi.NewServer(api)
defer server.Shutdown()
-/*
- parser := flags.NewParser(server, flags.Default)
- parser.ShortDescription = "fatcat"
- parser.LongDescription = "A scalable, versioned, API-oriented catalog of bibliographic entities and file metadata"
-
- server.ConfigureFlags()
- for _, optsGroup := range api.CommandLineOptionsGroups {
- _, err := parser.AddGroup(optsGroup.ShortDescription, optsGroup.LongDescription, optsGroup.Options)
- if err != nil {
- log.Fatalln(err)
- }
- }
-
- if _, err := parser.Parse(); err != nil {
- code := 1
- if fe, ok := err.(*flags.Error); ok {
- if fe.Type == flags.ErrHelp {
- code = 0
- }
- }
- os.Exit(code)
- }
-*/
-
server.Port = viper.GetInt("port")
- db := pg.Connect(&pg.Options{
- User: "bnewbold", // XXX:
- Database: "fatcat",
- })
+ db_options, err := pg.ParseURL(viper.GetString("db_url"))
+ if err != nil {
+ log.Panicf("parsing DB string: {}", err)
+ }
+ db := pg.Connect(db_options)
defer db.Close()
// register all the many handlers here
@@ -87,20 +60,19 @@ func main_serve() {
middle := interpose.New()
- // sentry and upstream
- // XXX: middle.UseHandler(sentry.Recovery(raven.DefaultClient, false))
+ // sentry
+ middle.Use(adaptors.FromNegroni(sentroni.NewRecovery(viper.GetString("sentry_dsn"))))
// logging
- negroniMiddleware := negronilogrus.NewMiddleware()
- middle.Use(adaptors.FromNegroni(negroniMiddleware))
+ middle.Use(adaptors.FromNegroni(negronilogrus.NewMiddleware()))
- // add clacks (TODO: only production)
+ // add clacks
middle.UseHandler(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
rw.Header().Set("X-Clacks-Overhead:", "GNU Aaron Swartz, John Perry Barlow")
}))
+ // actual handler
middle.UseHandler(api.Serve(nil))
-
server.SetHandler(middle)
if err := server.Serve(); err != nil {