From fe7bb0fe758d573fe353df028257ed66c6eccf66 Mon Sep 17 00:00:00 2001
From: Valery Piashchynski <piashchynski.valery@gmail.com>
Date: Fri, 18 Jun 2021 01:06:16 +0300
Subject: - Rework main parts

Signed-off-by: Valery Piashchynski <piashchynski.valery@gmail.com>
---
 pkg/interface/broadcast/broadcast.go |  4 +--
 pkg/interface/pubsub/interface.go    | 55 ++++++++++++++++++++++++++++++++++++
 2 files changed, 57 insertions(+), 2 deletions(-)
 create mode 100644 pkg/interface/pubsub/interface.go

(limited to 'pkg/interface')

diff --git a/pkg/interface/broadcast/broadcast.go b/pkg/interface/broadcast/broadcast.go
index c922c82e..4c49f7c5 100644
--- a/pkg/interface/broadcast/broadcast.go
+++ b/pkg/interface/broadcast/broadcast.go
@@ -1,7 +1,7 @@
 package broadcast
 
-import "github.com/spiral/roadrunner/v2/pkg/pubsub"
+import "github.com/spiral/roadrunner/v2/pkg/interface/pubsub"
 
 type Broadcaster interface {
-	GetDriver(key string) pubsub.SubReader
+	GetDriver(key string) (pubsub.SubReader, error)
 }
diff --git a/pkg/interface/pubsub/interface.go b/pkg/interface/pubsub/interface.go
new file mode 100644
index 00000000..30b544db
--- /dev/null
+++ b/pkg/interface/pubsub/interface.go
@@ -0,0 +1,55 @@
+package pubsub
+
+import websocketsv1 "github.com/spiral/roadrunner/v2/pkg/proto/websockets/v1beta"
+
+/*
+This interface is in BETA. It might be changed.
+*/
+
+// PubSub interface designed to implement on any storage type to provide pub-sub abilities
+// Publisher used to receive messages from the PHP app via RPC
+// Subscriber should be implemented to subscribe to a topics and provide a connections list per topic
+// Reader return next message from the channel
+type PubSub interface {
+	Publisher
+	Subscriber
+	Reader
+}
+
+type SubReader interface {
+	Subscriber
+	Reader
+}
+
+// Subscriber defines the ability to operate as message passing broker.
+// BETA interface
+type Subscriber interface {
+	// Subscribe broker to one or multiple topics.
+	Subscribe(connectionID string, topics ...string) error
+
+	// Unsubscribe from one or multiply topics
+	Unsubscribe(connectionID string, topics ...string) error
+
+	// Connections returns all connections associated with the particular topic
+	Connections(topic string, ret map[string]struct{})
+}
+
+// Publisher publish one or more messages
+// BETA interface
+type Publisher interface {
+	// Publish one or multiple Channel.
+	Publish(messages []byte) error
+
+	// PublishAsync publish message and return immediately
+	// If error occurred it will be printed into the logger
+	PublishAsync(messages []byte)
+}
+
+// Reader interface should return next message
+type Reader interface {
+	Next() (*websocketsv1.Message, error)
+}
+
+type PSProvider interface {
+	PSProvide(key string) (PubSub, error)
+}
-- 
cgit v1.2.3