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)
}
}
