🐛 Fix websocket gateway, finally

This commit is contained in:
2025-09-22 01:33:30 +08:00
parent d5a5721402
commit 5b31357fe9
5 changed files with 35 additions and 30 deletions

22
Caddyfile Normal file
View File

@@ -0,0 +1,22 @@
:5002 {
@cors_preflight method OPTIONS
header {
Access-Control-Allow-Origin "{header.origin}"
Vary Origin
Access-Control-Expose-Headers "*"
+Access-Control-Allow-Headers "*"
+Access-Control-Allow-Headers "Authorization,Content-Type,Accept,User-Agent"
Access-Control-Allow-Credentials "true"
}
handle @cors_preflight {
header {
Access-Control-Allow-Methods "GET, POST, PUT, PATCH, DELETE"
Access-Control-Max-Age "3600"
}
respond "" 204
}
reverse_proxy host.docker.internal:5001
}

View File

@@ -149,7 +149,7 @@ public class WebSocketService
AccountId = Guid.Parse(currentUser.Id), AccountId = Guid.Parse(currentUser.Id),
DeviceId = deviceId, DeviceId = deviceId,
PacketBytes = packet.ToBytes() PacketBytes = packet.ToBytes()
}).ToBase64() }).ToByteArray()
); );
return; return;
} }

View File

@@ -25,12 +25,8 @@ public class RingServiceGrpc(
public override Task<Empty> PushWebSocketPacket(PushWebSocketPacketRequest request, ServerCallContext context) public override Task<Empty> PushWebSocketPacket(PushWebSocketPacketRequest request, ServerCallContext context)
{ {
var packet = new Shared.Data.WebSocketPacket var packet = Shared.Data.WebSocketPacket.FromProtoValue(request.Packet);
{
Type = request.Packet.Type,
Data = GrpcTypeHelper.ConvertByteStringToObject<Dictionary<string, object?>>(request.Packet.Data),
ErrorMessage = request.Packet.ErrorMessage
};
WebSocketService.SendPacketToAccount(Guid.Parse(request.UserId), packet); WebSocketService.SendPacketToAccount(Guid.Parse(request.UserId), packet);
return Task.FromResult(new Empty()); return Task.FromResult(new Empty());
} }
@@ -38,12 +34,7 @@ public class RingServiceGrpc(
public override Task<Empty> PushWebSocketPacketToUsers(PushWebSocketPacketToUsersRequest request, public override Task<Empty> PushWebSocketPacketToUsers(PushWebSocketPacketToUsersRequest request,
ServerCallContext context) ServerCallContext context)
{ {
var packet = new Shared.Data.WebSocketPacket var packet = Shared.Data.WebSocketPacket.FromProtoValue(request.Packet);
{
Type = request.Packet.Type,
Data = GrpcTypeHelper.ConvertByteStringToObject<Dictionary<string, object?>>(request.Packet.Data),
ErrorMessage = request.Packet.ErrorMessage
};
foreach (var accountId in request.UserIds) foreach (var accountId in request.UserIds)
WebSocketService.SendPacketToAccount(Guid.Parse(accountId), packet); WebSocketService.SendPacketToAccount(Guid.Parse(accountId), packet);
@@ -54,12 +45,8 @@ public class RingServiceGrpc(
public override Task<Empty> PushWebSocketPacketToDevice(PushWebSocketPacketToDeviceRequest request, public override Task<Empty> PushWebSocketPacketToDevice(PushWebSocketPacketToDeviceRequest request,
ServerCallContext context) ServerCallContext context)
{ {
var packet = new Shared.Data.WebSocketPacket var packet = Shared.Data.WebSocketPacket.FromProtoValue(request.Packet);
{
Type = request.Packet.Type,
Data = GrpcTypeHelper.ConvertByteStringToObject<Dictionary<string, object?>>(request.Packet.Data),
ErrorMessage = request.Packet.ErrorMessage
};
websocket.SendPacketToDevice(request.DeviceId, packet); websocket.SendPacketToDevice(request.DeviceId, packet);
return Task.FromResult(new Empty()); return Task.FromResult(new Empty());
} }
@@ -67,17 +54,10 @@ public override Task<Empty> PushWebSocketPacketToDevice(PushWebSocketPacketToDev
public override Task<Empty> PushWebSocketPacketToDevices(PushWebSocketPacketToDevicesRequest request, public override Task<Empty> PushWebSocketPacketToDevices(PushWebSocketPacketToDevicesRequest request,
ServerCallContext context) ServerCallContext context)
{ {
var packet = new Shared.Data.WebSocketPacket var packet = Shared.Data.WebSocketPacket.FromProtoValue(request.Packet);
{
Type = request.Packet.Type,
Data = GrpcTypeHelper.ConvertByteStringToObject<Dictionary<string, object?>>(request.Packet.Data),
ErrorMessage = request.Packet.ErrorMessage
};
foreach (var deviceId in request.DeviceIds) foreach (var deviceId in request.DeviceIds)
{
websocket.SendPacketToDevice(deviceId, packet); websocket.SendPacketToDevice(deviceId, packet);
}
return Task.FromResult(new Empty()); return Task.FromResult(new Empty());
} }

View File

@@ -4,7 +4,7 @@ public class WebSocketPacketEvent
{ {
public static string Type => "websocket_msg"; public static string Type => "websocket_msg";
public static string SubjectPrefix = "websocket_"; public const string SubjectPrefix = "websocket_";
public Guid AccountId { get; set; } public Guid AccountId { get; set; }
public string DeviceId { get; set; } = null!; public string DeviceId { get; set; } = null!;

View File

@@ -176,11 +176,14 @@ public class BroadcastEventHandler(
await foreach (var msg in nats.SubscribeAsync<byte[]>( await foreach (var msg in nats.SubscribeAsync<byte[]>(
WebSocketPacketEvent.SubjectPrefix + "sphere", cancellationToken: stoppingToken)) WebSocketPacketEvent.SubjectPrefix + "sphere", cancellationToken: stoppingToken))
{ {
logger.LogDebug("Handling websocket packet...");
try try
{ {
var evt = JsonSerializer.Deserialize<WebSocketPacketEvent>(msg.Data, GrpcTypeHelper.SerializerOptions); var evt = JsonSerializer.Deserialize<WebSocketPacketEvent>(msg.Data, GrpcTypeHelper.SerializerOptions);
if (evt == null) throw new ArgumentNullException(nameof(evt)); if (evt == null) throw new ArgumentNullException(nameof(evt));
var packet = WebSocketPacket.FromBytes(evt.PacketBytes); var packet = WebSocketPacket.FromBytes(evt.PacketBytes);
logger.LogInformation("Handling websocket packet... {Type}", packet.Type);
switch (packet.Type) switch (packet.Type)
{ {
case "messages.read": case "messages.read":
@@ -255,12 +258,12 @@ public class BroadcastEventHandler(
var responsePacket = new WebSocketPacket var responsePacket = new WebSocketPacket
{ {
Type = "messages.typing", Type = "messages.typing",
Data = GrpcTypeHelper.ConvertObjectToByteString(new Data = new
{ {
room_id = sender.ChatRoomId, room_id = sender.ChatRoomId,
sender_id = sender.Id, sender_id = sender.Id,
sender = sender sender = sender
}) }
}; };
// Broadcast typing indicator to other room members // Broadcast typing indicator to other room members