Skip to content

Instantly share code, notes, and snippets.

@akerl
Created November 8, 2018 01:19
Show Gist options
  • Save akerl/fc80f88981fcfcf6c54c69674c4f3d2b to your computer and use it in GitHub Desktop.
Save akerl/fc80f88981fcfcf6c54c69674c4f3d2b to your computer and use it in GitHub Desktop.
package main
import (
"fmt"
"log"
"net/http"
"github.com/akerl/go-lambda/s3"
"github.com/aws/aws-lambda-go/lambda"
)
type target struct {
URL string `json:"url"`
Method string `json:"method,omitempty"`
}
type config struct {
Targets map[string]target `json:"targets"`
}
type targetList map[string]*http.Request
var targets targetList
func handler() error {
if len(targets) == 0 {
return fmt.Errorf("no targets found in config")
}
client := &http.Client{}
for name, req := range targets {
_, err := client.Do(req)
if err != nil {
return fmt.Errorf("failed to hit url: %s", name)
}
}
log.Printf("success: hit %d urls", len(targets))
return nil
}
func loadConfig() {
c := config{}
cf, err := s3.GetConfigFromEnv(&c)
if err != nil {
log.Print(err)
return
}
cf.OnError = func(_ *s3.ConfigFile, err error) {
log.Print(err)
}
cf.Autoreload(60)
tl := make(targetList)
for name, t := range c.Targets {
req, err := http.NewRequest(t.Method, t.URL, nil)
if err != nil {
log.Print(err)
return
}
tl[name] = req
}
targets = tl
}
func main() {
loadConfig()
lambda.Start(handler)
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment