- Switch from the flatbuffers to the protobuf

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

1 files changed, 12 insertions, 67 deletions

diff --git a/tests/plugins/websockets/websocket_plugin_test.go b/tests/plugins/websockets/websocket_plugin_test.go
index b2c756bf..5f472106 100644
--- a/tests/plugins/websockets/websocket_plugin_test.go
+++ b/tests/plugins/websockets/websocket_plugin_test.go
@@ -13,11 +13,9 @@ import (
 	"time"
 
 	"github.com/fasthttp/websocket"
-	flatbuffers "github.com/google/flatbuffers/go"
 	json "github.com/json-iterator/go"
 	endure "github.com/spiral/endure/pkg/container"
 	goridgeRpc "github.com/spiral/goridge/v3/pkg/rpc"
-	"github.com/spiral/roadrunner/v2/pkg/pubsub"
 	"github.com/spiral/roadrunner/v2/pkg/pubsub/message"
 	"github.com/spiral/roadrunner/v2/plugins/config"
 	httpPlugin "github.com/spiral/roadrunner/v2/plugins/http"
@@ -877,8 +875,8 @@ func publish2(command string, broker string, topics ...string) {
 		panic(err)
 	}
 }
-func messageWS(command string, broker string, payload []byte, topics ...string) *Msg {
-	return &Msg{
+func messageWS(command string, broker string, payload []byte, topics ...string) *message.Message {
+	return &message.Message{
 		Topics:  topics,
 		Command: command,
 		Broker:  broker,
@@ -886,70 +884,17 @@ func messageWS(command string, broker string, payload []byte, topics ...string)
 	}
 }
 
-func makeMessage(command string, broker string, payload []byte, topics ...string) []byte {
-	m := []pubsub.Message{
-		{
-			Topics:  topics,
-			Command: command,
-			Broker:  broker,
-			Payload: payload,
+func makeMessage(command string, broker string, payload []byte, topics ...string) *message.Messages {
+	m := &message.Messages{
+		Messages: []*message.Message{
+			{
+				Topics:  topics,
+				Command: command,
+				Broker:  broker,
+				Payload: payload,
+			},
 		},
 	}
 
-	b := flatbuffers.NewBuilder(1)
-
-	return msgs(b, m)
-}
-
-func serializeMsg(b *flatbuffers.Builder, msg pubsub.Message) flatbuffers.UOffsetT {
-	cmdOff := b.CreateString(msg.Command)
-	brokerOff := b.CreateString(msg.Broker)
-
-	offsets := make([]flatbuffers.UOffsetT, len(msg.Topics))
-	for j := len(msg.Topics) - 1; j >= 0; j-- {
-		offsets[j] = b.CreateString(msg.Topics[j])
-	}
-
-	message.MessageStartTopicsVector(b, len(offsets))
-
-	for j := len(offsets) - 1; j >= 0; j-- {
-		b.PrependUOffsetT(offsets[j])
-	}
-
-	tOff := b.EndVector(len(offsets))
-	pOff := b.CreateByteVector(msg.Payload)
-
-	message.MessageStart(b)
-
-	message.MessageAddCommand(b, cmdOff)
-	message.MessageAddBroker(b, brokerOff)
-	message.MessageAddTopics(b, tOff)
-	message.MessageAddPayload(b, pOff)
-
-	return message.MessageEnd(b)
-}
-
-func msgs(b *flatbuffers.Builder, msgs []pubsub.Message) []byte {
-	b.Reset()
-
-	mOff := make([]flatbuffers.UOffsetT, len(msgs))
-
-	for i := len(msgs) - 1; i >= 0; i-- {
-		mOff[i] = serializeMsg(b, msgs[i])
-	}
-
-	message.MessagesStartMessagesVector(b, len(mOff))
-
-	for i := len(mOff) - 1; i >= 0; i-- {
-		b.PrependUOffsetT(mOff[i])
-	}
-
-	msgsOff := b.EndVector(len(msgs))
-
-	message.MessagesStart(b)
-	message.MessagesAddMessages(b, msgsOff)
-	fOff := message.MessagesEnd(b)
-	b.Finish(fOff)
-
-	return b.Bytes[b.Head():]
+	return m
 }