2023-03-12 16:00:57 +01:00
|
|
|
// 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/>.
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2021-08-25 15:34:33 +02:00
|
|
|
package bundb
|
2021-08-20 12:26:56 +02:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"errors"
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
"fmt"
|
2022-12-01 16:06:09 +01:00
|
|
|
"strings"
|
2021-08-20 12:26:56 +02:00
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/config"
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
|
2023-05-07 19:53:21 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
2021-08-20 12:26:56 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
2023-05-22 16:32:36 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/id"
|
2022-07-19 10:47:55 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/log"
|
2022-12-08 18:35:14 +01:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/state"
|
2023-05-07 19:53:21 +02:00
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/util"
|
2021-08-25 15:34:33 +02:00
|
|
|
"github.com/uptrace/bun"
|
2022-06-10 10:56:49 +02:00
|
|
|
"github.com/uptrace/bun/dialect"
|
2021-08-20 12:26:56 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type accountDB struct {
|
2023-07-25 10:34:05 +02:00
|
|
|
db *WrappedDB
|
2022-12-08 18:35:14 +01:00
|
|
|
state *state.State
|
2022-11-15 19:45:15 +01:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByID(ctx context.Context, id string) (*gtsmodel.Account, error) {
|
2021-09-01 11:08:21 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
2022-11-15 19:45:15 +01:00
|
|
|
"ID",
|
2021-09-01 11:08:21 +02:00
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.id"), id).
|
|
|
|
Scan(ctx)
|
2021-09-01 11:08:21 +02:00
|
|
|
},
|
2022-11-15 19:45:15 +01:00
|
|
|
id,
|
2021-09-01 11:08:21 +02:00
|
|
|
)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-24 14:14:13 +02:00
|
|
|
func (a *accountDB) GetAccountsByIDs(ctx context.Context, ids []string) ([]*gtsmodel.Account, error) {
|
|
|
|
accounts := make([]*gtsmodel.Account, 0, len(ids))
|
|
|
|
|
|
|
|
for _, id := range ids {
|
|
|
|
// Attempt to fetch account from DB.
|
|
|
|
account, err := a.GetAccountByID(
|
|
|
|
gtscontext.SetBarebones(ctx),
|
|
|
|
id,
|
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
log.Errorf(ctx, "error getting account %q: %v", id, err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// Append account to return slice.
|
|
|
|
accounts = append(accounts, account)
|
|
|
|
}
|
|
|
|
|
|
|
|
return accounts, nil
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByURI(ctx context.Context, uri string) (*gtsmodel.Account, error) {
|
2021-09-01 11:08:21 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
2022-11-15 19:45:15 +01:00
|
|
|
"URI",
|
2021-09-01 11:08:21 +02:00
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.uri"), uri).
|
|
|
|
Scan(ctx)
|
2021-09-01 11:08:21 +02:00
|
|
|
},
|
2022-11-15 19:45:15 +01:00
|
|
|
uri,
|
2021-09-01 11:08:21 +02:00
|
|
|
)
|
|
|
|
}
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByURL(ctx context.Context, url string) (*gtsmodel.Account, error) {
|
2021-09-01 11:08:21 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
2022-11-15 19:45:15 +01:00
|
|
|
"URL",
|
2021-09-01 11:08:21 +02:00
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.url"), url).
|
|
|
|
Scan(ctx)
|
2021-09-01 11:08:21 +02:00
|
|
|
},
|
2022-11-15 19:45:15 +01:00
|
|
|
url,
|
2021-09-01 11:08:21 +02:00
|
|
|
)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByUsernameDomain(ctx context.Context, username string, domain string) (*gtsmodel.Account, error) {
|
2023-05-07 19:53:21 +02:00
|
|
|
if domain != "" {
|
|
|
|
// Normalize the domain as punycode
|
|
|
|
var err error
|
|
|
|
domain, err = util.Punify(domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-20 22:47:19 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
2022-11-15 19:45:15 +01:00
|
|
|
"Username.Domain",
|
2022-08-20 22:47:19 +02:00
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account)
|
2022-08-20 22:47:19 +02:00
|
|
|
|
|
|
|
if domain != "" {
|
2022-12-01 16:06:09 +01:00
|
|
|
q = q.
|
|
|
|
Where("LOWER(?) = ?", bun.Ident("account.username"), strings.ToLower(username)).
|
|
|
|
Where("? = ?", bun.Ident("account.domain"), domain)
|
2022-08-20 22:47:19 +02:00
|
|
|
} else {
|
2022-12-01 16:06:09 +01:00
|
|
|
q = q.
|
|
|
|
Where("? = ?", bun.Ident("account.username"), strings.ToLower(username)). // usernames on our instance are always lowercase
|
|
|
|
Where("? IS NULL", bun.Ident("account.domain"))
|
2022-08-20 22:47:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return q.Scan(ctx)
|
|
|
|
},
|
2022-11-15 19:45:15 +01:00
|
|
|
username,
|
|
|
|
domain,
|
2022-08-20 22:47:19 +02:00
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByPubkeyID(ctx context.Context, id string) (*gtsmodel.Account, error) {
|
2022-09-02 11:58:42 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
2022-11-15 19:45:15 +01:00
|
|
|
"PublicKeyURI",
|
2022-09-02 11:58:42 +02:00
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.public_key_uri"), id).
|
|
|
|
Scan(ctx)
|
2022-09-02 11:58:42 +02:00
|
|
|
},
|
2022-11-15 19:45:15 +01:00
|
|
|
id,
|
2022-09-02 11:58:42 +02:00
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByInboxURI(ctx context.Context, uri string) (*gtsmodel.Account, error) {
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
|
|
|
"InboxURI",
|
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.inbox_uri"), uri).
|
|
|
|
Scan(ctx)
|
|
|
|
},
|
|
|
|
uri,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByOutboxURI(ctx context.Context, uri string) (*gtsmodel.Account, error) {
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
|
|
|
"OutboxURI",
|
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.outbox_uri"), uri).
|
|
|
|
Scan(ctx)
|
|
|
|
},
|
|
|
|
uri,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByFollowersURI(ctx context.Context, uri string) (*gtsmodel.Account, error) {
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
|
|
|
"FollowersURI",
|
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.followers_uri"), uri).
|
|
|
|
Scan(ctx)
|
|
|
|
},
|
|
|
|
uri,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountByFollowingURI(ctx context.Context, uri string) (*gtsmodel.Account, error) {
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
return a.getAccount(
|
|
|
|
ctx,
|
|
|
|
"FollowingURI",
|
|
|
|
func(account *gtsmodel.Account) error {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.NewSelect().
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.following_uri"), uri).
|
|
|
|
Scan(ctx)
|
|
|
|
},
|
|
|
|
uri,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetInstanceAccount(ctx context.Context, domain string) (*gtsmodel.Account, error) {
|
2022-11-15 19:45:15 +01:00
|
|
|
var username string
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2022-11-15 19:45:15 +01:00
|
|
|
if domain == "" {
|
|
|
|
// I.e. our local instance account
|
|
|
|
username = config.GetHost()
|
|
|
|
} else {
|
|
|
|
// A remote instance account
|
|
|
|
username = domain
|
|
|
|
}
|
|
|
|
|
|
|
|
return a.GetAccountByUsernameDomain(ctx, username, domain)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) getAccount(ctx context.Context, lookup string, dbQuery func(*gtsmodel.Account) error, keyParts ...any) (*gtsmodel.Account, error) {
|
2022-11-18 18:29:25 +01:00
|
|
|
// Fetch account from database cache with loader callback
|
2022-12-08 18:35:14 +01:00
|
|
|
account, err := a.state.Caches.GTS.Account().Load(lookup, func() (*gtsmodel.Account, error) {
|
2022-11-15 19:45:15 +01:00
|
|
|
var account gtsmodel.Account
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2021-09-01 11:08:21 +02:00
|
|
|
// Not cached! Perform database query
|
2022-11-15 19:45:15 +01:00
|
|
|
if err := dbQuery(&account); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2021-09-01 11:08:21 +02:00
|
|
|
}
|
|
|
|
|
2022-11-15 19:45:15 +01:00
|
|
|
return &account, nil
|
|
|
|
}, keyParts...)
|
2022-11-18 18:29:25 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
if gtscontext.Barebones(ctx) {
|
|
|
|
// no need to fully populate.
|
|
|
|
return account, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Further populate the account fields where applicable.
|
|
|
|
if err := a.PopulateAccount(ctx, account); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return account, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (a *accountDB) PopulateAccount(ctx context.Context, account *gtsmodel.Account) error {
|
2023-05-07 19:53:21 +02:00
|
|
|
var (
|
|
|
|
err error
|
|
|
|
errs = make(gtserror.MultiError, 0, 3)
|
|
|
|
)
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
|
|
|
|
if account.AvatarMediaAttachment == nil && account.AvatarMediaAttachmentID != "" {
|
|
|
|
// Account avatar attachment is not set, fetch from database.
|
|
|
|
account.AvatarMediaAttachment, err = a.state.DB.GetAttachmentByID(
|
|
|
|
ctx, // these are already barebones
|
|
|
|
account.AvatarMediaAttachmentID,
|
|
|
|
)
|
2023-02-03 21:03:05 +01:00
|
|
|
if err != nil {
|
2023-05-07 19:53:21 +02:00
|
|
|
errs.Append(fmt.Errorf("error populating account avatar: %w", err))
|
2023-02-03 21:03:05 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
if account.HeaderMediaAttachment == nil && account.HeaderMediaAttachmentID != "" {
|
|
|
|
// Account header attachment is not set, fetch from database.
|
|
|
|
account.HeaderMediaAttachment, err = a.state.DB.GetAttachmentByID(
|
|
|
|
ctx, // these are already barebones
|
|
|
|
account.HeaderMediaAttachmentID,
|
|
|
|
)
|
2023-02-03 21:03:05 +01:00
|
|
|
if err != nil {
|
2023-05-07 19:53:21 +02:00
|
|
|
errs.Append(fmt.Errorf("error populating account header: %w", err))
|
2023-02-03 21:03:05 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
[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>
2023-03-28 15:03:14 +02:00
|
|
|
if !account.EmojisPopulated() {
|
|
|
|
// Account emojis are out-of-date with IDs, repopulate.
|
|
|
|
account.Emojis, err = a.state.DB.GetEmojisByIDs(
|
|
|
|
ctx, // these are already barebones
|
|
|
|
account.EmojiIDs,
|
|
|
|
)
|
2022-11-18 18:29:25 +01:00
|
|
|
if err != nil {
|
2023-05-07 19:53:21 +02:00
|
|
|
errs.Append(fmt.Errorf("error populating account emojis: %w", err))
|
2022-11-18 18:29:25 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-05-07 19:53:21 +02:00
|
|
|
return errs.Combine()
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) PutAccount(ctx context.Context, account *gtsmodel.Account) error {
|
2022-12-08 18:35:14 +01:00
|
|
|
return a.state.Caches.GTS.Account().Store(account, func() error {
|
2022-11-15 19:45:15 +01:00
|
|
|
// It is safe to run this database transaction within cache.Store
|
|
|
|
// as the cache does not attempt a mutex lock until AFTER hook.
|
|
|
|
//
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.RunInTx(ctx, func(tx bun.Tx) error {
|
2022-11-15 19:45:15 +01:00
|
|
|
// create links between this account and any emojis it uses
|
|
|
|
for _, i := range account.EmojiIDs {
|
|
|
|
if _, err := tx.NewInsert().Model(>smodel.AccountToEmoji{
|
|
|
|
AccountID: account.ID,
|
|
|
|
EmojiID: i,
|
|
|
|
}).Exec(ctx); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2022-09-26 11:56:01 +02:00
|
|
|
}
|
|
|
|
|
2022-11-15 19:45:15 +01:00
|
|
|
// insert the account
|
|
|
|
_, err := tx.NewInsert().Model(account).Exec(ctx)
|
|
|
|
return err
|
|
|
|
})
|
|
|
|
})
|
2022-09-26 11:56:01 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) UpdateAccount(ctx context.Context, account *gtsmodel.Account, columns ...string) error {
|
2021-08-25 15:34:33 +02:00
|
|
|
account.UpdatedAt = time.Now()
|
2023-03-20 19:10:08 +01:00
|
|
|
if len(columns) > 0 {
|
|
|
|
// If we're updating by column, ensure "updated_at" is included.
|
|
|
|
columns = append(columns, "updated_at")
|
|
|
|
}
|
2021-08-25 15:34:33 +02:00
|
|
|
|
2023-05-12 11:15:54 +02:00
|
|
|
return a.state.Caches.GTS.Account().Store(account, func() error {
|
2022-11-15 19:45:15 +01:00
|
|
|
// It is safe to run this database transaction within cache.Store
|
|
|
|
// as the cache does not attempt a mutex lock until AFTER hook.
|
|
|
|
//
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.RunInTx(ctx, func(tx bun.Tx) error {
|
2022-11-15 19:45:15 +01:00
|
|
|
// create links between this account and any emojis it uses
|
|
|
|
// first clear out any old emoji links
|
2022-10-08 13:50:48 +02:00
|
|
|
if _, err := tx.
|
2022-11-15 19:45:15 +01:00
|
|
|
NewDelete().
|
|
|
|
TableExpr("? AS ?", bun.Ident("account_to_emojis"), bun.Ident("account_to_emoji")).
|
|
|
|
Where("? = ?", bun.Ident("account_to_emoji.account_id"), account.ID).
|
|
|
|
Exec(ctx); err != nil {
|
2022-09-26 11:56:01 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-11-15 19:45:15 +01:00
|
|
|
// now populate new emoji links
|
|
|
|
for _, i := range account.EmojiIDs {
|
|
|
|
if _, err := tx.
|
|
|
|
NewInsert().
|
|
|
|
Model(>smodel.AccountToEmoji{
|
|
|
|
AccountID: account.ID,
|
|
|
|
EmojiID: i,
|
|
|
|
}).Exec(ctx); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
2021-09-01 11:08:21 +02:00
|
|
|
|
2022-11-15 19:45:15 +01:00
|
|
|
// update the account
|
|
|
|
_, err := tx.NewUpdate().
|
|
|
|
Model(account).
|
|
|
|
Where("? = ?", bun.Ident("account.id"), account.ID).
|
2023-03-20 19:10:08 +01:00
|
|
|
Column(columns...).
|
2022-11-15 19:45:15 +01:00
|
|
|
Exec(ctx)
|
|
|
|
return err
|
|
|
|
})
|
|
|
|
})
|
2021-08-25 15:34:33 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) DeleteAccount(ctx context.Context, id string) error {
|
2023-05-12 11:15:54 +02:00
|
|
|
defer a.state.Caches.GTS.Account().Invalidate("ID", id)
|
|
|
|
|
|
|
|
// Load account into cache before attempting a delete,
|
|
|
|
// as we need it cached in order to trigger the invalidate
|
|
|
|
// callback. This in turn invalidates others.
|
|
|
|
_, err := a.GetAccountByID(gtscontext.SetBarebones(ctx), id)
|
|
|
|
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
|
|
|
// NOTE: even if db.ErrNoEntries is returned, we
|
|
|
|
// still run the below transaction to ensure related
|
|
|
|
// objects are appropriately deleted.
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.RunInTx(ctx, func(tx bun.Tx) error {
|
2022-10-08 13:50:48 +02:00
|
|
|
// clear out any emoji links
|
|
|
|
if _, err := tx.
|
|
|
|
NewDelete().
|
|
|
|
TableExpr("? AS ?", bun.Ident("account_to_emojis"), bun.Ident("account_to_emoji")).
|
|
|
|
Where("? = ?", bun.Ident("account_to_emoji.account_id"), id).
|
|
|
|
Exec(ctx); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// delete the account
|
|
|
|
_, err := tx.
|
2022-11-15 19:45:15 +01:00
|
|
|
NewDelete().
|
2022-10-08 13:50:48 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("accounts"), bun.Ident("account")).
|
|
|
|
Where("? = ?", bun.Ident("account.id"), id).
|
|
|
|
Exec(ctx)
|
|
|
|
return err
|
2023-05-12 11:15:54 +02:00
|
|
|
})
|
2022-10-08 13:50:48 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountLastPosted(ctx context.Context, accountID string, webOnly bool) (time.Time, error) {
|
2022-10-08 14:00:39 +02:00
|
|
|
createdAt := time.Time{}
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewSelect().
|
2022-10-08 14:00:39 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
2022-10-08 13:50:48 +02:00
|
|
|
Column("status.created_at").
|
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID).
|
|
|
|
Order("status.id DESC").
|
|
|
|
Limit(1)
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2022-10-08 14:00:39 +02:00
|
|
|
if webOnly {
|
|
|
|
q = q.
|
2023-07-05 12:34:37 +02:00
|
|
|
Where("? IS NULL", bun.Ident("status.in_reply_to_uri")).
|
|
|
|
Where("? IS NULL", bun.Ident("status.boost_of_id")).
|
2022-10-08 14:00:39 +02:00
|
|
|
Where("? = ?", bun.Ident("status.visibility"), gtsmodel.VisibilityPublic).
|
|
|
|
Where("? = ?", bun.Ident("status.federated"), true)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := q.Scan(ctx, &createdAt); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return time.Time{}, a.db.ProcessError(err)
|
2021-08-29 16:41:41 +02:00
|
|
|
}
|
2022-10-08 14:00:39 +02:00
|
|
|
return createdAt, nil
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) SetAccountHeaderOrAvatar(ctx context.Context, mediaAttachment *gtsmodel.MediaAttachment, accountID string) error {
|
2022-08-15 12:35:05 +02:00
|
|
|
if *mediaAttachment.Avatar && *mediaAttachment.Header {
|
2021-08-20 12:26:56 +02:00
|
|
|
return errors.New("one media attachment cannot be both header and avatar")
|
|
|
|
}
|
|
|
|
|
2022-10-08 13:50:48 +02:00
|
|
|
var column bun.Ident
|
2021-11-22 08:46:19 +01:00
|
|
|
switch {
|
2022-08-15 12:35:05 +02:00
|
|
|
case *mediaAttachment.Avatar:
|
2022-10-08 13:50:48 +02:00
|
|
|
column = bun.Ident("account.avatar_media_attachment_id")
|
2022-08-15 12:35:05 +02:00
|
|
|
case *mediaAttachment.Header:
|
2022-10-08 13:50:48 +02:00
|
|
|
column = bun.Ident("account.header_media_attachment_id")
|
2021-11-22 08:46:19 +01:00
|
|
|
default:
|
2021-08-20 12:26:56 +02:00
|
|
|
return errors.New("given media attachment was neither a header nor an avatar")
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: there are probably more side effects here that need to be handled
|
2023-07-25 10:34:05 +02:00
|
|
|
if _, err := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewInsert().
|
|
|
|
Model(mediaAttachment).
|
|
|
|
Exec(ctx); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.ProcessError(err)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
2022-10-08 13:50:48 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
if _, err := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewUpdate().
|
2022-10-08 13:50:48 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("accounts"), bun.Ident("account")).
|
|
|
|
Set("? = ?", column, mediaAttachment.ID).
|
|
|
|
Where("? = ?", bun.Ident("account.id"), accountID).
|
2021-08-25 15:34:33 +02:00
|
|
|
Exec(ctx); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return a.db.ProcessError(err)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
2021-08-29 16:41:41 +02:00
|
|
|
|
2021-08-20 12:26:56 +02:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountCustomCSSByUsername(ctx context.Context, username string) (string, error) {
|
2022-09-12 13:14:29 +02:00
|
|
|
account, err := a.GetAccountByUsernameDomain(ctx, username, "")
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return account.CustomCSS, nil
|
|
|
|
}
|
|
|
|
|
2023-07-24 14:14:13 +02:00
|
|
|
func (a *accountDB) GetAccountsUsingEmoji(ctx context.Context, emojiID string) ([]*gtsmodel.Account, error) {
|
|
|
|
var accountIDs []string
|
|
|
|
|
|
|
|
// Create SELECT account query.
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.NewSelect().
|
2023-07-24 14:14:13 +02:00
|
|
|
Table("accounts").
|
|
|
|
Column("id")
|
|
|
|
|
|
|
|
// Append a WHERE LIKE clause to the query
|
|
|
|
// that checks the `emoji` column for any
|
|
|
|
// text containing this specific emoji ID.
|
|
|
|
//
|
|
|
|
// The reason we do this instead of doing a
|
|
|
|
// `WHERE ? IN (emojis)` is that the latter
|
|
|
|
// ends up being much MUCH slower, and the
|
|
|
|
// database stores this ID-array-column as
|
|
|
|
// text anyways, allowing a simple LIKE query.
|
|
|
|
q = whereLike(q, "emojis", emojiID)
|
|
|
|
|
|
|
|
// Execute the query, scanning destination into accountIDs.
|
|
|
|
if _, err := q.Exec(ctx, &accountIDs); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2023-07-24 14:14:13 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// Convert account IDs into account objects.
|
|
|
|
return a.GetAccountsByIDs(ctx, accountIDs)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountFaves(ctx context.Context, accountID string) ([]*gtsmodel.StatusFave, error) {
|
2021-08-25 15:34:33 +02:00
|
|
|
faves := new([]*gtsmodel.StatusFave)
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
if err := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewSelect().
|
|
|
|
Model(faves).
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? = ?", bun.Ident("status_fave.account_id"), accountID).
|
2021-08-25 15:34:33 +02:00
|
|
|
Scan(ctx); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
2021-08-29 16:41:41 +02:00
|
|
|
|
2021-08-25 15:34:33 +02:00
|
|
|
return *faves, nil
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) CountAccountStatuses(ctx context.Context, accountID string) (int, error) {
|
|
|
|
return a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewSelect().
|
2022-10-08 13:50:48 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID).
|
2021-08-25 15:34:33 +02:00
|
|
|
Count(ctx)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) CountAccountPinned(ctx context.Context, accountID string) (int, error) {
|
|
|
|
return a.db.
|
2023-02-25 13:16:30 +01:00
|
|
|
NewSelect().
|
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID).
|
|
|
|
Where("? IS NOT NULL", bun.Ident("status.pinned_at")).
|
|
|
|
Count(ctx)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountStatuses(ctx context.Context, accountID string, limit int, excludeReplies bool, excludeReblogs bool, maxID string, minID string, mediaOnly bool, publicOnly bool) ([]*gtsmodel.Status, error) {
|
2023-05-22 16:32:36 +02:00
|
|
|
// Ensure reasonable
|
|
|
|
if limit < 0 {
|
|
|
|
limit = 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make educated guess for slice size
|
|
|
|
var (
|
|
|
|
statusIDs = make([]string, 0, limit)
|
|
|
|
frontToBack = true
|
|
|
|
)
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewSelect().
|
2022-10-08 13:50:48 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
2023-05-22 16:32:36 +02:00
|
|
|
// Select only IDs from table
|
2022-10-08 13:50:48 +02:00
|
|
|
Column("status.id").
|
2023-05-22 16:32:36 +02:00
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID)
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2021-10-24 11:57:39 +02:00
|
|
|
if excludeReplies {
|
2023-05-22 16:32:36 +02:00
|
|
|
q = q.WhereGroup(" AND ", func(*bun.SelectQuery) *bun.SelectQuery {
|
2022-08-27 11:35:31 +02:00
|
|
|
return q.
|
2023-05-22 16:32:36 +02:00
|
|
|
// Do include self replies (threads), but
|
|
|
|
// don't include replies to other people.
|
|
|
|
Where("? = ?", bun.Ident("status.in_reply_to_account_id"), accountID).
|
|
|
|
WhereOr("? IS NULL", bun.Ident("status.in_reply_to_uri"))
|
|
|
|
})
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2022-04-15 14:33:01 +02:00
|
|
|
if excludeReblogs {
|
2023-05-22 16:32:36 +02:00
|
|
|
q = q.Where("? IS NULL", bun.Ident("status.boost_of_id"))
|
2021-10-24 11:57:39 +02:00
|
|
|
}
|
|
|
|
|
2021-08-25 15:34:33 +02:00
|
|
|
if mediaOnly {
|
2023-05-22 16:32:36 +02:00
|
|
|
// Attachments are stored as a json object; this
|
|
|
|
// implementation differs between SQLite and Postgres,
|
2022-06-10 10:56:49 +02:00
|
|
|
// so we have to be thorough to cover all eventualities
|
2021-08-25 15:34:33 +02:00
|
|
|
q = q.WhereGroup(" AND ", func(q *bun.SelectQuery) *bun.SelectQuery {
|
2023-07-25 10:34:05 +02:00
|
|
|
switch a.db.Dialect().Name() {
|
2022-06-10 10:56:49 +02:00
|
|
|
case dialect.PG:
|
|
|
|
return q.
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? IS NOT NULL", bun.Ident("status.attachments")).
|
|
|
|
Where("? != '{}'", bun.Ident("status.attachments"))
|
2022-06-10 10:56:49 +02:00
|
|
|
case dialect.SQLite:
|
|
|
|
return q.
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? IS NOT NULL", bun.Ident("status.attachments")).
|
|
|
|
Where("? != ''", bun.Ident("status.attachments")).
|
|
|
|
Where("? != 'null'", bun.Ident("status.attachments")).
|
|
|
|
Where("? != '{}'", bun.Ident("status.attachments")).
|
|
|
|
Where("? != '[]'", bun.Ident("status.attachments"))
|
2022-06-10 10:56:49 +02:00
|
|
|
default:
|
2023-02-17 12:02:29 +01:00
|
|
|
log.Panic(ctx, "db dialect was neither pg nor sqlite")
|
2022-06-10 10:56:49 +02:00
|
|
|
return q
|
|
|
|
}
|
2021-08-25 15:34:33 +02:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-10-24 11:57:39 +02:00
|
|
|
if publicOnly {
|
2022-10-08 13:50:48 +02:00
|
|
|
q = q.Where("? = ?", bun.Ident("status.visibility"), gtsmodel.VisibilityPublic)
|
2021-08-26 22:06:34 +02:00
|
|
|
}
|
|
|
|
|
2023-05-22 16:32:36 +02:00
|
|
|
// return only statuses LOWER (ie., older) than maxID
|
|
|
|
if maxID == "" {
|
|
|
|
maxID = id.Highest
|
|
|
|
}
|
|
|
|
q = q.Where("? < ?", bun.Ident("status.id"), maxID)
|
|
|
|
|
|
|
|
if minID != "" {
|
|
|
|
// return only statuses HIGHER (ie., newer) than minID
|
|
|
|
q = q.Where("? > ?", bun.Ident("status.id"), minID)
|
|
|
|
|
|
|
|
// page up
|
|
|
|
frontToBack = false
|
|
|
|
}
|
|
|
|
|
|
|
|
if limit > 0 {
|
|
|
|
// limit amount of statuses returned
|
|
|
|
q = q.Limit(limit)
|
|
|
|
}
|
|
|
|
|
|
|
|
if frontToBack {
|
|
|
|
// Page down.
|
|
|
|
q = q.Order("status.id DESC")
|
|
|
|
} else {
|
|
|
|
// Page up.
|
|
|
|
q = q.Order("status.id ASC")
|
|
|
|
}
|
|
|
|
|
2022-07-10 17:18:21 +02:00
|
|
|
if err := q.Scan(ctx, &statusIDs); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-05-22 16:32:36 +02:00
|
|
|
// If we're paging up, we still want statuses
|
|
|
|
// to be sorted by ID desc, so reverse ids slice.
|
|
|
|
// https://zchee.github.io/golang-wiki/SliceTricks/#reversing
|
|
|
|
if !frontToBack {
|
|
|
|
for l, r := 0, len(statusIDs)-1; l < r; l, r = l+1, r-1 {
|
|
|
|
statusIDs[l], statusIDs[r] = statusIDs[r], statusIDs[l]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-13 09:57:47 +02:00
|
|
|
return a.statusesFromIDs(ctx, statusIDs)
|
|
|
|
}
|
2021-08-20 12:26:56 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountPinnedStatuses(ctx context.Context, accountID string) ([]*gtsmodel.Status, error) {
|
2023-02-25 13:16:30 +01:00
|
|
|
statusIDs := []string{}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.
|
2023-02-25 13:16:30 +01:00
|
|
|
NewSelect().
|
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
|
|
|
Column("status.id").
|
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID).
|
|
|
|
Where("? IS NOT NULL", bun.Ident("status.pinned_at")).
|
|
|
|
Order("status.pinned_at DESC")
|
|
|
|
|
|
|
|
if err := q.Scan(ctx, &statusIDs); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2023-02-25 13:16:30 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return a.statusesFromIDs(ctx, statusIDs)
|
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountWebStatuses(ctx context.Context, accountID string, limit int, maxID string) ([]*gtsmodel.Status, error) {
|
2023-05-22 16:32:36 +02:00
|
|
|
// Ensure reasonable
|
|
|
|
if limit < 0 {
|
|
|
|
limit = 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make educated guess for slice size
|
|
|
|
statusIDs := make([]string, 0, limit)
|
2022-07-10 17:18:21 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
q := a.db.
|
2022-07-13 09:57:47 +02:00
|
|
|
NewSelect().
|
2022-10-08 13:50:48 +02:00
|
|
|
TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status")).
|
2023-05-22 16:32:36 +02:00
|
|
|
// Select only IDs from table
|
2022-10-08 13:50:48 +02:00
|
|
|
Column("status.id").
|
|
|
|
Where("? = ?", bun.Ident("status.account_id"), accountID).
|
2023-05-22 16:32:36 +02:00
|
|
|
// Don't show replies or boosts.
|
|
|
|
Where("? IS NULL", bun.Ident("status.in_reply_to_uri")).
|
|
|
|
Where("? IS NULL", bun.Ident("status.boost_of_id")).
|
|
|
|
// Only Public statuses.
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? = ?", bun.Ident("status.visibility"), gtsmodel.VisibilityPublic).
|
2023-05-22 16:32:36 +02:00
|
|
|
// Don't show local-only statuses on the web view.
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? = ?", bun.Ident("status.federated"), true)
|
2022-07-10 17:18:21 +02:00
|
|
|
|
2023-05-22 16:32:36 +02:00
|
|
|
// return only statuses LOWER (ie., older) than maxID
|
|
|
|
if maxID == "" {
|
|
|
|
maxID = id.Highest
|
|
|
|
}
|
|
|
|
q = q.Where("? < ?", bun.Ident("status.id"), maxID)
|
|
|
|
|
|
|
|
if limit > 0 {
|
|
|
|
// limit amount of statuses returned
|
|
|
|
q = q.Limit(limit)
|
|
|
|
}
|
|
|
|
|
|
|
|
if limit > 0 {
|
|
|
|
// limit amount of statuses returned
|
|
|
|
q = q.Limit(limit)
|
2022-07-10 17:18:21 +02:00
|
|
|
}
|
|
|
|
|
2023-05-22 16:32:36 +02:00
|
|
|
q = q.Order("status.id DESC")
|
2022-07-13 09:57:47 +02:00
|
|
|
|
|
|
|
if err := q.Scan(ctx, &statusIDs); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, a.db.ProcessError(err)
|
2022-07-13 09:57:47 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return a.statusesFromIDs(ctx, statusIDs)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) GetAccountBlocks(ctx context.Context, accountID string, maxID string, sinceID string, limit int) ([]*gtsmodel.Account, string, string, error) {
|
2021-08-20 12:26:56 +02:00
|
|
|
blocks := []*gtsmodel.Block{}
|
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
fq := a.db.
|
2021-08-25 15:34:33 +02:00
|
|
|
NewSelect().
|
|
|
|
Model(&blocks).
|
2022-10-08 13:50:48 +02:00
|
|
|
Where("? = ?", bun.Ident("block.account_id"), accountID).
|
2021-08-20 12:26:56 +02:00
|
|
|
Relation("TargetAccount").
|
|
|
|
Order("block.id DESC")
|
|
|
|
|
|
|
|
if maxID != "" {
|
2022-10-08 13:50:48 +02:00
|
|
|
fq = fq.Where("? < ?", bun.Ident("block.id"), maxID)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if sinceID != "" {
|
2022-10-08 13:50:48 +02:00
|
|
|
fq = fq.Where("? > ?", bun.Ident("block.id"), sinceID)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if limit > 0 {
|
|
|
|
fq = fq.Limit(limit)
|
|
|
|
}
|
|
|
|
|
2021-11-22 08:46:19 +01:00
|
|
|
if err := fq.Scan(ctx); err != nil {
|
2023-07-25 10:34:05 +02:00
|
|
|
return nil, "", "", a.db.ProcessError(err)
|
2021-08-20 12:26:56 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(blocks) == 0 {
|
|
|
|
return nil, "", "", db.ErrNoEntries
|
|
|
|
}
|
|
|
|
|
|
|
|
accounts := []*gtsmodel.Account{}
|
|
|
|
for _, b := range blocks {
|
|
|
|
accounts = append(accounts, b.TargetAccount)
|
|
|
|
}
|
|
|
|
|
|
|
|
nextMaxID := blocks[len(blocks)-1].ID
|
|
|
|
prevMinID := blocks[0].ID
|
|
|
|
return accounts, nextMaxID, prevMinID, nil
|
|
|
|
}
|
2022-07-13 09:57:47 +02:00
|
|
|
|
2023-07-25 10:34:05 +02:00
|
|
|
func (a *accountDB) statusesFromIDs(ctx context.Context, statusIDs []string) ([]*gtsmodel.Status, error) {
|
2022-07-13 09:57:47 +02:00
|
|
|
// Catch case of no statuses early
|
|
|
|
if len(statusIDs) == 0 {
|
|
|
|
return nil, db.ErrNoEntries
|
|
|
|
}
|
|
|
|
|
|
|
|
// Allocate return slice (will be at most len statusIDS)
|
|
|
|
statuses := make([]*gtsmodel.Status, 0, len(statusIDs))
|
|
|
|
|
|
|
|
for _, id := range statusIDs {
|
|
|
|
// Fetch from status from database by ID
|
2022-12-08 18:35:14 +01:00
|
|
|
status, err := a.state.DB.GetStatusByID(ctx, id)
|
2022-07-13 09:57:47 +02:00
|
|
|
if err != nil {
|
2023-02-17 12:02:29 +01:00
|
|
|
log.Errorf(ctx, "error getting status %q: %v", id, err)
|
2022-07-13 09:57:47 +02:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// Append to return slice
|
|
|
|
statuses = append(statuses, status)
|
|
|
|
}
|
|
|
|
|
|
|
|
return statuses, nil
|
|
|
|
}
|