Move drivers to the plugin's root.

Fix #771, add tests.

Signed-off-by: Valery Piashchynski <[email protected]>

1 files changed, 75 insertions, 0 deletions

diff --git a/plugins/beanstalk/encode_test.go b/plugins/beanstalk/encode_test.go
new file mode 100644
index 00000000..e43207eb
--- /dev/null
+++ b/plugins/beanstalk/encode_test.go
@@ -0,0 +1,75 @@
+package beanstalk
+
+import (
+	"bytes"
+	"crypto/rand"
+	"encoding/gob"
+	"testing"
+
+	json "github.com/json-iterator/go"
+	"github.com/spiral/roadrunner/v2/utils"
+)
+
+func BenchmarkEncodeGob(b *testing.B) {
+	tb := make([]byte, 1024*10)
+	_, err := rand.Read(tb)
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	item := &Item{
+		Job:     "/super/test/php/class/loooooong",
+		Ident:   "12341234-asdfasdfa-1234234-asdfasdfas",
+		Payload: utils.AsString(tb),
+		Headers: map[string][]string{"Test": {"test1", "test2"}},
+		Options: &Options{
+			Priority: 10,
+			Pipeline: "test-local-pipe",
+			Delay:    10,
+		},
+	}
+
+	b.ResetTimer()
+	b.ReportAllocs()
+
+	for i := 0; i < b.N; i++ {
+		bb := new(bytes.Buffer)
+		err := gob.NewEncoder(bb).Encode(item)
+		if err != nil {
+			b.Fatal(err)
+		}
+		_ = bb.Bytes()
+		bb.Reset()
+	}
+}
+
+func BenchmarkEncodeJsonIter(b *testing.B) {
+	tb := make([]byte, 1024*10)
+	_, err := rand.Read(tb)
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	item := &Item{
+		Job:     "/super/test/php/class/loooooong",
+		Ident:   "12341234-asdfasdfa-1234234-asdfasdfas",
+		Payload: utils.AsString(tb),
+		Headers: map[string][]string{"Test": {"test1", "test2"}},
+		Options: &Options{
+			Priority: 10,
+			Pipeline: "test-local-pipe",
+			Delay:    10,
+		},
+	}
+
+	b.ResetTimer()
+	b.ReportAllocs()
+
+	for i := 0; i < b.N; i++ {
+		bb, err := json.Marshal(item)
+		if err != nil {
+			b.Fatal(err)
+		}
+		_ = bb
+	}
+}