summaryrefslogtreecommitdiff
path: root/internal/processing/workers
diff options
context:
space:
mode:
authorLibravatar tobi <31960611+tsmethurst@users.noreply.github.com>2024-04-16 13:10:13 +0200
committerLibravatar GitHub <noreply@github.com>2024-04-16 13:10:13 +0200
commit3cceed11b28b5f42a653d85ed779d652fd8c26ad (patch)
tree0a7f0994e477609ca705a45f382dfb62056b196e /internal/processing/workers
parent[performance] cached oauth database types (#2838) (diff)
downloadgotosocial-3cceed11b28b5f42a653d85ed779d652fd8c26ad.tar.xz
[feature/performance] Store account stats in separate table (#2831)
* [feature/performance] Store account stats in separate table, get stats from remote * test account stats * add some missing increment / decrement calls * change stats function signatures * rejig logging a bit * use lock when updating stats
Diffstat (limited to 'internal/processing/workers')
-rw-r--r--internal/processing/workers/fromclientapi.go54
-rw-r--r--internal/processing/workers/fromclientapi_test.go66
-rw-r--r--internal/processing/workers/fromfediapi.go64
-rw-r--r--internal/processing/workers/fromfediapi_test.go74
-rw-r--r--internal/processing/workers/util.go255
5 files changed, 447 insertions, 66 deletions
diff --git a/internal/processing/workers/fromclientapi.go b/internal/processing/workers/fromclientapi.go
index 37c330cf0..1412ea003 100644
--- a/internal/processing/workers/fromclientapi.go
+++ b/internal/processing/workers/fromclientapi.go
@@ -247,6 +247,11 @@ func (p *clientAPI) CreateStatus(ctx context.Context, cMsg messages.FromClientAP
return gtserror.Newf("%T not parseable as *gtsmodel.Status", cMsg.GTSModel)
}
+ // Update stats for the actor account.
+ if err := p.utilF.incrementStatusesCount(ctx, cMsg.OriginAccount, status); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.surface.timelineAndNotifyStatus(ctx, status); err != nil {
log.Errorf(ctx, "error timelining and notifying status: %v", err)
}
@@ -311,6 +316,11 @@ func (p *clientAPI) CreateFollowReq(ctx context.Context, cMsg messages.FromClien
return gtserror.Newf("%T not parseable as *gtsmodel.FollowRequest", cMsg.GTSModel)
}
+ // Update stats for the target account.
+ if err := p.utilF.incrementFollowRequestsCount(ctx, cMsg.TargetAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.surface.notifyFollowRequest(ctx, followRequest); err != nil {
log.Errorf(ctx, "error notifying follow request: %v", err)
}
@@ -360,6 +370,11 @@ func (p *clientAPI) CreateAnnounce(ctx context.Context, cMsg messages.FromClient
return gtserror.Newf("%T not parseable as *gtsmodel.Status", cMsg.GTSModel)
}
+ // Update stats for the actor account.
+ if err := p.utilF.incrementStatusesCount(ctx, cMsg.OriginAccount, boost); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
// Timeline and notify the boost wrapper status.
if err := p.surface.timelineAndNotifyStatus(ctx, boost); err != nil {
log.Errorf(ctx, "error timelining and notifying status: %v", err)
@@ -485,6 +500,20 @@ func (p *clientAPI) AcceptFollow(ctx context.Context, cMsg messages.FromClientAP
return gtserror.Newf("%T not parseable as *gtsmodel.Follow", cMsg.GTSModel)
}
+ // Update stats for the target account.
+ if err := p.utilF.decrementFollowRequestsCount(ctx, cMsg.TargetAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ if err := p.utilF.incrementFollowersCount(ctx, cMsg.TargetAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ // Update stats for the origin account.
+ if err := p.utilF.incrementFollowingCount(ctx, cMsg.OriginAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.surface.notifyFollow(ctx, follow); err != nil {
log.Errorf(ctx, "error notifying follow: %v", err)
}
@@ -502,6 +531,11 @@ func (p *clientAPI) RejectFollowRequest(ctx context.Context, cMsg messages.FromC
return gtserror.Newf("%T not parseable as *gtsmodel.FollowRequest", cMsg.GTSModel)
}
+ // Update stats for the target account.
+ if err := p.utilF.decrementFollowRequestsCount(ctx, cMsg.TargetAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.federate.RejectFollow(
ctx,
p.converter.FollowRequestToFollow(ctx, followReq),
@@ -518,6 +552,16 @@ func (p *clientAPI) UndoFollow(ctx context.Context, cMsg messages.FromClientAPI)
return gtserror.Newf("%T not parseable as *gtsmodel.Follow", cMsg.GTSModel)
}
+ // Update stats for the origin account.
+ if err := p.utilF.decrementFollowingCount(ctx, cMsg.OriginAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ // Update stats for the target account.
+ if err := p.utilF.decrementFollowersCount(ctx, cMsg.TargetAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.federate.UndoFollow(ctx, follow); err != nil {
log.Errorf(ctx, "error federating follow undo: %v", err)
}
@@ -565,6 +609,11 @@ func (p *clientAPI) UndoAnnounce(ctx context.Context, cMsg messages.FromClientAP
return gtserror.Newf("db error deleting status: %w", err)
}
+ // Update stats for the origin account.
+ if err := p.utilF.decrementStatusesCount(ctx, cMsg.OriginAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.surface.deleteStatusFromTimelines(ctx, status.ID); err != nil {
log.Errorf(ctx, "error removing timelined status: %v", err)
}
@@ -603,6 +652,11 @@ func (p *clientAPI) DeleteStatus(ctx context.Context, cMsg messages.FromClientAP
log.Errorf(ctx, "error wiping status: %v", err)
}
+ // Update stats for the origin account.
+ if err := p.utilF.decrementStatusesCount(ctx, cMsg.OriginAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if status.InReplyToID != "" {
// Interaction counts changed on the replied status;
// uncache the prepared version from all timelines.
diff --git a/internal/processing/workers/fromclientapi_test.go b/internal/processing/workers/fromclientapi_test.go
index 3d3630b11..5e294597d 100644
--- a/internal/processing/workers/fromclientapi_test.go
+++ b/internal/processing/workers/fromclientapi_test.go
@@ -182,11 +182,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusWithNotification() {
nil,
nil,
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Update the follow from receiving account -> posting account so
@@ -212,6 +207,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusWithNotification() {
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
@@ -285,11 +286,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusReply() {
suite.testStatuses["local_account_2_status_1"],
nil,
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Process the new status.
@@ -305,6 +301,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusReply() {
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
@@ -451,11 +453,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusListRepliesPolicyLis
suite.testStatuses["local_account_2_status_1"],
nil,
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Modify replies policy of test list to show replies
@@ -480,6 +477,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusListRepliesPolicyLis
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
@@ -518,11 +521,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusListRepliesPolicyLis
suite.testStatuses["local_account_2_status_1"],
nil,
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Modify replies policy of test list to show replies
@@ -552,6 +550,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusListRepliesPolicyLis
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
@@ -590,11 +594,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusReplyListRepliesPoli
suite.testStatuses["local_account_2_status_1"],
nil,
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Modify replies policy of test list.
@@ -619,6 +618,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusReplyListRepliesPoli
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
@@ -654,11 +659,6 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusBoost() {
nil,
suite.testStatuses["local_account_2_status_1"],
)
- statusJSON = suite.statusJSON(
- ctx,
- status,
- receivingAccount,
- )
)
// Process the new status.
@@ -674,6 +674,12 @@ func (suite *FromClientAPITestSuite) TestProcessCreateStatusBoost() {
suite.FailNow(err.Error())
}
+ statusJSON := suite.statusJSON(
+ ctx,
+ status,
+ receivingAccount,
+ )
+
// Check message in home stream.
suite.checkStreamed(
homeStream,
diff --git a/internal/processing/workers/fromfediapi.go b/internal/processing/workers/fromfediapi.go
index 7b0e72490..0b1106a9e 100644
--- a/internal/processing/workers/fromfediapi.go
+++ b/internal/processing/workers/fromfediapi.go
@@ -122,7 +122,7 @@ func (p *Processor) ProcessFromFediAPI(ctx context.Context, fMsg messages.FromFe
// UPDATE SOMETHING
case ap.ActivityUpdate:
- switch fMsg.APObjectType { //nolint:gocritic
+ switch fMsg.APObjectType {
// UPDATE NOTE/STATUS
case ap.ObjectNote:
@@ -133,6 +133,15 @@ func (p *Processor) ProcessFromFediAPI(ctx context.Context, fMsg messages.FromFe
return p.fediAPI.UpdateAccount(ctx, fMsg)
}
+ // ACCEPT SOMETHING
+ case ap.ActivityAccept:
+ switch fMsg.APObjectType { //nolint:gocritic
+
+ // ACCEPT FOLLOW
+ case ap.ActivityFollow:
+ return p.fediAPI.AcceptFollow(ctx, fMsg)
+ }
+
// DELETE SOMETHING
case ap.ActivityDelete:
switch fMsg.APObjectType {
@@ -220,6 +229,11 @@ func (p *fediAPI) CreateStatus(ctx context.Context, fMsg messages.FromFediAPI) e
return nil
}
+ // Update stats for the remote account.
+ if err := p.utilF.incrementStatusesCount(ctx, fMsg.RequestingAccount, status); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if status.InReplyToID != "" {
// Interaction counts changed on the replied status; uncache the
// prepared version from all timelines. The status dereferencer
@@ -290,14 +304,20 @@ func (p *fediAPI) CreateFollowReq(ctx context.Context, fMsg messages.FromFediAPI
}
if *followRequest.TargetAccount.Locked {
- // Account on our instance is locked: just notify the follow request.
+ // Local account is locked: just notify the follow request.
if err := p.surface.notifyFollowRequest(ctx, followRequest); err != nil {
log.Errorf(ctx, "error notifying follow request: %v", err)
}
+
+ // And update stats for the local account.
+ if err := p.utilF.incrementFollowRequestsCount(ctx, fMsg.ReceivingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
return nil
}
- // Account on our instance is not locked:
+ // Local account is not locked:
// Automatically accept the follow request
// and notify about the new follower.
follow, err := p.state.DB.AcceptFollowRequest(
@@ -309,6 +329,16 @@ func (p *fediAPI) CreateFollowReq(ctx context.Context, fMsg messages.FromFediAPI
return gtserror.Newf("error accepting follow request: %w", err)
}
+ // Update stats for the local account.
+ if err := p.utilF.incrementFollowersCount(ctx, fMsg.ReceivingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ // Update stats for the remote account.
+ if err := p.utilF.incrementFollowingCount(ctx, fMsg.RequestingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if err := p.federate.AcceptFollow(ctx, follow); err != nil {
log.Errorf(ctx, "error federating follow request accept: %v", err)
}
@@ -369,6 +399,11 @@ func (p *fediAPI) CreateAnnounce(ctx context.Context, fMsg messages.FromFediAPI)
return gtserror.Newf("error dereferencing announce: %w", err)
}
+ // Update stats for the remote account.
+ if err := p.utilF.incrementStatusesCount(ctx, fMsg.RequestingAccount, boost); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
// Timeline and notify the announce.
if err := p.surface.timelineAndNotifyStatus(ctx, boost); err != nil {
log.Errorf(ctx, "error timelining and notifying status: %v", err)
@@ -509,6 +544,24 @@ func (p *fediAPI) UpdateAccount(ctx context.Context, fMsg messages.FromFediAPI)
return nil
}
+func (p *fediAPI) AcceptFollow(ctx context.Context, fMsg messages.FromFediAPI) error {
+ // Update stats for the remote account.
+ if err := p.utilF.decrementFollowRequestsCount(ctx, fMsg.RequestingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ if err := p.utilF.incrementFollowersCount(ctx, fMsg.RequestingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ // Update stats for the local account.
+ if err := p.utilF.incrementFollowingCount(ctx, fMsg.ReceivingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
+ return nil
+}
+
func (p *fediAPI) UpdateStatus(ctx context.Context, fMsg messages.FromFediAPI) error {
// Cast the existing Status model attached to msg.
existing, ok := fMsg.GTSModel.(*gtsmodel.Status)
@@ -567,6 +620,11 @@ func (p *fediAPI) DeleteStatus(ctx context.Context, fMsg messages.FromFediAPI) e
log.Errorf(ctx, "error wiping status: %v", err)
}
+ // Update stats for the remote account.
+ if err := p.utilF.decrementStatusesCount(ctx, fMsg.RequestingAccount); err != nil {
+ log.Errorf(ctx, "error updating account stats: %v", err)
+ }
+
if status.InReplyToID != "" {
// Interaction counts changed on the replied status;
// uncache the prepared version from all timelines.
diff --git a/internal/processing/workers/fromfediapi_test.go b/internal/processing/workers/fromfediapi_test.go
index 51f61bd12..eb3d73e0c 100644
--- a/internal/processing/workers/fromfediapi_test.go
+++ b/internal/processing/workers/fromfediapi_test.go
@@ -55,10 +55,11 @@ func (suite *FromFediAPITestSuite) TestProcessFederationAnnounce() {
announceStatus.Visibility = boostedStatus.Visibility
err := suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
- APObjectType: ap.ActivityAnnounce,
- APActivityType: ap.ActivityCreate,
- GTSModel: announceStatus,
- ReceivingAccount: suite.testAccounts["local_account_1"],
+ APObjectType: ap.ActivityAnnounce,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: announceStatus,
+ ReceivingAccount: suite.testAccounts["local_account_1"],
+ RequestingAccount: boostingAccount,
})
suite.NoError(err)
@@ -115,10 +116,11 @@ func (suite *FromFediAPITestSuite) TestProcessReplyMention() {
// Send the replied status off to the fedi worker to be further processed.
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
- APObjectType: ap.ObjectNote,
- APActivityType: ap.ActivityCreate,
- APObjectModel: replyingStatusable,
- ReceivingAccount: suite.testAccounts["local_account_1"],
+ APObjectType: ap.ObjectNote,
+ APActivityType: ap.ActivityCreate,
+ APObjectModel: replyingStatusable,
+ ReceivingAccount: repliedAccount,
+ RequestingAccount: replyingAccount,
})
suite.NoError(err)
@@ -178,10 +180,11 @@ func (suite *FromFediAPITestSuite) TestProcessFave() {
suite.NoError(err)
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
- APObjectType: ap.ActivityLike,
- APActivityType: ap.ActivityCreate,
- GTSModel: fave,
- ReceivingAccount: favedAccount,
+ APObjectType: ap.ActivityLike,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: fave,
+ ReceivingAccount: favedAccount,
+ RequestingAccount: favingAccount,
})
suite.NoError(err)
@@ -247,10 +250,11 @@ func (suite *FromFediAPITestSuite) TestProcessFaveWithDifferentReceivingAccount(
suite.NoError(err)
err = suite.processor.Workers().ProcessFromFediAPI(context.Background(), messages.FromFediAPI{
- APObjectType: ap.ActivityLike,
- APActivityType: ap.ActivityCreate,
- GTSModel: fave,
- ReceivingAccount: receivingAccount,
+ APObjectType: ap.ActivityLike,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: fave,
+ ReceivingAccount: receivingAccount,
+ RequestingAccount: favingAccount,
})
suite.NoError(err)
@@ -318,10 +322,11 @@ func (suite *FromFediAPITestSuite) TestProcessAccountDelete() {
// now they are mufos!
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
- APObjectType: ap.ObjectProfile,
- APActivityType: ap.ActivityDelete,
- GTSModel: deletedAccount,
- ReceivingAccount: receivingAccount,
+ APObjectType: ap.ObjectProfile,
+ APActivityType: ap.ActivityDelete,
+ GTSModel: deletedAccount,
+ ReceivingAccount: receivingAccount,
+ RequestingAccount: deletedAccount,
})
suite.NoError(err)
@@ -398,10 +403,11 @@ func (suite *FromFediAPITestSuite) TestProcessFollowRequestLocked() {
suite.NoError(err)
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
- APObjectType: ap.ActivityFollow,
- APActivityType: ap.ActivityCreate,
- GTSModel: satanFollowRequestTurtle,
- ReceivingAccount: targetAccount,
+ APObjectType: ap.ActivityFollow,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: satanFollowRequestTurtle,
+ ReceivingAccount: targetAccount,
+ RequestingAccount: originAccount,
})
suite.NoError(err)
@@ -451,10 +457,11 @@ func (suite *FromFediAPITestSuite) TestProcessFollowRequestUnlocked() {
suite.NoError(err)
err = suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
- APObjectType: ap.ActivityFollow,
- APActivityType: ap.ActivityCreate,
- GTSModel: satanFollowRequestTurtle,
- ReceivingAccount: targetAccount,
+ APObjectType: ap.ActivityFollow,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: satanFollowRequestTurtle,
+ ReceivingAccount: targetAccount,
+ RequestingAccount: originAccount,
})
suite.NoError(err)
@@ -526,11 +533,12 @@ func (suite *FromFediAPITestSuite) TestCreateStatusFromIRI() {
statusCreator := suite.testAccounts["remote_account_2"]
err := suite.processor.Workers().ProcessFromFediAPI(ctx, messages.FromFediAPI{
- APObjectType: ap.ObjectNote,
- APActivityType: ap.ActivityCreate,
- GTSModel: nil, // gtsmodel is nil because this is a forwarded status -- we want to dereference it using the iri
- ReceivingAccount: receivingAccount,
- APIri: testrig.URLMustParse("http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1"),
+ APObjectType: ap.ObjectNote,
+ APActivityType: ap.ActivityCreate,
+ GTSModel: nil, // gtsmodel is nil because this is a forwarded status -- we want to dereference it using the iri
+ ReceivingAccount: receivingAccount,
+ RequestingAccount: statusCreator,
+ APIri: testrig.URLMustParse("http://example.org/users/Some_User/statuses/afaba698-5740-4e32-a702-af61aa543bc1"),
})
suite.NoError(err)
diff --git a/internal/processing/workers/util.go b/internal/processing/workers/util.go
index a38ecd336..cd936f428 100644
--- a/internal/processing/workers/util.go
+++ b/internal/processing/workers/util.go
@@ -238,3 +238,258 @@ func (u *utilF) redirectFollowers(
return true
}
+
+func (u *utilF) incrementStatusesCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+ status *gtsmodel.Status,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by incrementing status
+ // count by one and setting last posted.
+ *account.Stats.StatusesCount++
+ account.Stats.LastStatusAt = status.CreatedAt
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "statuses_count",
+ "last_status_at",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) decrementStatusesCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by decrementing
+ // status count by one.
+ //
+ // Clamp to 0 to avoid funny business.
+ *account.Stats.StatusesCount--
+ if *account.Stats.StatusesCount < 0 {
+ *account.Stats.StatusesCount = 0
+ }
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "statuses_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) incrementFollowersCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by incrementing followers
+ // count by one and setting last posted.
+ *account.Stats.FollowersCount++
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "followers_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) decrementFollowersCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by decrementing
+ // followers count by one.
+ //
+ // Clamp to 0 to avoid funny business.
+ *account.Stats.FollowersCount--
+ if *account.Stats.FollowersCount < 0 {
+ *account.Stats.FollowersCount = 0
+ }
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "followers_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) incrementFollowingCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by incrementing
+ // followers count by one.
+ *account.Stats.FollowingCount++
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "following_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) decrementFollowingCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by decrementing
+ // following count by one.
+ //
+ // Clamp to 0 to avoid funny business.
+ *account.Stats.FollowingCount--
+ if *account.Stats.FollowingCount < 0 {
+ *account.Stats.FollowingCount = 0
+ }
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "following_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) incrementFollowRequestsCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by incrementing
+ // follow requests count by one.
+ *account.Stats.FollowRequestsCount++
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "follow_requests_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}
+
+func (u *utilF) decrementFollowRequestsCount(
+ ctx context.Context,
+ account *gtsmodel.Account,
+) error {
+ // Lock on this account since we're changing stats.
+ unlock := u.state.AccountLocks.Lock(account.URI)
+ defer unlock()
+
+ // Populate stats.
+ if account.Stats == nil {
+ if err := u.state.DB.PopulateAccountStats(ctx, account); err != nil {
+ return gtserror.Newf("db error getting account stats: %w", err)
+ }
+ }
+
+ // Update stats by decrementing
+ // follow requests count by one.
+ //
+ // Clamp to 0 to avoid funny business.
+ *account.Stats.FollowRequestsCount--
+ if *account.Stats.FollowRequestsCount < 0 {
+ *account.Stats.FollowRequestsCount = 0
+ }
+ if err := u.state.DB.UpdateAccountStats(
+ ctx,
+ account.Stats,
+ "follow_requests_count",
+ ); err != nil {
+ return gtserror.Newf("db error updating account stats: %w", err)
+ }
+
+ return nil
+}