Add context for everything

Signed-off-by: Asai Neko <sugar@sne.moe>

data/user.go

@@ -1,6 +1,8 @@
 package data
 
 import (
+	"context"
+
 	"github.com/go-viper/mapstructure/v2"
 	"github.com/google/uuid"
 	"github.com/meilisearch/meilisearch-go"
@@ -31,10 +33,10 @@ type UserSearchDoc struct {
 	Avatar   string `json:"avatar"`
 }
 
-func (self *User) GetByEmail(email string) (*User, error) {
+func (self *User) GetByEmail(ctx context.Context, email string) (*User, error) {
 	var user User
 
-	err := Database.
+	err := Database.WithContext(ctx).
 		Where("email = ?", email).
 		First(&user).Error
 
@@ -48,10 +50,10 @@ func (self *User) GetByEmail(email string) (*User, error) {
 	return &user, nil
 }
 
-func (self *User) GetByUserId(userId uuid.UUID) (*User, error) {
+func (self *User) GetByUserId(ctx context.Context, userId uuid.UUID) (*User, error) {
 	var user User
 
-	err := Database.
+	err := Database.WithContext(ctx).
 		Where("user_id = ?", userId).
 		First(&user).Error
 
@@ -65,12 +67,12 @@ func (self *User) GetByUserId(userId uuid.UUID) (*User, error) {
 	return &user, err
 }
 
-func (self *User) Create() error {
+func (self *User) Create(ctx context.Context) error {
 	self.UUID = uuid.New()
 	self.UserId = uuid.New()
 
 	// DB transaction only
-	if err := Database.Transaction(func(tx *gorm.DB) error {
+	if err := Database.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
 		if err := tx.Create(self).Error; err != nil {
 			return err
 		}
@@ -80,7 +82,7 @@ func (self *User) Create() error {
 	}
 
 	// Search index (eventual consistency)
-	if err := self.UpdateSearchIndex(); err != nil {
+	if err := self.UpdateSearchIndex(ctx); err != nil {
 		// TODO: async retry / log
 		return err
 	}
@@ -88,8 +90,8 @@ func (self *User) Create() error {
 	return nil
 }
 
-func (self *User) UpdateByUserID(userId uuid.UUID) error {
-	return Database.Transaction(func(tx *gorm.DB) error {
+func (self *User) UpdateByUserID(ctx context.Context, userId uuid.UUID) error {
+	return Database.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
 		if err := tx.Model(&User{}).Where("user_id = ?", userId).Updates(&self).Error; err != nil {
 			return err
 		}
@@ -97,20 +99,20 @@ func (self *User) UpdateByUserID(userId uuid.UUID) error {
 	})
 }
 
-func (self *User) GetFullTable() (*[]User, error) {
+func (self *User) GetFullTable(ctx context.Context) (*[]User, error) {
 	var users []User
-	err := Database.Find(&users).Error
+	err := Database.WithContext(ctx).Find(&users).Error
 	if err != nil {
 		return nil, err
 	}
 	return &users, nil
 }
 
-func (self *User) FastListUsers(limit, offset int64) (*[]UserSearchDoc, error) {
+func (self *User) FastListUsers(ctx context.Context, limit, offset int64) (*[]UserSearchDoc, error) {
 	index := MeiliSearch.Index("user")
 
 	// Fast read from MeiliSearch, no DB involved
-	result, err := index.Search("", &meilisearch.SearchRequest{
+	result, err := index.SearchWithContext(ctx, "", &meilisearch.SearchRequest{
 		Limit:  limit,
 		Offset: offset,
 	})
@@ -126,7 +128,7 @@ func (self *User) FastListUsers(limit, offset int64) (*[]UserSearchDoc, error) {
 	return &list, nil
 }
 
-func (self *User) UpdateSearchIndex() error {
+func (self *User) UpdateSearchIndex(ctx context.Context) error {
 	doc := UserSearchDoc{
 		UserId:   self.UserId.String(),
 		Email:    self.Email,
@@ -142,7 +144,8 @@ func (self *User) UpdateSearchIndex() error {
 		PrimaryKey: &primaryKey,
 	}
 
-	if _, err := index.UpdateDocuments(
+	if _, err := index.UpdateDocumentsWithContext(
+		ctx,
 		[]UserSearchDoc{doc},
 		opts,
 	); err != nil {
@@ -152,8 +155,8 @@ func (self *User) UpdateSearchIndex() error {
 	return nil
 }
 
-func (self *User) DeleteSearchIndex() error {
+func (self *User) DeleteSearchIndex(ctx context.Context) error {
 	index := MeiliSearch.Index("user")
-	_, err := index.DeleteDocument(self.UserId.String(), nil)
+	_, err := index.DeleteDocumentWithContext(ctx, self.UserId.String(), nil)
 	return err
 }