can pub and sub

weed/mq/client/cmd/weed_sub/subscriber.go

@@ -1,50 +1,30 @@
 package main
 
 import (
-	"context"
 	"fmt"
-	"github.com/seaweedfs/seaweedfs/weed/pb"
-	"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/credentials/insecure"
+	"github.com/seaweedfs/seaweedfs/weed/mq/client/sub_client"
 )
 
 func main() {
-	err := pb.WithBrokerGrpcClient(true,
-		"localhost:17777",
-		grpc.WithTransportCredentials(insecure.NewCredentials()),
-		func(client mq_pb.SeaweedMessagingClient) error {
-			subClient, err := client.Subscribe(context.Background(), &mq_pb.SubscribeRequest{
-				Init: &mq_pb.SubscribeRequest_InitMessage{
-					Topic: &mq_pb.Topic{
-						Namespace: "test",
-						Name:      "test",
-					},
-				},
-			})
-			if err != nil {
-				return err
-			}
 
-			for {
-				resp, err := subClient.Recv()
-				if err != nil {
-					return err
-				}
-				if resp.GetCtrl() != nil {
-					if resp.GetCtrl().Error != "" {
-						return fmt.Errorf("ctrl error: %v", resp.GetCtrl().Error)
-					}
-				}
-				if resp.GetData() != nil {
-					println(string(resp.GetData().Key), "=>", string(resp.GetData().Value))
-				}
+	subscriber := sub_client.NewTopicSubscriber(
+		&sub_client.SubscriberConfiguration{
+			ConsumerGroup: "test",
+			ConsumerId:    "test",
+		},
+		"test", "test")
+	if err := subscriber.Connect("localhost:17777"); err != nil {
+		fmt.Println(err)
+		return
+	}
 
-			}
-			return nil
-		})
-
-	if err != nil {
+	if err := subscriber.Subscribe(func(key, value []byte) bool {
+		println(string(key), "=>", string(value))
+		return true
+	}, func() {
+		println("done subscribing")
+	}); err != nil {
 		fmt.Println(err)
 	}
+
 }