Merge pull request #55 from h0n9/feature/enable-gzip-compression

Support Gzip compression
h0n9 · Jul 8, 2024 · 0096b35 · 0096b35
2 parents e3504fd + 2e143e6
commit 0096b35
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/cli/agent/agent.go b/cli/agent/agent.go
@@ -14,6 +14,7 @@ import (
 	"github.com/rs/zerolog"
 	"github.com/spf13/cobra"
 	"google.golang.org/grpc"
+	_ "google.golang.org/grpc/encoding/gzip"
 
 	"github.com/h0n9/msg-lake/lake"
 	pb "github.com/h0n9/msg-lake/proto"

diff --git a/client/client.go b/client/client.go
@@ -9,6 +9,7 @@ import (
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/encoding/gzip"
 
 	pb "github.com/h0n9/msg-lake/proto"
 	"github.com/postie-labs/go-postie-lib/crypto"
@@ -139,16 +140,20 @@ func (c *Client) Publish(ctx context.Context, topicID, message string) error {
 	}
 
 	// publish the message
-	pubRes, err := c.msgLakeClient.Publish(ctx, &pb.PublishReq{
-		TopicId: topicID,
-		MsgCapsule: &pb.MsgCapsule{
-			Data: data,
-			Signature: &pb.Signature{
-				PubKey: c.privKey.PubKey().Bytes(),
-				Data:   sigDataBytes,
+	pubRes, err := c.msgLakeClient.Publish(
+		ctx,
+		&pb.PublishReq{
+			TopicId: topicID,
+			MsgCapsule: &pb.MsgCapsule{
+				Data: data,
+				Signature: &pb.Signature{
+					PubKey: c.privKey.PubKey().Bytes(),
+					Data:   sigDataBytes,
+				},
 			},
 		},
-	})
+		grpc.UseCompressor(gzip.Name),
+	)
 	if err != nil {
 		return err
 	}