Implement RDB.DeleteAggregationSet

internal/rdb/rdb.go

@@ -1124,9 +1124,22 @@ func (r *RDB) ReadAggregationSet(qname, gname, setID string) ([]*base.TaskMessag
 	return msgs, time.Unix(int64(deadlineUnix), 0), nil
 }
 
-// DeleteAggregationSet deletes the aggregation set identified by the parameters.
+// KEYS[1] -> asynq:{<qname>}:g:<gname>:<aggregation_set_id>
+// -------
+// ARGV[1] -> task key prefix
+var deleteAggregationSetCmd = redis.NewScript(`
+local ids = redis.call("SMEMBERS", KEYS[1])
+for _, id in ipairs(ids)  do
+	redis.call("DEL", ARGV[1] .. id)
+end
+redis.call("DEL", KEYS[1])
+return redis.status_reply("OK")
+`)
+
+// DeleteAggregationSet deletes the aggregation set and its members identified by the parameters.
 func (r *RDB) DeleteAggregationSet(ctx context.Context, qname, gname, setID string) error {
-	return nil
+	var op errors.Op = "RDB.DeleteAggregationSet"
+	return r.runScript(ctx, op, deleteAggregationSetCmd, []string{base.AggregationSetKey(qname, gname, setID)}, base.TaskKeyPrefix(qname))
 }
 
 // KEYS[1] -> asynq:{<qname>}:completed

internal/rdb/rdb_test.go

@@ -3284,3 +3284,50 @@ func TestAggregationCheck(t *testing.T) {
 		}
 	}
 }
+
+func TestDeleteAggregationSet(t *testing.T) {
+	r := setup(t)
+	defer r.Close()
+
+	ctx := context.Background()
+	setID := uuid.NewString()
+	msg1 := h.NewTaskMessageBuilder().SetType("foo").SetQueue("default").SetGroup("mygroup").Build()
+	msg2 := h.NewTaskMessageBuilder().SetType("bar").SetQueue("default").SetGroup("mygroup").Build()
+	msg3 := h.NewTaskMessageBuilder().SetType("baz").SetQueue("default").SetGroup("mygroup").Build()
+
+	tests := []struct {
+		aggregationSet []*base.TaskMessage
+		qname          string
+		gname          string
+		setID          string
+	}{
+		{
+			aggregationSet: []*base.TaskMessage{msg1, msg2, msg3},
+			qname:          "default",
+			gname:          "mygroup",
+			setID:          setID,
+		},
+	}
+
+	for _, tc := range tests {
+		h.FlushDB(t, r.client)
+		h.SeedAggregationSet(t, r.client, tc.aggregationSet, tc.qname, tc.gname, tc.setID)
+
+		if err := r.DeleteAggregationSet(ctx, tc.qname, tc.gname, tc.setID); err != nil {
+			t.Fatalf("DeleteAggregationSet returned error: %v", err)
+		}
+		key := base.AggregationSetKey(tc.qname, tc.gname, tc.setID)
+		// Check if the set is deleted.
+		if r.client.Exists(ctx, key).Val() != 0 {
+			t.Errorf("aggregation set key %q still exists", key)
+		}
+
+		// Check all tasks in the set are deleted.
+		for _, m := range tc.aggregationSet {
+			taskKey := base.TaskKey(m.Queue, m.ID)
+			if r.client.Exists(ctx, taskKey).Val() != 0 {
+				t.Errorf("task key %q still exists", taskKey)
+			}
+		}
+	}
+}