package http import ( "fmt" "io/ioutil" "net/http" "os" "strings" "github.com/Shugyousha/stasher/input" "github.com/Shugyousha/stasher/registry" "github.com/Shugyousha/stasher/work" ) func init() { registry.Inputregistry["http"] = New } type HTTPInput struct { retchan chan *work.Work prefix string port string } func New(conf map[string]string) input.Input { prefix := conf["prefix"] if prefix == "" { fmt.Fprintf(os.Stderr, "Need a prefix when setting up http input. Exiting.\n") os.Exit(1) } prefix = strings.Replace(prefix, "\"", "", -1) port := conf["port"] if port == "" { fmt.Fprintf(os.Stderr, "Need a port number when setting up http input. Exiting.\n") os.Exit(1) } port = strings.Replace(port, "\"", "", -1) if port[0] != ':' { port = ":" + port } return &HTTPInput{prefix: prefix, port: port} } func (hi *HTTPInput) httphandler(w http.ResponseWriter, r *http.Request) { if r.Method != "POST" { fmt.Printf("Expected POST method was: %q\n", r.Method) return } all, err := ioutil.ReadAll(r.Body) wrk := &work.Work{Data: all} if err != nil { fmt.Printf("Error when reading HTTP request body: %q\n", err) wrk.SetError(err) } hi.retchan <- wrk } func (hi *HTTPInput) Start() chan *work.Work { hi.retchan = make(chan *work.Work, 100) go func() { http.HandleFunc("/"+hi.prefix, hi.httphandler) err := http.ListenAndServe(hi.port, nil) fmt.Printf("Error when serving HTTP: %q\n", err) close(hi.retchan) }() return hi.retchan }