func main() {
log.SetFlags(log.LstdFlags | log.Lshortfile)
// Set up validation for config from our environment.
envVars := map[string]*re.Regexp{
"PORT": digits,
"DATABASE_URL": nows,
"RELAYMSG_PG_DB": word,
"RELAYMSG_PG_SCHEMA": word,
"RELAYMSG_PG_USER": word,
"RELAYMSG_PG_PASS": nows,
"RELAYMSG_PG_MAX_CONNS": digits,
"RELAYMSG_BATCH_INTERVAL": digits,
"RELAYMSG_INBOUND_DOMAIN": nows,
"RELAYMSG_ALLOWED_ORIGIN": nows,
}
// Config container
cfg := map[string]string{}
for k, v := range envVars {
cfg[k] = os.Getenv(k)
if !v.MatchString(cfg[k]) {
log.Fatalf("Unsupported value for %s, double check your parameters.", k)
}
}
// Set defaults
if cfg["PORT"] == "" {
cfg["PORT"] = "5000"
}
if cfg["RELAYMSG_BATCH_INTERVAL"] == "" {
cfg["RELAYMSG_BATCH_INTERVAL"] = "10"
}
batchInterval, err := strconv.Atoi(cfg["RELAYMSG_BATCH_INTERVAL"])
if err != nil {
log.Fatal(err)
}
if cfg["RELAYMSG_INBOUND_DOMAIN"] == "" {
cfg["RELAYMSG_INBOUND_DOMAIN"] = "hey.avocado.industries"
}
if cfg["RELAYMSG_PG_MAX_CONNS"] == "" {
cfg["RELAYMSG_PG_MAX_CONNS"] = "18"
}
maxConns, err := strconv.Atoi(cfg["RELAYMSG_PG_MAX_CONNS"])
if err != nil {
log.Fatal(err)
}
pgcfg := &pg.PGConfig{
Db: cfg["RELAYMSG_PG_DB"],
User: cfg["RELAYMSG_PG_USER"],
Pass: cfg["RELAYMSG_PG_PASS"],
Opts: map[string]string{
"sslmode": "disable",
},
Url: cfg["DATABASE_URL"],
}
dbh, err := pgcfg.Connect()
if err != nil {
log.Fatal(err)
}
if maxConns > 0 {
dbh.SetMaxOpenConns(maxConns)
}
// Configure PostgreSQL dumper with connection details.
schema := cfg["RELAYMSG_PG_SCHEMA"]
if schema == "" {
schema = "request_dump"
}
pgDumper := &pg.PgDumper{Schema: schema}
// make sure schema and raw_requests table exist
err = pg.SchemaInit(dbh, schema)
if err != nil {
log.Fatal(err)
}
// make sure relay_messages table exists
err = SchemaInit(dbh, schema)
if err != nil {
log.Fatal(err)
}
pgDumper.Dbh = dbh
// Set up our handler which writes to, and reads from PostgreSQL.
reqDumper := storage.HandlerFactory(pgDumper)
// Set up our handler which writes individual events to PostgreSQL.
msgParser := &RelayMsgParser{
Dbh: dbh,
Schema: schema,
Domain: cfg["RELAYMSG_INBOUND_DOMAIN"],
}
// recurring job to transform blobs of webhook data into relay_messages
interval := time.Duration(batchInterval) * time.Second
ticker := time.NewTicker(interval)
go func() {
for {
select {
case <-ticker.C:
go func() {
_, err := storage.ProcessBatch(pgDumper, msgParser)
if err != nil {
log.Printf("%s\n", err)
}
}()
}
}
}()
router := vestigo.NewRouter()
router.SetGlobalCors(&vestigo.CorsAccessControl{
AllowOrigin: []string{cfg["RELAYMSG_ALLOWED_ORIGIN"]},
ExposeHeaders: []string{"accept"},
AllowHeaders: []string{"accept"},
})
// Install handler to store votes in database (incoming webhook events)
router.Post("/incoming", reqDumper)
router.Get("/summary/:localpart", msgParser.SummaryHandler())
portSpec := fmt.Sprintf(":%s", cfg["PORT"])
log.Fatal(http.ListenAndServe(portSpec, router))
}
func main() {
log.SetFlags(log.LstdFlags | log.Lshortfile)
flag.Parse()
// Env vars we'll be checking for, mapped to the regular expressions
// we'll use to validate their values.
envVars := map[string]*re.Regexp{
"LOGGLY_TOKEN": uuid,
"POSTGRESQL_DB": word,
"POSTGRESQL_USER": word,
"POSTGRESQL_PASS": pass,
"POSTGRESQL_SCHEMA": word,
}
opts := map[string]string{}
for k, v := range envVars {
opts[k] = os.Getenv(k)
if !v.MatchString(opts[k]) {
log.Fatalf("Unexpected value for %s, double check your parameters.", k)
}
}
pgcfg := &pg.PGConfig{
Db: opts["POSTGRESQL_DB"],
User: opts["POSTGRESQL_USER"],
Pass: opts["POSTGRESQL_PASS"],
Opts: map[string]string{
"sslmode": "disable",
},
}
dbh, err := pgcfg.Connect()
if err != nil {
log.Fatal(err)
}
// Configure the PostgreSQL dumper.
pgDumper := &pg.PgDumper{Schema: opts["POSTGRESQL_SCHEMA"]}
pgDumper.Dbh = dbh
err = pg.SchemaInit(dbh, pgDumper.Schema)
if err != nil {
log.Fatal(err)
}
// Configure the Loggly processor
loggly := &Loggly{
Endpoint: fmt.Sprintf("https://logs-01.loggly.com/bulk/%s/tag/bulk/", opts["LOGGLY_TOKEN"]),
Client: &http.Client{},
BatchMax: 5 * 1024 * 1024,
EventMax: 1024 * 1024,
}
loggly.buf = bytes.NewBuffer(make([]byte, 0, loggly.BatchMax))
// Set up our handler which writes to, and reads from PostgreSQL.
reqDumper := storage.HandlerFactory(pgDumper)
// Start up recurring job to process events stored in PostgreSQL.
interval := time.Duration(*batchInterval) * time.Second
ticker := time.NewTicker(interval)
go func() {
for {
select {
case <-ticker.C:
go func() {
_, err := storage.ProcessBatch(pgDumper, loggly)
if err != nil {
log.Printf("%s\n", err)
}
}()
}
}
}()
// Spin up HTTP listener on the requested port.
http.HandleFunc("/", reqDumper)
portSpec := fmt.Sprintf(":%d", *port)
log.Fatal(http.ListenAndServe(portSpec, nil))
}
