From 4e8885276c2edf039681e9b4d9dffc609f8f95f7 Mon Sep 17 00:00:00 2001
From: Ken Hibino <ken.hibino7@gmail.com>
Date: Sun, 6 Mar 2022 05:53:59 -0800
Subject: [PATCH] Update client to store groupKey under TaskMessage

---
 client.go      | 1 +
 client_test.go | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/client.go b/client.go
index 34aafd5..248b452 100644
--- a/client.go
+++ b/client.go
@@ -375,6 +375,7 @@ func (c *Client) EnqueueContext(ctx context.Context, task *Task, opts ...Option)
 		Deadline:  deadline.Unix(),
 		Timeout:   int64(timeout.Seconds()),
 		UniqueKey: uniqueKey,
+		GroupKey:  opt.groupKey,
 		Retention: int64(opt.retention.Seconds()),
 	}
 	now := time.Now()
diff --git a/client_test.go b/client_test.go
index 06cfec5..d305d76 100644
--- a/client_test.go
+++ b/client_test.go
@@ -528,6 +528,7 @@ func TestClientEnqueueWithGroupOption(t *testing.T) {
 								Queue:    "default",
 								Timeout:  int64(defaultTimeout.Seconds()),
 								Deadline: noDeadline.Unix(),
+								GroupKey: "mygroup",
 							},
 							Score: now.Unix(),
 						},
@@ -576,6 +577,7 @@ func TestClientEnqueueWithGroupOption(t *testing.T) {
 							Queue:    "default",
 							Timeout:  int64(defaultTimeout.Seconds()),
 							Deadline: noDeadline.Unix(),
+							GroupKey: "mygroup",
 						},
 						Score: now.Add(30 * time.Minute).Unix(),
 					},