fix: sync drive TDLib username cache convergence

This commit is contained in:
iamxvbaba 2026-08-02 19:32:17 +08:00
parent f67e87689c
commit 8bc1560c16
2 changed files with 38 additions and 10 deletions

View file

@ -124,19 +124,26 @@ func TestDispatchPushesCompleteSelfProfileOnceWhenSessionBecomesReady(t *testing
got.receives, got.receivesCalls, got.sessionPushCalls) got.receives, got.receivesCalls, got.sessionPushCalls)
} }
updates, ok := got.message.(*tg.Updates) updates, ok := got.message.(*tg.Updates)
if !ok || len(updates.Updates) != 1 || len(updates.Users) != 1 { if !ok || len(updates.Updates) != 2 || len(updates.Users) != 1 {
t.Fatalf("self refresh = %T %+v, want one update and one user", got.message, got.message) t.Fatalf("self refresh = %T %+v, want two updates and one user", got.message, got.message)
} }
refresh, ok := updates.Updates[0].(*tg.UpdateUser) nameRefresh, ok := updates.Updates[0].(*tg.UpdateUserName)
if !ok || nameRefresh.UserID != userID || nameRefresh.FirstName != "Alice" || nameRefresh.LastName != "" {
t.Fatalf("self name refresh = %T %+v, want updateUserName(%d, Alice)", updates.Updates[0], updates.Updates[0], userID)
}
wantUsernames := []string{"Alice", "aliceCollect0728b", "aliceCollect0728a"}
if !reflect.DeepEqual(usernameStrings(nameRefresh.Usernames), wantUsernames) {
t.Fatalf("self updateUserName usernames = %v, want %v", usernameStrings(nameRefresh.Usernames), wantUsernames)
}
refresh, ok := updates.Updates[1].(*tg.UpdateUser)
if !ok || refresh.UserID != userID { if !ok || refresh.UserID != userID {
t.Fatalf("self refresh update = %T %+v, want updateUser(%d)", updates.Updates[0], updates.Updates[0], userID) t.Fatalf("self refresh update = %T %+v, want updateUser(%d)", updates.Updates[1], updates.Updates[1], userID)
} }
projected, ok := updates.Users[0].(*tg.User) projected, ok := updates.Users[0].(*tg.User)
if !ok { if !ok {
t.Fatalf("self refresh user = %T, want *tg.User", updates.Users[0]) t.Fatalf("self refresh user = %T, want *tg.User", updates.Users[0])
} }
vector, set := projected.GetUsernames() vector, set := projected.GetUsernames()
wantUsernames := []string{"Alice", "aliceCollect0728b", "aliceCollect0728a"}
if !set || !reflect.DeepEqual(usernameStrings(vector), wantUsernames) { if !set || !reflect.DeepEqual(usernameStrings(vector), wantUsernames) {
t.Fatalf("self refresh usernames = %v (set %v), want %v", usernameStrings(vector), set, wantUsernames) t.Fatalf("self refresh usernames = %v (set %v), want %v", usernameStrings(vector), set, wantUsernames)
} }
@ -156,9 +163,14 @@ func TestDispatchPushesCompleteSelfProfileOnceWhenSessionBecomesReady(t *testing
t.Fatalf("decode Layer 228 self refresh: %v", err) t.Fatalf("decode Layer 228 self refresh: %v", err)
} }
decodedUpdates, ok := decoded.(*tg.Updates) decodedUpdates, ok := decoded.(*tg.Updates)
if !ok || len(decodedUpdates.Users) != 1 { if !ok || len(decodedUpdates.Updates) != 2 || len(decodedUpdates.Users) != 1 {
t.Fatalf("decoded Layer 228 self refresh = %T %+v", decoded, decoded) t.Fatalf("decoded Layer 228 self refresh = %T %+v", decoded, decoded)
} }
decodedNameRefresh, ok := decodedUpdates.Updates[0].(*tg.UpdateUserName)
if !ok || !reflect.DeepEqual(usernameStrings(decodedNameRefresh.Usernames), wantUsernames) {
t.Fatalf("decoded Layer 228 updateUserName = %T usernames=%v, want %v",
decodedUpdates.Updates[0], usernameStrings(decodedNameRefresh.Usernames), wantUsernames)
}
decodedUser := decodedUpdates.Users[0].(*tg.User) decodedUser := decodedUpdates.Users[0].(*tg.User)
decodedVector, decodedSet := decodedUser.GetUsernames() decodedVector, decodedSet := decodedUser.GetUsernames()
if !decodedSet || !reflect.DeepEqual(usernameStrings(decodedVector), wantUsernames) { if !decodedSet || !reflect.DeepEqual(usernameStrings(decodedVector), wantUsernames) {

View file

@ -75,10 +75,26 @@ func (r *Router) updatesReadySelfProfile(ctx context.Context, userID int64) (*tg
r.applyStoryMaxIDsToPeerObjects(ctx, userID, users, nil) r.applyStoryMaxIDsToPeerObjects(ctx, userID, users, nil)
r.applyBotVerificationIconsToPeerObjects(ctx, users, nil) r.applyBotVerificationIconsToPeerObjects(ctx, users, nil)
usernames := tgUsernames(u.Username)
if vector, ok := self.GetUsernames(); ok && len(vector) != 0 {
usernames = vector
}
return &tg.Updates{ return &tg.Updates{
Updates: []tg.UpdateClass{&tg.UpdateUser{UserID: userID}}, Updates: []tg.UpdateClass{
Users: users, // updateUserName is the authoritative basic-name/username cache write.
Date: int(r.clock.Now().Unix()), // TDLib handles it by calling on_update_user_usernames directly.
Seq: 0, &tg.UpdateUserName{
UserID: userID,
FirstName: u.FirstName,
LastName: u.LastName,
Usernames: usernames,
},
// updateUser additionally invalidates userFull while the complete basic
// User remains bundled in the outer users vector.
&tg.UpdateUser{UserID: userID},
},
Users: users,
Date: int(r.clock.Now().Unix()),
Seq: 0,
}, nil }, nil
} }