diff options
Diffstat (limited to 'weed/mq/client/cmd/weed_sub/subscriber.go')
| -rw-r--r-- | weed/mq/client/cmd/weed_sub/subscriber.go | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/weed/mq/client/cmd/weed_sub/subscriber.go b/weed/mq/client/cmd/weed_sub/subscriber.go index 6cb18c574..d5bd8f12d 100644 --- a/weed/mq/client/cmd/weed_sub/subscriber.go +++ b/weed/mq/client/cmd/weed_sub/subscriber.go @@ -1,13 +1,23 @@ package main import ( + "flag" "fmt" "github.com/seaweedfs/seaweedfs/weed/mq/client/sub_client" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" + "strings" + "time" +) + +var ( + namespace = flag.String("ns", "test", "namespace") + topic = flag.String("topic", "test", "topic") + seedBrokers = flag.String("brokers", "localhost:17777", "seed brokers") ) func main() { + flag.Parse() subscriberConfig := &sub_client.SubscriberConfiguration{ ClientId: "testSubscriber", @@ -17,12 +27,14 @@ func main() { } contentConfig := &sub_client.ContentConfiguration{ - Namespace: "test", - Topic: "test", + Namespace: *namespace, + Topic: *topic, Filter: "", + StartTime: time.Now(), } - subscriber := sub_client.NewTopicSubscriber("localhost:17777", subscriberConfig, contentConfig) + brokers := strings.Split(*seedBrokers, ",") + subscriber := sub_client.NewTopicSubscriber(brokers, subscriberConfig, contentConfig) subscriber.SetEachMessageFunc(func(key, value []byte) bool { println(string(key), "=>", string(value)) |