mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-20 14:27:51 +08:00
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package sub_client
|
|
|
|
import (
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type SubscriberConfiguration struct {
|
|
ClientId string
|
|
GroupId string
|
|
GroupInstanceId string
|
|
BootstrapServers []string
|
|
GrpcDialOption grpc.DialOption
|
|
}
|
|
|
|
type ContentConfiguration struct {
|
|
Namespace string
|
|
Topic string
|
|
Filter string
|
|
}
|
|
|
|
type OnEachMessageFunc func(key, value []byte) (shouldContinue bool)
|
|
type OnCompletionFunc func()
|
|
|
|
type TopicSubscriber struct {
|
|
SubscriberConfig *SubscriberConfiguration
|
|
ContentConfig *ContentConfiguration
|
|
brokerPartitionAssignments []*mq_pb.BrokerPartitionAssignment
|
|
OnEachMessageFunc OnEachMessageFunc
|
|
OnCompletionFunc OnCompletionFunc
|
|
bootstrapBroker string
|
|
}
|
|
|
|
func NewTopicSubscriber(bootstrapBroker string, subscriber *SubscriberConfiguration, content *ContentConfiguration) *TopicSubscriber {
|
|
return &TopicSubscriber{
|
|
SubscriberConfig: subscriber,
|
|
ContentConfig: content,
|
|
bootstrapBroker: bootstrapBroker,
|
|
}
|
|
}
|
|
|
|
func (sub *TopicSubscriber) SetEachMessageFunc(onEachMessageFn OnEachMessageFunc) {
|
|
sub.OnEachMessageFunc = onEachMessageFn
|
|
}
|
|
|
|
func (sub *TopicSubscriber) SetCompletionFunc(onCompeletionFn OnCompletionFunc) {
|
|
sub.OnCompletionFunc = onCompeletionFn
|
|
}
|