diff options
| author | 2025-06-24 17:24:34 +0200 | |
|---|---|---|
| committer | 2025-06-24 17:24:34 +0200 | |
| commit | 996da6e0291b158093d917ca76933584f464d668 (patch) | |
| tree | 38c12b20f76076f08ef5c8b8715ba3d8629fa0fb /internal/db/bundb/filterstatus.go | |
| parent | [bugfix] update the default configuration to not set a db type or address, to... (diff) | |
| download | gotosocial-996da6e0291b158093d917ca76933584f464d668.tar.xz | |
[performance] filter model and database table improvements (#4277)
- removes unnecessary fields / columns (created_at, updated_at)
- replaces filter.context_* columns with singular filter.contexts bit field which should save both struct memory and database space
- replaces filter.action string with integer enum type which should save both struct memory and database space
- adds links from filter to filter_* tables with Filter{}.KeywordIDs and Filter{}.StatusIDs fields (this also means we now have those ID slices cached, which reduces some lookups)
- removes account_id fields from filter_* tables, since there's a more direct connection between filter and filter_* tables, and filter.account_id already exists
- refactors a bunch of the filter processor logic to save on code repetition, factor in the above changes, fix a few bugs with missed error returns and bring it more in-line with some of our newer code
Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4277
Co-authored-by: kim <grufwub@gmail.com>
Co-committed-by: kim <grufwub@gmail.com>
Diffstat (limited to 'internal/db/bundb/filterstatus.go')
| -rw-r--r-- | internal/db/bundb/filterstatus.go | 101 |
1 files changed, 14 insertions, 87 deletions
diff --git a/internal/db/bundb/filterstatus.go b/internal/db/bundb/filterstatus.go index a14e2a7b4..d15705a14 100644 --- a/internal/db/bundb/filterstatus.go +++ b/internal/db/bundb/filterstatus.go @@ -19,86 +19,38 @@ package bundb import ( "context" - "slices" - "time" - "code.superseriousbusiness.org/gotosocial/internal/gtscontext" - "code.superseriousbusiness.org/gotosocial/internal/gtserror" "code.superseriousbusiness.org/gotosocial/internal/gtsmodel" "code.superseriousbusiness.org/gotosocial/internal/util/xslices" "github.com/uptrace/bun" ) func (f *filterDB) GetFilterStatusByID(ctx context.Context, id string) (*gtsmodel.FilterStatus, error) { - filterStatus, err := f.state.Caches.DB.FilterStatus.LoadOne( + return f.state.Caches.DB.FilterStatus.LoadOne( "ID", func() (*gtsmodel.FilterStatus, error) { var filterStatus gtsmodel.FilterStatus - err := f.db. + if err := f.db. NewSelect(). Model(&filterStatus). Where("? = ?", bun.Ident("id"), id). - Scan(ctx) - return &filterStatus, err + Scan(ctx); err != nil { + return nil, err + } + return &filterStatus, nil }, id, ) - if err != nil { - return nil, err - } - - if !gtscontext.Barebones(ctx) { - err = f.populateFilterStatus(ctx, filterStatus) - if err != nil { - return nil, err - } - } - - return filterStatus, nil -} - -func (f *filterDB) populateFilterStatus(ctx context.Context, filterStatus *gtsmodel.FilterStatus) error { - if filterStatus.Filter == nil { - // Filter is not set, fetch from the cache or database. - filter, err := f.state.DB.GetFilterByID( - // Don't populate the filter with all of its keywords and statuses or we'll just end up back here. - gtscontext.SetBarebones(ctx), - filterStatus.FilterID, - ) - if err != nil { - return err - } - filterStatus.Filter = filter - } - - return nil } -func (f *filterDB) GetFilterStatusesForFilterID(ctx context.Context, filterID string) ([]*gtsmodel.FilterStatus, error) { - return f.getFilterStatuses(ctx, "filter_id", filterID) -} - -func (f *filterDB) GetFilterStatusesForAccountID(ctx context.Context, accountID string) ([]*gtsmodel.FilterStatus, error) { - return f.getFilterStatuses(ctx, "account_id", accountID) -} - -func (f *filterDB) getFilterStatuses(ctx context.Context, idColumn string, id string) ([]*gtsmodel.FilterStatus, error) { - var filterStatusIDs []string - if err := f.db. - NewSelect(). - Model((*gtsmodel.FilterStatus)(nil)). - Column("id"). - Where("? = ?", bun.Ident(idColumn), id). - Scan(ctx, &filterStatusIDs); err != nil { - return nil, err - } - if len(filterStatusIDs) == 0 { +func (f *filterDB) GetFilterStatusesByIDs(ctx context.Context, ids []string) ([]*gtsmodel.FilterStatus, error) { + if len(ids) == 0 { return nil, nil } // Get each filter status by ID from the cache or DB. filterStatuses, err := f.state.Caches.DB.FilterStatus.LoadIDs("ID", - filterStatusIDs, + ids, func(uncached []string) ([]*gtsmodel.FilterStatus, error) { filterStatuses := make([]*gtsmodel.FilterStatus, 0, len(uncached)) if err := f.db. @@ -116,29 +68,11 @@ func (f *filterDB) getFilterStatuses(ctx context.Context, idColumn string, id st } // Put the filter status structs in the same order as the filter status IDs. - xslices.OrderBy(filterStatuses, filterStatusIDs, func(filterStatus *gtsmodel.FilterStatus) string { + xslices.OrderBy(filterStatuses, ids, func(filterStatus *gtsmodel.FilterStatus) string { return filterStatus.ID }) - if gtscontext.Barebones(ctx) { - return filterStatuses, nil - } - - // Populate the filter statuses. Remove any that we can't populate from the return slice. - errs := gtserror.NewMultiError(len(filterStatuses)) - filterStatuses = slices.DeleteFunc(filterStatuses, func(filterStatus *gtsmodel.FilterStatus) bool { - if err := f.populateFilterStatus(ctx, filterStatus); err != nil { - errs.Appendf( - "error populating filter status %s: %w", - filterStatus.ID, - err, - ) - return true - } - return false - }) - - return filterStatuses, errs.Combine() + return filterStatuses, nil } func (f *filterDB) PutFilterStatus(ctx context.Context, filterStatus *gtsmodel.FilterStatus) error { @@ -152,11 +86,6 @@ func (f *filterDB) PutFilterStatus(ctx context.Context, filterStatus *gtsmodel.F } func (f *filterDB) UpdateFilterStatus(ctx context.Context, filterStatus *gtsmodel.FilterStatus, columns ...string) error { - filterStatus.UpdatedAt = time.Now() - if len(columns) > 0 { - columns = append(columns, "updated_at") - } - return f.state.Caches.DB.FilterStatus.Store(filterStatus, func() error { _, err := f.db. NewUpdate(). @@ -168,16 +97,14 @@ func (f *filterDB) UpdateFilterStatus(ctx context.Context, filterStatus *gtsmode }) } -func (f *filterDB) DeleteFilterStatusByID(ctx context.Context, id string) error { +func (f *filterDB) DeleteFilterStatusesByIDs(ctx context.Context, ids ...string) error { if _, err := f.db. NewDelete(). Model((*gtsmodel.FilterStatus)(nil)). - Where("? = ?", bun.Ident("id"), id). + Where("? IN (?)", bun.Ident("id"), bun.In(ids)). Exec(ctx); err != nil { return err } - - f.state.Caches.DB.FilterStatus.Invalidate("ID", id) - + f.state.Caches.DB.FilterStatus.InvalidateIDs("ID", ids) return nil } |
