Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Several callbacks can be added to a same message type and disposed (#197) #199

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Several callbacks can be added to a same message type and disposed (#197) #199

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
81 changes: 56 additions & 25 deletions Assets/Colyseus/Runtime/Scripts/Room/ColyseusRoom.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -282,18 +282,53 @@ public async Task Send(string type, object message)
await colyseusConnection.Send(bytes);
}

/// <summary>
/// Disposable class to return with OnMessage function, so that a message callback can be removed.
/// </summary>
/// <typeparam name="MessageType">The type of object the message responds with</typeparam>
private class MessageRemover<MessageType> : IDisposable
{
private string _type;
private Action<MessageType> _targetHandler;
private Dictionary<string, IColyseusMessageHandler> _onMessageHandlers;

public MessageRemover(string type,
Action<MessageType> targetHandler,
Dictionary<string, IColyseusMessageHandler> onMessageHandlers)
{
_type = type;
_targetHandler = targetHandler;
_onMessageHandlers = onMessageHandlers;
}

public void Dispose()
{
var messageHandler = _onMessageHandlers[_type] as ColyseusMessageHandler<MessageType>;
messageHandler.Action -= _targetHandler;
if (messageHandler.Action == null) {
_onMessageHandlers.Remove(_type);
}
}
}

/// <summary>
/// Method to add new message handlers to the room
/// </summary>
/// <param name="type">The type of message received</param>
/// <param name="handler"></param>
/// <typeparam name="MessageType">The type of object this message should respond with</typeparam>
public void OnMessage<MessageType>(string type, Action<MessageType> handler)
/// <returns>IDisposable object to remove the message handler when called</returns>
public IDisposable OnMessage<MessageType>(string type, Action<MessageType> handler)
{
OnMessageHandlers.Add(type, new ColyseusMessageHandler<MessageType>
{
Action = handler
});
if (OnMessageHandlers.ContainsKey(type)) {
(OnMessageHandlers[type] as ColyseusMessageHandler<MessageType>).Action += handler;
} else {
OnMessageHandlers.Add(type, new ColyseusMessageHandler<MessageType>
{
Action = handler
});
}
return new MessageRemover<MessageType>(type, handler, OnMessageHandlers);
}

/// <summary>
Expand All @@ -302,25 +337,21 @@ public void OnMessage<MessageType>(string type, Action<MessageType> handler)
/// <param name="type">The type of message received</param>
/// <param name="handler"></param>
/// <typeparam name="MessageType">The type of object this message should respond with</typeparam>
public void OnMessage<MessageType>(byte type, Action<MessageType> handler)
/// <returns>IDisposable object to remove the message handler when called</returns>
public IDisposable OnMessage<MessageType>(byte type, Action<MessageType> handler)
{
OnMessageHandlers.Add("i" + type, new ColyseusMessageHandler<MessageType>
{
Action = handler
});
return this.OnMessage<MessageType>("i" + type, handler);
}

/// <summary>
/// Method to add new message handlers to the room
/// </summary>
/// <param name="handler"></param>
/// <typeparam name="MessageType">The type of object this message should respond with</typeparam>
public void OnMessage<MessageType>(Action<MessageType> handler) where MessageType : Schema.Schema, new()
/// <returns>IDisposable object to remove the message handler when called</returns>
public IDisposable OnMessage<MessageType>(Action<MessageType> handler) where MessageType : Schema.Schema, new()
{
OnMessageHandlers.Add("s" + typeof(MessageType), new ColyseusMessageHandler<MessageType>
{
Action = handler
});
return this.OnMessage<MessageType>("s" + typeof(MessageType), handler);
}

/// <summary>
Expand Down Expand Up @@ -372,15 +403,15 @@ protected async void ParseMessage(byte[] bytes)

if (bytes.Length > offset)
{
try {
serializer.Handshake(bytes, offset);
}
catch (Exception e)
{
await Leave(false);
OnError?.Invoke(ColyseusErrorCode.SCHEMA_MISMATCH, e.Message);
return;
}
try {
serializer.Handshake(bytes, offset);
}
catch (Exception e)
{
await Leave(false);
OnError?.Invoke(ColyseusErrorCode.SCHEMA_MISMATCH, e.Message);
return;
}
}

OnJoin?.Invoke();
Expand Down Expand Up @@ -423,7 +454,7 @@ protected async void ParseMessage(byte[] bytes)
}
else if (code == ColyseusProtocol.ROOM_STATE)
{
SetState(bytes, 1);
SetState(bytes, 1);
}
else if (code == ColyseusProtocol.ROOM_STATE_PATCH)
{
Expand Down