pub sub initial tests

weed/mq/client/pub_client/publisher.go

@@ -1,5 +1,59 @@
 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"
+)
+
 func main() {
 
+	err := pb.WithBrokerGrpcClient(true,
+		"localhost:17777",
+		grpc.WithTransportCredentials(insecure.NewCredentials()),
+		func(client mq_pb.SeaweedMessagingClient) error {
+			pubClient, err := client.Publish(context.Background())
+			if err != nil {
+				return err
+			}
+			if initErr := pubClient.Send(&mq_pb.PublishRequest{
+				Message: &mq_pb.PublishRequest_Init{
+					Init: &mq_pb.PublishRequest_InitMessage{
+						Topic: &mq_pb.Topic{
+							Namespace: "test",
+							Name:      "test",
+						},
+						Partition: &mq_pb.Partition{
+							RangeStart: 0,
+							RangeStop:  1,
+							RingSize:   1,
+						},
+					},
+				},
+			}); initErr != nil {
+				return initErr
+			}
+
+			for i := 0; i < 10; i++ {
+				if dataErr := pubClient.Send(&mq_pb.PublishRequest{
+					Message: &mq_pb.PublishRequest_Data{
+						Data: &mq_pb.DataMessage{
+							Key:   []byte(fmt.Sprintf("key-%d", i)),
+							Value: []byte(fmt.Sprintf("value-%d", i)),
+						},
+					},
+				}); dataErr != nil {
+					return dataErr
+				}
+			}
+			return nil
+		})
+
+	if err != nil {
+		fmt.Println(err)
+	}
+
 }