mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package http
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"time"
|
|
|
|
"github.com/buger/goreplay/pkg/plugin"
|
|
"github.com/buger/goreplay/pkg/proto"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
var inputLogger = log.With().Str("component", "input_http").Logger()
|
|
|
|
// HTTPInput used for sending requests to Gor via http
|
|
type HTTPInput struct {
|
|
data chan []byte
|
|
address string
|
|
listener net.Listener
|
|
stop chan bool // Channel used only to indicate goroutine should shutdown
|
|
}
|
|
|
|
// NewHTTPInput constructor for HTTPInput. Accepts address with port which it will listen on.
|
|
func NewHTTPInput(address string) (i *HTTPInput) {
|
|
i = new(HTTPInput)
|
|
i.data = make(chan []byte, 1000)
|
|
i.stop = make(chan bool)
|
|
|
|
i.listen(address)
|
|
|
|
return
|
|
}
|
|
|
|
// PluginRead reads message from this plugin
|
|
func (i *HTTPInput) PluginRead() (*plugin.Message, error) {
|
|
var msg plugin.Message
|
|
select {
|
|
case <-i.stop:
|
|
return nil, plugin.ErrorStopped
|
|
case buf := <-i.data:
|
|
msg.Data = buf
|
|
msg.Meta = proto.PayloadHeader(proto.RequestPayload, proto.UUID(), time.Now().UnixNano(), -1)
|
|
return &msg, nil
|
|
}
|
|
}
|
|
|
|
// Close closes this plugin
|
|
func (i *HTTPInput) Close() error {
|
|
close(i.stop)
|
|
return nil
|
|
}
|
|
|
|
func (i *HTTPInput) handler(w http.ResponseWriter, r *http.Request) {
|
|
r.URL.Scheme = "http"
|
|
r.URL.Host = i.address
|
|
|
|
buf, _ := httputil.DumpRequestOut(r, true)
|
|
http.Error(w, http.StatusText(200), 200)
|
|
i.data <- buf
|
|
}
|
|
|
|
func (i *HTTPInput) listen(address string) {
|
|
var err error
|
|
|
|
mux := http.NewServeMux()
|
|
|
|
mux.HandleFunc("/", i.handler)
|
|
|
|
i.listener, err = net.Listen("tcp", address)
|
|
if err != nil {
|
|
inputLogger.Fatal().Err(err).Msg("HTTP input listener failure")
|
|
}
|
|
i.address = i.listener.Addr().String()
|
|
|
|
go func() {
|
|
err = http.Serve(i.listener, mux)
|
|
if err != nil && err != http.ErrServerClosed {
|
|
inputLogger.Fatal().Err(err).Msg("HTTP input serve failure")
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (i *HTTPInput) String() string {
|
|
return "HTTP input: " + i.address
|
|
}
|