From 2dc9fc1626507bb54417fc4a1920b847cafb27a2 Mon Sep 17 00:00:00 2001
From: tobi <31960611+tsmethurst@users.noreply.github.com>
Date: Wed, 25 Aug 2021 15:34:33 +0200
Subject: Pg to bun (#148)

* start moving to bun

* changing more stuff

* more

* and yet more

* tests passing

* seems stable now

* more big changes

* small fix

* little fixes
---
 internal/db/status.go | 36 ++++++++++++++++++++----------------
 1 file changed, 20 insertions(+), 16 deletions(-)

(limited to 'internal/db/status.go')

diff --git a/internal/db/status.go b/internal/db/status.go
index 9d206c198..7430433c4 100644
--- a/internal/db/status.go
+++ b/internal/db/status.go
@@ -18,58 +18,62 @@
 
 package db
 
-import "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+import (
+	"context"
+
+	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+)
 
 // Status contains functions for getting statuses, creating statuses, and checking various other fields on statuses.
 type Status interface {
 	// GetStatusByID returns one status from the database, with all rel fields populated (if possible).
-	GetStatusByID(id string) (*gtsmodel.Status, Error)
+	GetStatusByID(ctx context.Context, id string) (*gtsmodel.Status, Error)
 
 	// GetStatusByURI returns one status from the database, with all rel fields populated (if possible).
-	GetStatusByURI(uri string) (*gtsmodel.Status, Error)
+	GetStatusByURI(ctx context.Context, uri string) (*gtsmodel.Status, Error)
 
 	// GetStatusByURL returns one status from the database, with all rel fields populated (if possible).
-	GetStatusByURL(uri string) (*gtsmodel.Status, Error)
+	GetStatusByURL(ctx context.Context, uri string) (*gtsmodel.Status, Error)
 
 	// PutStatus stores one status in the database.
-	PutStatus(status *gtsmodel.Status) Error
+	PutStatus(ctx context.Context, status *gtsmodel.Status) Error
 
 	// CountStatusReplies returns the amount of replies recorded for a status, or an error if something goes wrong
-	CountStatusReplies(status *gtsmodel.Status) (int, Error)
+	CountStatusReplies(ctx context.Context, status *gtsmodel.Status) (int, Error)
 
 	// CountStatusReblogs returns the amount of reblogs/boosts recorded for a status, or an error if something goes wrong
-	CountStatusReblogs(status *gtsmodel.Status) (int, Error)
+	CountStatusReblogs(ctx context.Context, status *gtsmodel.Status) (int, Error)
 
 	// CountStatusFaves returns the amount of faves/likes recorded for a status, or an error if something goes wrong
-	CountStatusFaves(status *gtsmodel.Status) (int, Error)
+	CountStatusFaves(ctx context.Context, status *gtsmodel.Status) (int, Error)
 
 	// GetStatusParents gets the parent statuses of a given status.
 	//
 	// If onlyDirect is true, only the immediate parent will be returned.
-	GetStatusParents(status *gtsmodel.Status, onlyDirect bool) ([]*gtsmodel.Status, Error)
+	GetStatusParents(ctx context.Context, status *gtsmodel.Status, onlyDirect bool) ([]*gtsmodel.Status, Error)
 
 	// GetStatusChildren gets the child statuses of a given status.
 	//
 	// If onlyDirect is true, only the immediate children will be returned.
-	GetStatusChildren(status *gtsmodel.Status, onlyDirect bool, minID string) ([]*gtsmodel.Status, Error)
+	GetStatusChildren(ctx context.Context, status *gtsmodel.Status, onlyDirect bool, minID string) ([]*gtsmodel.Status, Error)
 
 	// IsStatusFavedBy checks if a given status has been faved by a given account ID
-	IsStatusFavedBy(status *gtsmodel.Status, accountID string) (bool, Error)
+	IsStatusFavedBy(ctx context.Context, status *gtsmodel.Status, accountID string) (bool, Error)
 
 	// IsStatusRebloggedBy checks if a given status has been reblogged/boosted by a given account ID
-	IsStatusRebloggedBy(status *gtsmodel.Status, accountID string) (bool, Error)
+	IsStatusRebloggedBy(ctx context.Context, status *gtsmodel.Status, accountID string) (bool, Error)
 
 	// IsStatusMutedBy checks if a given status has been muted by a given account ID
-	IsStatusMutedBy(status *gtsmodel.Status, accountID string) (bool, Error)
+	IsStatusMutedBy(ctx context.Context, status *gtsmodel.Status, accountID string) (bool, Error)
 
 	// IsStatusBookmarkedBy checks if a given status has been bookmarked by a given account ID
-	IsStatusBookmarkedBy(status *gtsmodel.Status, accountID string) (bool, Error)
+	IsStatusBookmarkedBy(ctx context.Context, status *gtsmodel.Status, accountID string) (bool, Error)
 
 	// GetStatusFaves returns a slice of faves/likes of the given status.
 	// This slice will be unfiltered, not taking account of blocks and whatnot, so filter it before serving it back to a user.
-	GetStatusFaves(status *gtsmodel.Status) ([]*gtsmodel.StatusFave, Error)
+	GetStatusFaves(ctx context.Context, status *gtsmodel.Status) ([]*gtsmodel.StatusFave, Error)
 
 	// GetStatusReblogs returns a slice of statuses that are a boost/reblog of the given status.
 	// This slice will be unfiltered, not taking account of blocks and whatnot, so filter it before serving it back to a user.
-	GetStatusReblogs(status *gtsmodel.Status) ([]*gtsmodel.Status, Error)
+	GetStatusReblogs(ctx context.Context, status *gtsmodel.Status) ([]*gtsmodel.Status, Error)
 }
-- 
cgit v1.2.3