func Main(c *cli.Context) { db, err := bouncer.NewDB(c.String("db-dsn")) if err != nil { log.Fatalf("Could not open DB: %v", err) } defer db.Close() bouncerHandler := &BouncerHandler{ db: db, CacheTime: time.Duration(c.Int("cache-time")) * time.Second, PinnedBaseURLHttp: c.String("pinned-baseurl-http"), PinnedBaseURLHttps: c.String("pinned-baseurl-https"), StubRootURL: c.String("stub-root-url"), } healthHandler := &HealthHandler{ db: db, CacheTime: 5 * time.Second, } mux := http.NewServeMux() mux.Handle("/__heartbeat__", healthHandler) mux.Handle("/", bouncerHandler) server := &http.Server{ Addr: c.String("addr"), Handler: mux, } err = server.ListenAndServe() if err != nil { log.Fatal(err) } }
func init() { testDB, err := bouncer.NewDB("root@tcp(127.0.0.1:3306)/bouncer_test") if err != nil { log.Fatal(err) } bouncerHandler = &BouncerHandler{ db: testDB, StubRootURL: "https://stub/", } bouncerHandlerPinned = &BouncerHandler{ db: testDB, PinnedBaseURLHttp: "download-sha1.cdn.mozilla.net/pub", PinnedBaseURLHttps: "download-sha1.cdn.mozilla.net/pub", } }
func Main(c *cli.Context) { db, err := bouncer.NewDB(c.String("db-dsn")) if err != nil { log.Fatalf("Could not open DB: %v", err) } defer db.Close() sentry, err := sentry.New(db, c.Bool("checknow"), c.String("mirror"), c.Int("mirror-routines"), c.Int("location-routines")) if err != nil { log.Fatal(err) } sentry.Verbose = c.Bool("verbose") err = sentry.Run() if err != nil { log.Fatal(err) } }