From 1163655f49826ee12a42acc0a238a6c2ae4bf987 Mon Sep 17 00:00:00 2001 From: kolaente Date: Tue, 16 Jan 2024 12:47:53 +0100 Subject: [PATCH] fix: update function signatures --- pkg/models/listeners.go | 5 +++-- pkg/models/task_search.go | 3 ++- pkg/models/typesense.go | 11 ++++++----- 3 files changed, 11 insertions(+), 8 deletions(-) diff --git a/pkg/models/listeners.go b/pkg/models/listeners.go index 67c64873d..445cf44ff 100644 --- a/pkg/models/listeners.go +++ b/pkg/models/listeners.go @@ -17,6 +17,7 @@ package models import ( + "context" "encoding/json" "strconv" "time" @@ -526,7 +527,7 @@ func (s *RemoveTaskFromTypesense) Handle(msg *message.Message) (err error) { _, err = typesenseClient. Collection("tasks"). Document(strconv.FormatInt(event.Task.ID, 10)). - Delete() + Delete(context.Background()) return err } @@ -558,7 +559,7 @@ func (l *AddTaskToTypesense) Handle(msg *message.Message) (err error) { _, err = typesenseClient.Collection("tasks"). Documents(). - Create(ttask) + Create(context.Background(), ttask) return } diff --git a/pkg/models/task_search.go b/pkg/models/task_search.go index 1fd6c00a9..a8071b273 100644 --- a/pkg/models/task_search.go +++ b/pkg/models/task_search.go @@ -17,6 +17,7 @@ package models import ( + "context" "strconv" "strings" "time" @@ -411,7 +412,7 @@ func (t *typesenseTaskSearcher) Search(opts *taskSearchOptions) (tasks []*Task, result, err := typesenseClient.Collection("tasks"). Documents(). - Search(params) + Search(context.Background(), params) if err != nil { return } diff --git a/pkg/models/typesense.go b/pkg/models/typesense.go index f7a4eea64..ee152861e 100644 --- a/pkg/models/typesense.go +++ b/pkg/models/typesense.go @@ -17,6 +17,7 @@ package models import ( + "context" "fmt" "time" @@ -195,9 +196,9 @@ func CreateTypesenseCollections() error { } // delete any collection which might exist - _, _ = typesenseClient.Collection("tasks").Delete() + _, _ = typesenseClient.Collection("tasks").Delete(context.Background()) - _, err := typesenseClient.Collections().Create(taskSchema) + _, err := typesenseClient.Collections().Create(context.Background(), taskSchema) return err } @@ -304,7 +305,7 @@ func reindexTasks(s *xorm.Session, tasks map[int64]*Task) (err error) { _, err = typesenseClient.Collection("tasks"). Documents(). - Import(typesenseTasks, &api.ImportDocumentsParams{ + Import(context.Background(), typesenseTasks, &api.ImportDocumentsParams{ Action: pointer.String("upsert"), BatchSize: pointer.Int(100), }) @@ -381,14 +382,14 @@ func indexDummyTask() (err error) { _, err = typesenseClient.Collection("tasks"). Documents(). - Create(dummyTask) + Create(context.Background(), dummyTask) if err != nil { return } _, err = typesenseClient.Collection("tasks"). Document(dummyTask.ID). - Delete() + Delete(context.Background()) return }