mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
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",
|
|
ConsumerGroup: "test",
|
|
ConsumerGroupInstanceId: "test",
|
|
GrpcDialOption: grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
}
|
|
|
|
contentConfig := &sub_client.ContentConfiguration{
|
|
Namespace: *namespace,
|
|
Topic: *topic,
|
|
Filter: "",
|
|
StartTime: time.Unix(0, 0),
|
|
}
|
|
|
|
processorConfig := sub_client.ProcessorConfiguration{
|
|
ConcurrentPartitionLimit: 6,
|
|
}
|
|
|
|
brokers := strings.Split(*seedBrokers, ",")
|
|
subscriber := sub_client.NewTopicSubscriber(brokers, subscriberConfig, contentConfig, processorConfig)
|
|
|
|
subscriber.SetEachMessageFunc(func(key, value []byte) (bool, error) {
|
|
println(string(key), "=>", string(value))
|
|
return true, nil
|
|
})
|
|
|
|
subscriber.SetCompletionFunc(func() {
|
|
println("done subscribing")
|
|
})
|
|
|
|
if err := subscriber.Subscribe(); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
}
|