//Router sets up the Router - extracted for testability
func router(services map[string]Service, healthHandler func(http.ResponseWriter, *http.Request)) *mux.Router {
m := mux.NewRouter()
gtgChecker := make([]gtg.StatusChecker, 0)
for path, service := range services {
handlers := httpHandlers{service}
m.HandleFunc(fmt.Sprintf("/%s/__count", path), handlers.countHandler).Methods("GET")
m.HandleFunc(fmt.Sprintf("/%s/__ids", path), handlers.idsHandler).Methods("GET")
m.HandleFunc(fmt.Sprintf("/%s/{uuid}", path), handlers.getHandler).Methods("GET")
m.HandleFunc(fmt.Sprintf("/%s/{uuid}", path), handlers.putHandler).Methods("PUT")
m.HandleFunc(fmt.Sprintf("/%s/{uuid}", path), handlers.deleteHandler).Methods("DELETE")
gtgChecker = append(gtgChecker, func() gtg.Status {
if err := service.Check(); err != nil {
return gtg.Status{GoodToGo: false, Message: err.Error()}
}
return gtg.Status{GoodToGo: true}
})
}
m.HandleFunc("/__health", healthHandler)
// The top one of these feels more correct, but the lower one matches what we have in Dropwizard,
// so it's what apps expect currently
m.HandleFunc(status.PingPath, status.PingHandler)
m.HandleFunc(status.PingPathDW, status.PingHandler)
// The top one of these feels more correct, but the lower one matches what we have in Dropwizard,
// so it's what apps expect currently same as ping, the content of build-info needs more definition
m.HandleFunc(status.BuildInfoPath, status.BuildInfoHandler)
m.HandleFunc(status.BuildInfoPathDW, status.BuildInfoHandler)
m.HandleFunc(status.GTGPath, status.NewGoodToGoHandler(gtg.FailFastParallelCheck(gtgChecker)))
return m
}
func main() {
app := cli.App("annotations-rw-neo4j", "A RESTful API for managing Annotations in neo4j")
neoURL := app.String(cli.StringOpt{
Name: "neo-url",
Value: "http://localhost:7474/db/data",
Desc: "neo4j endpoint URL",
EnvVar: "NEO_URL",
})
graphiteTCPAddress := app.String(cli.StringOpt{
Name: "graphiteTCPAddress",
Value: "",
Desc: "Graphite TCP address, e.g. graphite.ft.com:2003. Leave as default if you do NOT want to output to graphite (e.g. if running locally",
EnvVar: "GRAPHITE_ADDRESS",
})
graphitePrefix := app.String(cli.StringOpt{
Name: "graphitePrefix",
Value: "",
Desc: "Prefix to use. Should start with content, include the environment, and the host name. e.g. coco.pre-prod.roles-rw-neo4j.1 or content.test.people.rw.neo4j.ftaps58938-law1a-eu-t",
EnvVar: "GRAPHITE_PREFIX",
})
port := app.Int(cli.IntOpt{
Name: "port",
Value: 8080,
Desc: "Port to listen on",
EnvVar: "APP_PORT",
})
batchSize := app.Int(cli.IntOpt{
Name: "batchSize",
Value: 1024,
Desc: "Maximum number of statements to execute per batch",
EnvVar: "BATCH_SIZE",
})
logMetrics := app.Bool(cli.BoolOpt{
Name: "logMetrics",
Value: false,
Desc: "Whether to log metrics. Set to true if running locally and you want metrics output",
EnvVar: "LOG_METRICS",
})
logLevel := app.String(cli.StringOpt{
Name: "log-level",
Value: "INFO",
Desc: "Logging level (DEBUG, INFO, WARN, ERROR)",
EnvVar: "LOG_LEVEL",
})
platformVersion := app.String(cli.StringOpt{
Name: "platformVersion",
Value: "",
Desc: "Annotation source platform. Possible values are: v1 or v2.",
EnvVar: "PLATFORM_VERSION",
})
app.Action = func() {
parsedLogLevel, err := log.ParseLevel(*logLevel)
if err != nil {
log.WithFields(log.Fields{"logLevel": logLevel, "err": err}).Fatal("Incorrect log level")
}
log.SetLevel(parsedLogLevel)
log.Infof("annotations-rw-neo4j will listen on port: %d, connecting to: %s", *port, *neoURL)
conf := neoutils.DefaultConnectionConfig()
conf.BatchSize = *batchSize
db, err := neoutils.Connect(*neoURL, conf)
if err != nil {
log.Fatalf("Error connecting to neo4j %s", err)
}
annotationsService := annotations.NewCypherAnnotationsService(db, *platformVersion)
httpHandlers := httpHandlers{annotationsService}
// Healthchecks and standards first
http.HandleFunc("/__health", v1a.Handler("Annotations RW Healthchecks",
"Checks for accessing neo4j", httpHandlers.HealthCheck()))
http.HandleFunc(status.PingPath, status.PingHandler)
http.HandleFunc(status.PingPathDW, status.PingHandler)
http.HandleFunc(status.BuildInfoPath, status.BuildInfoHandler)
http.HandleFunc(status.BuildInfoPathDW, status.BuildInfoHandler)
gtgChecker := make([]gtg.StatusChecker, 0)
gtgChecker = append(gtgChecker, func() gtg.Status {
if err := httpHandlers.AnnotationsService.Check(); err != nil {
return gtg.Status{GoodToGo: false, Message: err.Error()}
}
return gtg.Status{GoodToGo: true}
})
http.HandleFunc(status.GTGPath, status.NewGoodToGoHandler(gtg.FailFastParallelCheck(gtgChecker)))
r := router(httpHandlers)
http.Handle("/", r)
baseftrwapp.OutputMetricsIfRequired(*graphiteTCPAddress, *graphitePrefix, *logMetrics)
if err := http.ListenAndServe(fmt.Sprintf(":%d", *port), nil); err != nil {
log.Fatalf("Unable to start server: %v", err)
}
}
log.Infof("Application started with args %s", os.Args)
app.Run(os.Args)
}