mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
Urban IshimweandGitHub
fdc8b094f0
### performance - handling of the very big packet(any size that can be buffered) - speeding up TCP sessions by using message hints: Added **proto.HasFullPayload** that helps to validate the entire HTTP request, it supports `Chunked` encoding too! Added **proto.HasRequestTitle** and **proto.HasResponseTitle** for validating the beginning of HTTP request. Those methods are used `input_raw.go` with `TCP`. - supports Keep-Alive: the above functions helps to support keep-alive ### Packaging - **capture:** engines(capture/doc.go) - **tcp:** tcp message parser (tcp/doc.go) ### benchmarking - **capture.BenchmarkPcapDump:** the benchmarks regarding dumping packets in a pcap file - **capture.BenchmarkPcapFile:** the benchmarks of reading packets from a pcap file - **capture.BenchmarkPcap:** the benchmarks of parsing packets from the loopback interface with pcap handles - **proto.BenchmarkHasFullPayload:**: benchmarking this function which validates the HTTP payload - **tcp.BenchmarkPacketParseAndSort:** benchmarks of parsing and sorting packets - **tcp.BenchmarkMessageParserWithoutHint:** benchmarks of message reasembling by using `SYN` and `FIN` flag - **tcp.BenchmarkMessageParserWithHint:** benchmarks of message reasembling by using `proto.HasRequestTitle` and `proto.HasFullPayload` flag ### issues see linked issues ### tests - fixed input raw and engine tests **Most of the changed of the files, was about using functionalities of** `tcp` **and** `capture` **in existing functionalities**
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Shopify/sarama"
|
|
"github.com/Shopify/sarama/mocks"
|
|
)
|
|
|
|
func TestOutputKafkaRAW(t *testing.T) {
|
|
config := sarama.NewConfig()
|
|
config.Producer.Return.Successes = true
|
|
producer := mocks.NewAsyncProducer(t, config)
|
|
producer.ExpectInputAndSucceed()
|
|
|
|
output := NewKafkaOutput("", &OutputKafkaConfig{
|
|
producer: producer,
|
|
Topic: "test",
|
|
UseJSON: false,
|
|
})
|
|
|
|
output.Write([]byte("1 2 3\nGET / HTTP1.1\r\nHeader: 1\r\n\r\n"))
|
|
|
|
resp := <-producer.Successes()
|
|
|
|
data, _ := resp.Value.Encode()
|
|
|
|
if string(data) != "1 2 3\nGET / HTTP1.1\r\nHeader: 1\r\n\r\n" {
|
|
t.Error("Message not properly encoded: ", string(data))
|
|
}
|
|
}
|
|
|
|
func TestOutputKafkaJSON(t *testing.T) {
|
|
config := sarama.NewConfig()
|
|
config.Producer.Return.Successes = true
|
|
producer := mocks.NewAsyncProducer(t, config)
|
|
producer.ExpectInputAndSucceed()
|
|
|
|
output := NewKafkaOutput("", &OutputKafkaConfig{
|
|
producer: producer,
|
|
Topic: "test",
|
|
UseJSON: true,
|
|
})
|
|
|
|
output.Write([]byte("1 2 3\nGET / HTTP1.1\r\nHeader: 1\r\n\r\n"))
|
|
|
|
resp := <-producer.Successes()
|
|
|
|
data, _ := resp.Value.Encode()
|
|
|
|
if string(data) != `{"Req_URL":"","Req_Type":"1","Req_ID":"2","Req_Ts":"3","Req_Method":"GET"}` {
|
|
t.Error("Message not properly encoded: ", string(data))
|
|
}
|
|
}
|