diff options
| author |  Anshul Gupta <ansg191@anshulg.com>
| 2024-08-07 18:56:01 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2024-08-07 18:56:01 -0700 |
| commit | 08993e2f8497341079010d3d06361c99492c4c07 (patch) | |
| tree | c65d6d571c928410faace1fa51c2ea3f49fce003 /backend/internal/worker/analyzer | |
| parent | 3de4ebb7560851ccbefe296c197456fe80c22901 (diff) | |
| parent | b8aef1a7fb24815c7d93bc30c7b289b4f5896779 (diff) | |
| download | ibd-trader-08993e2f8497341079010d3d06361c99492c4c07.tar.gz ibd-trader-08993e2f8497341079010d3d06361c99492c4c07.tar.zst ibd-trader-08993e2f8497341079010d3d06361c99492c4c07.zip | |
Merge pull request #1 from ansg191/refactor-database
Diffstat (limited to 'backend/internal/worker/analyzer')
| -rw-r--r-- | backend/internal/worker/analyzer/analyzer.go | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/backend/internal/worker/analyzer/analyzer.go b/backend/internal/worker/analyzer/analyzer.go index 79a35ee..ea8069e 100644 --- a/backend/internal/worker/analyzer/analyzer.go +++ b/backend/internal/worker/analyzer/analyzer.go @@ -24,7 +24,7 @@ func RunAnalyzer( ctx context.Context, redis *redis.Client, analyzer analyzer.Analyzer, - db database.StockStore, + db database.Executor, name string, ) error { queue, err := taskqueue.New( @@ -52,7 +52,7 @@ func waitForTask( ctx context.Context, queue taskqueue.TaskQueue[TaskInfo], analyzer analyzer.Analyzer, - db database.StockStore, + db database.Executor, ) { task, err := queue.Dequeue(ctx, lockTimeout, dequeueTimeout) if err != nil { @@ -111,8 +111,8 @@ func waitForTask( } } -func analyzeStock(ctx context.Context, a analyzer.Analyzer, db database.StockStore, id string) error { - info, err := db.GetStockInfo(ctx, id) +func analyzeStock(ctx context.Context, a analyzer.Analyzer, db database.Executor, id string) error { + info, err := database.GetStockInfo(ctx, db, id) if err != nil { return err } @@ -127,7 +127,7 @@ func analyzeStock(ctx context.Context, a analyzer.Analyzer, db database.StockSto return err } - return db.AddAnalysis(ctx, id, analysis) + return database.AddAnalysis(ctx, db, id, analysis) } type TaskInfo struct { |