Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/inputs/cloud_pubsub/subscription_gcp.go

@@ -0,0 +1,85 @@
+package cloud_pubsub
+
+import (
+	"context"
+	"time"
+
+	"cloud.google.com/go/pubsub"
+)
+
+type (
+	subscription interface {
+		// ID returns the unique identifier of the subscription.
+		ID() string
+		// Receive starts receiving messages from the subscription and processes them using the provided function.
+		Receive(ctx context.Context, f func(context.Context, message)) error
+	}
+
+	message interface {
+		// Ack acknowledges the message, indicating successful processing.
+		Ack()
+		// Nack negatively acknowledges the message, indicating it should be redelivered.
+		Nack()
+		// ID returns the unique identifier of the message.
+		ID() string
+		// Data returns the payload of the message.
+		Data() []byte
+		// Attributes returns the attributes of the message as a key-value map.
+		Attributes() map[string]string
+		// PublishTime returns the time when the message was published.
+		PublishTime() time.Time
+	}
+
+	gcpSubscription struct {
+		sub *pubsub.Subscription
+	}
+
+	gcpMessage struct {
+		msg *pubsub.Message
+	}
+)
+
+// ID returns the unique identifier of the subscription.
+func (s *gcpSubscription) ID() string {
+	if s.sub == nil {
+		return ""
+	}
+	return s.sub.ID()
+}
+
+// Receive starts receiving messages from the subscription and processes them using the provided function.
+func (s *gcpSubscription) Receive(ctx context.Context, f func(context.Context, message)) error {
+	return s.sub.Receive(ctx, func(cctx context.Context, m *pubsub.Message) {
+		f(cctx, &gcpMessage{m})
+	})
+}
+
+// Ack acknowledges the message, indicating successful processing.
+func (env *gcpMessage) Ack() {
+	env.msg.Ack()
+}
+
+// Nack negatively acknowledges the message, indicating it should be redelivered.
+func (env *gcpMessage) Nack() {
+	env.msg.Nack()
+}
+
+// ID returns the unique identifier of the message.
+func (env *gcpMessage) ID() string {
+	return env.msg.ID
+}
+
+// Data returns the payload of the message.
+func (env *gcpMessage) Data() []byte {
+	return env.msg.Data
+}
+
+// Attributes returns the attributes of the message as a key-value map.
+func (env *gcpMessage) Attributes() map[string]string {
+	return env.msg.Attributes
+}
+
+// PublishTime returns the time when the message was published.
+func (env *gcpMessage) PublishTime() time.Time {
+	return env.msg.PublishTime
+}