mirrors/gotosocial
1
0
Fork 0
mirror of https://github.com/superseriousbusiness/gotosocial.git synced 2025-11-28 01:43:33 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

[feature] add streaming of statuses and status updates to LOCAL / PUBLIC timelines (#4353)

Browse source 
This adds streaming of statuses and edits to LOCAL and PUBLIC timeline types. Note that in the future we should probably rearrange some of the surface code so we don't perform so many repeated mute and visibility checks on the same status in sequence.

closes https://codeberg.org/superseriousbusiness/gotosocial/issues/4342

Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4353
Co-authored-by: kim <grufwub@gmail.com>
Co-committed-by: kim <grufwub@gmail.com>
This commit is contained in:
kim 2025-07-30 17:54:07 +02:00 • committed by kim
commit 7e45168d33
5 changed files with 275 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

74
internal/db/bundb/user.go
View file

@ -19,12 +19,15 @@ 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/log"
"code.superseriousbusiness.org/gotosocial/internal/state"
"code.superseriousbusiness.org/gotosocial/internal/util/xslices"
"github.com/uptrace/bun"
)
@ -45,27 +48,47 @@ func (u *userDB) GetUserByID(ctx context.Context, id string) (*gtsmodel.User, er
}
func (u *userDB) GetUsersByIDs(ctx context.Context, ids []string) ([]*gtsmodel.User, error) {
var (
users = make([]*gtsmodel.User, 0, len(ids))
// Load all input user IDs via cache loader callback.
users, err := u.state.Caches.DB.User.LoadIDs("ID",
ids,
func(uncached []string) ([]*gtsmodel.User, error) {
// Preallocate expected length of uncached users.
users := make([]*gtsmodel.User, 0, len(uncached))
// Collect errors instead of
// returning early on any.
errs gtserror.MultiError
// Perform database query scanning
// the remaining (uncached) user IDs.
if err := u.db.NewSelect().
Model(&users).
Where("? IN (?)", bun.Ident("id"), bun.In(uncached)).
Scan(ctx); err != nil {
return nil, err
}
return users, nil
},
)
for _, id := range ids {
// Attempt to fetch user from DB.
user, err := u.GetUserByID(ctx, id)
if err != nil {
errs.Appendf("error getting user %s: %w", id, err)
continue
}
// Append user to return slice.
users = append(users, user)
if err != nil {
return nil, err
}
return users, errs.Combine()
// Reorder the users by their
// IDs to ensure in correct order.
getID := func(s *gtsmodel.User) string { return s.ID }
xslices.OrderBy(users, ids, getID)
if gtscontext.Barebones(ctx) {
// no need to fully populate.
return users, nil
}
// Populate all loaded users.
for _, user := range users {
if err := u.PopulateUser(ctx, user); err != nil {
log.Errorf(ctx, "error populating user %s: %v", user.ID, err)
}
}
return users, nil
}
func (u *userDB) GetUserByAccountID(ctx context.Context, accountID string) (*gtsmodel.User, error) {
@ -161,7 +184,11 @@ func (u *userDB) PopulateUser(ctx context.Context, user *gtsmodel.User) error {
return errs.Combine()
}
func (u *userDB) GetAllUsers(ctx context.Context) ([]*gtsmodel.User, error) {
func (u *userDB) GetAllUserIDs(ctx context.Context) ([]string, error) {
if p := u.state.Caches.DB.LocalInstance.UserIDs.Load(); p != nil {
return slices.Clone(*p), nil
}
var userIDs []string
// Scan all user IDs into slice.
@ -172,7 +199,16 @@ func (u *userDB) GetAllUsers(ctx context.Context) ([]*gtsmodel.User, error) {
return nil, err
}
// Transform user IDs into user slice.
// Store the scanned user IDs in our local cache ptr.
u.state.Caches.DB.LocalInstance.UserIDs.Store(&userIDs)
return userIDs, nil
}
func (u *userDB) GetAllUsers(ctx context.Context) ([]*gtsmodel.User, error) {
userIDs, err := u.GetAllUserIDs(ctx)
if err != nil {
return nil, err
}
return u.GetUsersByIDs(ctx, userIDs)
}