From 73107ab51c3fcb1d7db472ae79c561799e69c94d Mon Sep 17 00:00:00 2001
From: Naman Khator <naman@maplegraph.com>
Date: Tue, 31 Dec 2024 12:35:28 +0530
Subject: [PATCH] Handle dynamic table columns, names or joins

---
 internal/core/core.go | 27 +++++++++++++++++++++------
 models/models.go      |  3 +++
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/internal/core/core.go b/internal/core/core.go
index de0423c..623bf35 100644
--- a/internal/core/core.go
+++ b/internal/core/core.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"log/slog"
 	"strconv"
+	"strings"
 	"sync"
 	"time"
 
@@ -377,10 +378,17 @@ func (co *Core) makeJob(j models.JobReq, taskName string) (tasqueue.Job, error)
 		args[i] = j.Args[i]
 	}
 
+	// DynamicParams
+	var dynamicParams = make([]interface{}, len(j.DynamicParams))
+	for i := range j.DynamicParams {
+		dynamicParams[i] = j.DynamicParams[i]
+	}
+
 	b, err := msgpack.Marshal(taskMeta{
-		Args: args,
-		DB:   j.DB,
-		TTL:  int(ttl),
+		Args:          args,
+		DB:            j.DB,
+		TTL:           int(ttl),
+		DynamicParams: dynamicParams,
 	})
 	if err != nil {
 		return tasqueue.Job{}, err
@@ -395,9 +403,10 @@ func (co *Core) makeJob(j models.JobReq, taskName string) (tasqueue.Job, error)
 }
 
 type taskMeta struct {
-	Args []interface{} `json:"args"`
-	DB   string        `json:"db"`
-	TTL  int           `json:"ttl"`
+	Args          []interface{} `json:"args"`
+	DB            string        `json:"db"`
+	TTL           int           `json:"ttl"`
+	DynamicParams []interface{} `json:"dynamic_params"`
 }
 
 // initQueue creates and returns a distributed queue system (Tasqueue) and registers
@@ -428,6 +437,12 @@ func (co *Core) initQueue() (*tasqueue.Server, error) {
 				return fmt.Errorf("could not unmarshal args : %w", err)
 			}
 
+			// replace dynamic params in query (args.DynamicParams)
+			// example dynamic columns, dynamic joins, dynamic table names
+			for i := 0; i < len(args.DynamicParams); i++ {
+				query.Raw = strings.Replace(query.Raw, "$"+strconv.Itoa(i+1), fmt.Sprintf("%v", args.DynamicParams[i]), -1)
+			}
+
 			count, err := co.execJob(jctx.Meta.ID, name, args.DB, time.Duration(args.TTL)*time.Second, args.Args, query)
 			if err != nil {
 				return fmt.Errorf("could not execute job : %w", err)
diff --git a/models/models.go b/models/models.go
index 2d8bfcd..33fc94e 100644
--- a/models/models.go
+++ b/models/models.go
@@ -15,6 +15,9 @@ type JobReq struct {
 	Args     []string `json:"args"`
 	DB       string   `json:"db"`
 
+	// Optional params
+	DynamicParams []string `json:"dynamic_params"`
+
 	ttlDuration time.Duration
 }