Change protoreflect.ProtoMessage to proto.Message

fuzzylabs · Oct 20, 2021 · d19fc8c · d19fc8c
1 parent 590c15f
commit d19fc8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pubsub.go b/pubsub.go
@@ -10,6 +10,7 @@ import (
 	"fmt"
 	log "github.com/sirupsen/logrus"
 	"google.golang.org/protobuf/encoding/protojson"
+	"google.golang.org/protobuf/proto"
 	"google.golang.org/protobuf/reflect/protoreflect"
 	"io"
 )
@@ -85,8 +86,7 @@ func NewPubSubDecoder() (IPubSub, error) {
 }
 
 // PublishMessage publishes the message to the Pub/Sub topic
-//TODO should use proto.Message instead
-func (p *PubSub) PublishMessage(topicID string, message protoreflect.ProtoMessage) error {
+func (p *PubSub) PublishMessage(topicID string, message proto.Message) error {
 	messageBytes, err := protojson.MarshalOptions{UseProtoNames: true}.Marshal(message)
 	if err != nil {
 		return err