♻️ Better event broadcast status changes

This commit is contained in:
2026-01-02 15:08:02 +08:00
parent c588b6f234
commit eeb583d78d
2 changed files with 100 additions and 54 deletions

View File

@@ -25,9 +25,10 @@ public class AccountEventService(
{
private static readonly Random Random = new();
private const string StatusCacheKey = "account:status:";
private const string PreviousStatusCacheKey = "account:status:prev:";
private const string ActivityCacheKey = "account:activities:";
public async Task<bool> GetAccountIsConnected(Guid userId)
private async Task<bool> GetAccountIsConnected(Guid userId)
{
var resp = await pusher.GetWebsocketConnectionStatusAsync(
new GetWebsocketConnectionStatusRequest { UserId = userId.ToString() }
@@ -49,6 +50,8 @@ public class AccountEventService(
{
var cacheKey = $"{StatusCacheKey}{userId}";
cache.RemoveAsync(cacheKey);
var prevCacheKey = $"{PreviousStatusCacheKey}{userId}";
cache.RemoveAsync(prevCacheKey);
}
public void PurgeActivityCache(Guid userId)
@@ -70,18 +73,30 @@ public class AccountEventService(
);
}
private static bool StatusesEqual(SnAccountStatus a, SnAccountStatus b)
{
return a.Attitude == b.Attitude &&
a.IsOnline == b.IsOnline &&
a.IsCustomized == b.IsCustomized &&
a.Label == b.Label &&
a.IsInvisible == b.IsInvisible &&
a.IsNotDisturb == b.IsNotDisturb;
}
public async Task<SnAccountStatus> GetStatus(Guid userId)
{
var cacheKey = $"{StatusCacheKey}{userId}";
var cachedStatus = await cache.GetAsync<SnAccountStatus>(cacheKey);
SnAccountStatus status;
if (cachedStatus is not null)
{
cachedStatus!.IsOnline = !cachedStatus.IsInvisible && await GetAccountIsConnected(userId);
return cachedStatus;
status = cachedStatus;
}
else
{
var now = SystemClock.Instance.GetCurrentInstant();
var status = await db.AccountStatuses
status = await db.AccountStatuses
.Where(e => e.AccountId == userId)
.Where(e => e.ClearedAt == null || e.ClearedAt > now)
.OrderByDescending(e => e.CreatedAt)
@@ -92,12 +107,12 @@ public class AccountEventService(
status.IsOnline = !status.IsInvisible && isOnline;
await cache.SetWithGroupsAsync(cacheKey, status, [$"{AccountService.AccountCachePrefix}{status.AccountId}"],
TimeSpan.FromMinutes(5));
return status;
}
else
{
if (isOnline)
{
return new SnAccountStatus
status = new SnAccountStatus
{
Attitude = Shared.Models.StatusAttitude.Neutral,
IsOnline = true,
@@ -106,8 +121,9 @@ public class AccountEventService(
AccountId = userId,
};
}
return new SnAccountStatus
else
{
status = new SnAccountStatus
{
Attitude = Shared.Models.StatusAttitude.Neutral,
IsOnline = false,
@@ -116,6 +132,13 @@ public class AccountEventService(
AccountId = userId,
};
}
}
}
await cache.SetAsync($"{PreviousStatusCacheKey}{userId}", status, TimeSpan.FromMinutes(5));
return status;
}
public async Task<Dictionary<Guid, SnAccountStatus>> GetStatuses(List<Guid> userIds)
{

View File

@@ -1,6 +1,7 @@
using System.Text.Json;
using DysonNetwork.Pass.Account;
using DysonNetwork.Pass.Wallet;
using DysonNetwork.Shared.Cache;
using DysonNetwork.Shared.Models;
using DysonNetwork.Shared.Proto;
using DysonNetwork.Shared.Queue;
@@ -18,6 +19,16 @@ public class BroadcastEventHandler(
IServiceProvider serviceProvider
) : BackgroundService
{
private static bool StatusesEqual(SnAccountStatus a, SnAccountStatus b)
{
return a.Attitude == b.Attitude &&
a.IsOnline == b.IsOnline &&
a.IsCustomized == b.IsCustomized &&
a.Label == b.Label &&
a.IsInvisible == b.IsInvisible &&
a.IsNotDisturb == b.IsNotDisturb;
}
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
var paymentTask = HandlePaymentEventsAsync(stoppingToken);
@@ -159,15 +170,20 @@ public class BroadcastEventHandler(
{
var evt =
GrpcTypeHelper.ConvertByteStringToObject<WebSocketConnectedEvent>(ByteString.CopyFrom(msg.Data));
if (evt is null) continue;
logger.LogInformation("Received WebSocket connected event for user {AccountId}, device {DeviceId}",
evt.AccountId, evt.DeviceId);
await using var scope = serviceProvider.CreateAsyncScope();
var accountEventService = scope.ServiceProvider.GetRequiredService<AccountEventService>();
var cache = scope.ServiceProvider.GetRequiredService<ICacheService>();
var previous = await cache.GetAsync<SnAccountStatus>($"account:status:prev:{evt.AccountId}");
var status = await accountEventService.GetStatus(evt.AccountId);
if (previous != null && !StatusesEqual(previous, status))
{
await nats.PublishAsync(
AccountStatusUpdatedEvent.Type,
ByteString.CopyFromUtf8(JsonSerializer.Serialize(new AccountStatusUpdatedEvent
@@ -175,10 +191,12 @@ public class BroadcastEventHandler(
AccountId = evt.AccountId,
Status = status,
UpdatedAt = SystemClock.Instance.GetCurrentInstant()
}, GrpcTypeHelper.SerializerOptionsWithoutIgnore)).ToByteArray()
}, GrpcTypeHelper.SerializerOptionsWithoutIgnore)).ToByteArray(),
cancellationToken: stoppingToken
);
}
logger.LogInformation("Broadcasted status update for user {AccountId}", evt.AccountId);
logger.LogInformation("Handled status update for user {AccountId} on connect", evt.AccountId);
}
catch (Exception ex)
{
@@ -204,9 +222,13 @@ public class BroadcastEventHandler(
await using var scope = serviceProvider.CreateAsyncScope();
var accountEventService = scope.ServiceProvider.GetRequiredService<AccountEventService>();
var cache = scope.ServiceProvider.GetRequiredService<ICacheService>();
var previous = await cache.GetAsync<SnAccountStatus>($"account:status:prev:{evt.AccountId}");
var status = await accountEventService.GetStatus(evt.AccountId);
if (previous != null && !StatusesEqual(previous, status))
{
await nats.PublishAsync(
AccountStatusUpdatedEvent.Type,
ByteString.CopyFromUtf8(JsonSerializer.Serialize(new AccountStatusUpdatedEvent
@@ -216,8 +238,9 @@ public class BroadcastEventHandler(
UpdatedAt = SystemClock.Instance.GetCurrentInstant()
}, GrpcTypeHelper.SerializerOptionsWithoutIgnore)).ToByteArray()
);
}
logger.LogInformation("Broadcasted status update for user {AccountId}", evt.AccountId);
logger.LogInformation("Handled status update for user {AccountId} on disconnect", evt.AccountId);
}
catch (Exception ex)
{