publish, benchmark

weed/mq/client/pub_client/publisher.go

@@ -1,32 +1,37 @@
 package pub_client
 
 import (
-	cmap "github.com/orcaman/concurrent-map/v2"
 	"github.com/rdleal/intervalst/interval"
 	"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
+	"sync"
 )
 
 type PublisherConfiguration struct {
 }
+
+type PublishClient struct {
+	mq_pb.SeaweedMessaging_PublishClient
+	Broker string
+	Err    error
+}
 type TopicPublisher struct {
-	namespace            string
-	topic                string
-	partition2Broker     *interval.SearchTree[string, int32]
-	broker2PublishClient cmap.ConcurrentMap[string, mq_pb.SeaweedMessaging_PublishClient]
-	grpcDialOption       grpc.DialOption
+	namespace        string
+	topic            string
+	partition2Broker *interval.SearchTree[*PublishClient, int32]
+	grpcDialOption   grpc.DialOption
+	sync.Mutex       // protects grpc
 }
 
 func NewTopicPublisher(namespace, topic string) *TopicPublisher {
 	return &TopicPublisher{
 		namespace: namespace,
 		topic:     topic,
-		partition2Broker: interval.NewSearchTree[string](func(a, b int32) int {
+		partition2Broker: interval.NewSearchTree[*PublishClient](func(a, b int32) int {
 			return int(a - b)
 		}),
-		broker2PublishClient: cmap.New[mq_pb.SeaweedMessaging_PublishClient](),
-		grpcDialOption:       grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpcDialOption: grpc.WithTransportCredentials(insecure.NewCredentials()),
 	}
 }