You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.5 KiB
61 lines
1.5 KiB
package sub
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
"github.com/golang/protobuf/proto"
|
|
"gocloud.dev/pubsub"
|
|
_ "gocloud.dev/pubsub/awssnssqs"
|
|
// _ "gocloud.dev/pubsub/azuresb"
|
|
_ "gocloud.dev/pubsub/gcppubsub"
|
|
_ "gocloud.dev/pubsub/natspubsub"
|
|
_ "gocloud.dev/pubsub/rabbitpubsub"
|
|
)
|
|
|
|
func init() {
|
|
NotificationInputs = append(NotificationInputs, &GoCDKPubSubInput{})
|
|
}
|
|
|
|
type GoCDKPubSubInput struct {
|
|
sub *pubsub.Subscription
|
|
}
|
|
|
|
func (k *GoCDKPubSubInput) GetName() string {
|
|
return "gocdk_pub_sub"
|
|
}
|
|
|
|
func (k *GoCDKPubSubInput) Initialize(configuration util.Configuration, prefix string) error {
|
|
subURL := configuration.GetString(prefix + "sub_url")
|
|
glog.V(0).Infof("notification.gocdk_pub_sub.sub_url: %v", subURL)
|
|
sub, err := pubsub.OpenSubscription(context.Background(), subURL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
k.sub = sub
|
|
return nil
|
|
}
|
|
|
|
func (k *GoCDKPubSubInput) ReceiveMessage() (key string, message *filer_pb.EventNotification, onSuccessFn func(), onFailureFn func(), err error) {
|
|
msg, err := k.sub.Receive(context.Background())
|
|
if err != nil {
|
|
return
|
|
}
|
|
onSuccessFn = func() {
|
|
msg.Ack()
|
|
}
|
|
onFailureFn = func() {
|
|
if msg.Nackable() {
|
|
msg.Nack()
|
|
}
|
|
}
|
|
key = msg.Metadata["key"]
|
|
message = &filer_pb.EventNotification{}
|
|
err = proto.Unmarshal(msg.Body, message)
|
|
if err != nil {
|
|
return "", nil, onSuccessFn, onFailureFn, err
|
|
}
|
|
return key, message, onSuccessFn, onFailureFn, nil
|
|
}
|