apollo-backend/internal/cmd/root.go
André Medeiros 9341e08123 feat(ops): add distributed tracing (#109)
* tracing

* instrument redis

* add instrumentation around queue acks

* adjust backoff schedules for reddit

* set proper http client on reddit

* populate account last message ID on upsert

* pass last message id on upsert

* schedule accounts in a predictable order
2022-11-05 15:59:33 -04:00

89 lines
1.8 KiB
Go

package cmd
import (
"context"
"net/http"
_ "net/http/pprof"
"os"
"runtime"
"runtime/pprof"
"github.com/bugsnag/bugsnag-go/v2"
_ "github.com/heroku/x/hmetrics/onload"
"github.com/joho/godotenv"
"github.com/spf13/cobra"
_ "github.com/honeycombio/honeycomb-opentelemetry-go"
"github.com/honeycombio/opentelemetry-go-contrib/launcher"
)
func Execute(ctx context.Context) int {
_ = godotenv.Load()
if key, ok := os.LookupEnv("BUGSNAG_API_KEY"); ok {
bugsnag.Configure(bugsnag.Configuration{
APIKey: key,
ReleaseStage: os.Getenv("ENV"),
ProjectPackages: []string{"main", "github.com/christianselig/apollo-backend"},
AppVersion: os.Getenv("RENDER_GIT_COMMIT"),
})
}
otelShutdown, err := launcher.ConfigureOpenTelemetry()
if err == nil {
defer otelShutdown()
}
profile := false
rootCmd := &cobra.Command{
Use: "apollo",
Short: "Apollo is an amazing Reddit client for iOS. This isn't it, but it helps.",
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
if !profile {
return nil
}
f, perr := os.Create("cpu.pprof")
if perr != nil {
return perr
}
_ = pprof.StartCPUProfile(f)
return nil
},
PersistentPostRunE: func(cmd *cobra.Command, args []string) error {
if !profile {
return nil
}
pprof.StopCPUProfile()
f, perr := os.Create("mem.pprof")
if perr != nil {
return perr
}
defer f.Close()
runtime.GC()
err := pprof.WriteHeapProfile(f)
return err
},
}
rootCmd.PersistentFlags().BoolVarP(&profile, "profile", "p", false, "record CPU pprof")
rootCmd.AddCommand(APICmd(ctx))
rootCmd.AddCommand(SchedulerCmd(ctx))
rootCmd.AddCommand(WorkerCmd(ctx))
go func() {
_ = http.ListenAndServe("localhost:6060", nil)
}()
if err := rootCmd.Execute(); err != nil {
return 1
}
return 0
}