diff options
author | 2023-03-28 14:03:14 +0100 | |
---|---|---|
committer | 2023-03-28 14:03:14 +0100 | |
commit | de6e3e5f2a8ea639d76e310a11cb9bc093fef3a9 (patch) | |
tree | e2b7044e22c943425a4d351a02f862fbde783657 /internal/visibility/relevantaccounts.go | |
parent | [feature] Add list command to admin account (#1648) (diff) | |
download | gotosocial-de6e3e5f2a8ea639d76e310a11cb9bc093fef3a9.tar.xz |
[performance] refactoring + add fave / follow / request / visibility caching (#1607)
* refactor visibility checking, add caching for visibility
* invalidate visibility cache items on account / status deletes
* fix requester ID passed to visibility cache nil ptr
* de-interface caches, fix home / public timeline caching + visibility
* finish adding code comments for visibility filter
* fix angry goconst linter warnings
* actually finish adding filter visibility code comments for timeline functions
* move home timeline status author check to after visibility
* remove now-unused code
* add more code comments
* add TODO code comment, update printed cache start names
* update printed cache names on stop
* start adding separate follow(request) delete db functions, add specific visibility cache tests
* add relationship type caching
* fix getting local account follows / followed-bys, other small codebase improvements
* simplify invalidation using cache hooks, add more GetAccountBy___() functions
* fix boosting to return 404 if not boostable but no error (to not leak status ID)
* remove dead code
* improved placement of cache invalidation
* update license headers
* add example follow, follow-request config entries
* add example visibility cache configuration to config file
* use specific PutFollowRequest() instead of just Put()
* add tests for all GetAccountBy()
* add GetBlockBy() tests
* update block to check primitive fields
* update and finish adding Get{Account,Block,Follow,FollowRequest}By() tests
* fix copy-pasted code
* update envparsing test
* whitespace
* fix bun struct tag
* add license header to gtscontext
* fix old license header
* improved error creation to not use fmt.Errorf() when not needed
* fix various rebase conflicts, fix account test
* remove commented-out code, fix-up mention caching
* fix mention select bun statement
* ensure mention target account populated, pass in context to customrenderer logging
* remove more uncommented code, fix typeutil test
* add statusfave database model caching
* add status fave cache configuration
* add status fave cache example config
* woops, catch missed error. nice catch linter!
* add back testrig panic on nil db
* update example configuration to match defaults, slight tweak to cache configuration defaults
* update envparsing test with new defaults
* fetch followingget to use the follow target account
* use accounnt.IsLocal() instead of empty domain check
* use constants for the cache visibility type check
* use bun.In() for notification type restriction in db query
* include replies when fetching PublicTimeline() (to account for single-author threads in Visibility{}.StatusPublicTimelineable())
* use bun query building for nested select statements to ensure working with postgres
* update public timeline future status checks to match visibility filter
* same as previous, for home timeline
* update public timeline tests to dynamically check for appropriate statuses
* migrate accounts to allow unique constraint on public_key
* provide minimal account with publicKey
---------
Signed-off-by: kim <grufwub@gmail.com>
Co-authored-by: tsmethurst <tobi.smethurst@protonmail.com>
Diffstat (limited to 'internal/visibility/relevantaccounts.go')
-rw-r--r-- | internal/visibility/relevantaccounts.go | 230 |
1 files changed, 0 insertions, 230 deletions
diff --git a/internal/visibility/relevantaccounts.go b/internal/visibility/relevantaccounts.go deleted file mode 100644 index 2389b8544..000000000 --- a/internal/visibility/relevantaccounts.go +++ /dev/null @@ -1,230 +0,0 @@ -// GoToSocial -// Copyright (C) GoToSocial Authors admin@gotosocial.org -// SPDX-License-Identifier: AGPL-3.0-or-later -// -// This program is free software: you can redistribute it and/or modify -// it under the terms of the GNU Affero General Public License as published by -// the Free Software Foundation, either version 3 of the License, or -// (at your option) any later version. -// -// This program is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -// GNU Affero General Public License for more details. -// -// You should have received a copy of the GNU Affero General Public License -// along with this program. If not, see <http://www.gnu.org/licenses/>. - -package visibility - -import ( - "context" - "errors" - "fmt" - - "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" -) - -// relevantAccounts denotes accounts that are replied to, boosted by, or mentioned in a status. -type relevantAccounts struct { - // Who wrote the status - Account *gtsmodel.Account - // Who is the status replying to - InReplyToAccount *gtsmodel.Account - // Which accounts are mentioned (tagged) in the status - MentionedAccounts []*gtsmodel.Account - // Who authed the boosted status - BoostedAccount *gtsmodel.Account - // If the boosted status replies to another account, who does it reply to? - BoostedInReplyToAccount *gtsmodel.Account - // Who is mentioned (tagged) in the boosted status - BoostedMentionedAccounts []*gtsmodel.Account -} - -func (f *filter) relevantAccounts(ctx context.Context, status *gtsmodel.Status, getBoosted bool) (*relevantAccounts, error) { - relAccts := &relevantAccounts{ - MentionedAccounts: []*gtsmodel.Account{}, - BoostedMentionedAccounts: []*gtsmodel.Account{}, - } - - /* - Here's what we need to try and extract from the status: - - // 1. Who wrote the status - Account *gtsmodel.Account - - // 2. Who is the status replying to - InReplyToAccount *gtsmodel.Account - - // 3. Which accounts are mentioned (tagged) in the status - MentionedAccounts []*gtsmodel.Account - - if getBoosted: - // 4. Who wrote the boosted status - BoostedAccount *gtsmodel.Account - - // 5. If the boosted status replies to another account, who does it reply to? - BoostedInReplyToAccount *gtsmodel.Account - - // 6. Who is mentioned (tagged) in the boosted status - BoostedMentionedAccounts []*gtsmodel.Account - */ - - // 1. Account. - // Account might be set on the status already - if status.Account != nil { - // it was set - relAccts.Account = status.Account - } else { - // it wasn't set, so get it from the db - account, err := f.db.GetAccountByID(ctx, status.AccountID) - if err != nil { - return nil, fmt.Errorf("relevantAccounts: error getting account with id %s: %s", status.AccountID, err) - } - // set it on the status in case we need it further along - status.Account = account - // set it on relevant accounts - relAccts.Account = account - } - - // 2. InReplyToAccount - // only get this if InReplyToAccountID is set - if status.InReplyToAccountID != "" { - // InReplyToAccount might be set on the status already - if status.InReplyToAccount != nil { - // it was set - relAccts.InReplyToAccount = status.InReplyToAccount - } else { - // it wasn't set, so get it from the db - inReplyToAccount, err := f.db.GetAccountByID(ctx, status.InReplyToAccountID) - if err != nil { - return nil, fmt.Errorf("relevantAccounts: error getting inReplyToAccount with id %s: %s", status.InReplyToAccountID, err) - } - // set it on the status in case we need it further along - status.InReplyToAccount = inReplyToAccount - // set it on relevant accounts - relAccts.InReplyToAccount = inReplyToAccount - } - } - - // 3. MentionedAccounts - // First check if status.Mentions is populated with all mentions that correspond to status.MentionIDs - for _, mID := range status.MentionIDs { - if mID == "" { - continue - } - if !idIn(mID, status.Mentions) { - // mention with ID isn't in status.Mentions - mention, err := f.db.GetMention(ctx, mID) - if err != nil { - return nil, fmt.Errorf("relevantAccounts: error getting mention with id %s: %s", mID, err) - } - if mention == nil { - return nil, fmt.Errorf("relevantAccounts: mention with id %s was nil", mID) - } - status.Mentions = append(status.Mentions, mention) - } - } - // now filter mentions to make sure we only have mentions with a corresponding ID - nm := []*gtsmodel.Mention{} - for _, m := range status.Mentions { - if m == nil { - continue - } - if mentionIn(m, status.MentionIDs) { - nm = append(nm, m) - relAccts.MentionedAccounts = append(relAccts.MentionedAccounts, m.TargetAccount) - } - } - status.Mentions = nm - - if len(status.Mentions) != len(status.MentionIDs) { - return nil, errors.New("relevantAccounts: mentions length did not correspond with mentionIDs length") - } - - // if getBoosted is set, we should check the same properties on the boosted account as well - if getBoosted { - // 4, 5, 6. Boosted status items - // get the boosted status if it's not set on the status already - if status.BoostOfID != "" && status.BoostOf == nil { - boostedStatus, err := f.db.GetStatusByID(ctx, status.BoostOfID) - if err != nil { - return nil, fmt.Errorf("relevantAccounts: error getting boosted status with id %s: %s", status.BoostOfID, err) - } - status.BoostOf = boostedStatus - } - - if status.BoostOf != nil { - // return relevant accounts for the boosted status - boostedRelAccts, err := f.relevantAccounts(ctx, status.BoostOf, false) // false because we don't want to recurse - if err != nil { - return nil, fmt.Errorf("relevantAccounts: error getting relevant accounts of boosted status %s: %s", status.BoostOf.ID, err) - } - relAccts.BoostedAccount = boostedRelAccts.Account - relAccts.BoostedInReplyToAccount = boostedRelAccts.InReplyToAccount - relAccts.BoostedMentionedAccounts = boostedRelAccts.MentionedAccounts - } - } - - return relAccts, nil -} - -// domainBlockedRelevant checks through all relevant accounts attached to a status -// to make sure none of them are domain blocked by this instance. -func (f *filter) domainBlockedRelevant(ctx context.Context, r *relevantAccounts) (bool, error) { - domains := []string{} - - if r.Account != nil { - domains = append(domains, r.Account.Domain) - } - - if r.InReplyToAccount != nil { - domains = append(domains, r.InReplyToAccount.Domain) - } - - for _, a := range r.MentionedAccounts { - if a != nil { - domains = append(domains, a.Domain) - } - } - - if r.BoostedAccount != nil { - domains = append(domains, r.BoostedAccount.Domain) - } - - if r.BoostedInReplyToAccount != nil { - domains = append(domains, r.BoostedInReplyToAccount.Domain) - } - - for _, a := range r.BoostedMentionedAccounts { - if a != nil { - domains = append(domains, a.Domain) - } - } - - return f.db.AreDomainsBlocked(ctx, domains) -} - -func idIn(id string, mentions []*gtsmodel.Mention) bool { - for _, m := range mentions { - if m == nil { - continue - } - if m.ID == id { - return true - } - } - return false -} - -func mentionIn(mention *gtsmodel.Mention, ids []string) bool { - if mention == nil { - return false - } - for _, i := range ids { - if mention.ID == i { - return true - } - } - return false -} |