1
0
mirror of https://github.com/JKorf/CryptoExchange.Net synced 2025-06-13 02:46:20 +00:00

added lock for events, fixed some forgotten configureawait(false) calls

This commit is contained in:
Jan Korf 2018-12-28 21:44:38 +01:00
parent 36169d08a3
commit b423d4b6c7
2 changed files with 50 additions and 29 deletions

View File

@ -26,6 +26,7 @@ namespace CryptoExchange.Net.Sockets
private bool lostTriggered; private bool lostTriggered;
private readonly List<SocketEvent> waitingForEvents; private readonly List<SocketEvent> waitingForEvents;
private object eventLock = new object();
public SocketSubscription(IWebsocket socket) public SocketSubscription(IWebsocket socket)
@ -44,8 +45,11 @@ namespace CryptoExchange.Net.Sockets
Socket.DisconnectTime = DateTime.UtcNow; Socket.DisconnectTime = DateTime.UtcNow;
lostTriggered = true; lostTriggered = true;
lock (eventLock)
{
foreach (var events in Events) foreach (var events in Events)
events.Reset(); events.Reset();
}
if (Socket.ShouldReconnect) if (Socket.ShouldReconnect)
ConnectionLost?.Invoke(); ConnectionLost?.Invoke();
@ -62,10 +66,13 @@ namespace CryptoExchange.Net.Sockets
public void AddEvent(string name) public void AddEvent(string name)
{ {
lock (eventLock)
Events.Add(new SocketEvent(name)); Events.Add(new SocketEvent(name));
} }
public void SetEventByName(string name, bool success, Error error) public void SetEventByName(string name, bool success, Error error)
{
lock (eventLock)
{ {
var waitingEvent = waitingForEvents.SingleOrDefault(e => e.Name == name); var waitingEvent = waitingForEvents.SingleOrDefault(e => e.Name == name);
if (waitingEvent != null) if (waitingEvent != null)
@ -74,8 +81,11 @@ namespace CryptoExchange.Net.Sockets
waitingForEvents.Remove(waitingEvent); waitingForEvents.Remove(waitingEvent);
} }
} }
}
public void SetEventById(string id, bool success, Error error) public void SetEventById(string id, bool success, Error error)
{
lock (eventLock)
{ {
var waitingEvent = waitingForEvents.SingleOrDefault(e => e.WaitingId == id); var waitingEvent = waitingForEvents.SingleOrDefault(e => e.WaitingId == id);
if (waitingEvent != null) if (waitingEvent != null)
@ -84,50 +94,61 @@ namespace CryptoExchange.Net.Sockets
waitingForEvents.Remove(waitingEvent); waitingForEvents.Remove(waitingEvent);
} }
} }
}
public SocketEvent GetWaitingEvent(string name) public SocketEvent GetWaitingEvent(string name)
{ {
lock (eventLock)
return waitingForEvents.SingleOrDefault(w => w.Name == name); return waitingForEvents.SingleOrDefault(w => w.Name == name);
} }
public Task<CallResult<bool>> WaitForEvent(string name, TimeSpan timeout) public Task<CallResult<bool>> WaitForEvent(string name, TimeSpan timeout)
{ {
lock (eventLock)
return WaitForEvent(name, (int)Math.Round(timeout.TotalMilliseconds, 0)); return WaitForEvent(name, (int)Math.Round(timeout.TotalMilliseconds, 0));
} }
public Task<CallResult<bool>> WaitForEvent(string name, int timeout) public Task<CallResult<bool>> WaitForEvent(string name, int timeout)
{
lock (eventLock)
{ {
var evnt = Events.Single(e => e.Name == name); var evnt = Events.Single(e => e.Name == name);
waitingForEvents.Add(evnt); waitingForEvents.Add(evnt);
return Task.Run(() => evnt.Wait(timeout)); return Task.Run(() => evnt.Wait(timeout));
} }
}
public Task<CallResult<bool>> WaitForEvent(string name, string id, TimeSpan timeout) public Task<CallResult<bool>> WaitForEvent(string name, string id, TimeSpan timeout)
{ {
lock (eventLock)
return WaitForEvent(name, id, (int)Math.Round(timeout.TotalMilliseconds, 0)); return WaitForEvent(name, id, (int)Math.Round(timeout.TotalMilliseconds, 0));
} }
public Task<CallResult<bool>> WaitForEvent(string name, string id, int timeout) public Task<CallResult<bool>> WaitForEvent(string name, string id, int timeout)
{
lock (eventLock)
{ {
var evnt = Events.Single(e => e.Name == name); var evnt = Events.Single(e => e.Name == name);
evnt.WaitingId = id; evnt.WaitingId = id;
waitingForEvents.Add(evnt); waitingForEvents.Add(evnt);
return Task.Run(() => evnt.Wait(timeout)); return Task.Run(() => evnt.Wait(timeout));
} }
}
public void ResetEvents() public void ResetEvents()
{ {
foreach (var waiting in new List<SocketEvent>(waitingForEvents)) lock (eventLock)
{
foreach (var waiting in waitingForEvents)
waiting.Set(false, new UnknownError("Connection reset")); waiting.Set(false, new UnknownError("Connection reset"));
waitingForEvents.Clear(); waitingForEvents.Clear();
} }
}
public async Task Close() public async Task Close()
{ {
Socket.ShouldReconnect = false; Socket.ShouldReconnect = false;
await Socket.Close(); await Socket.Close().ConfigureAwait(false);
Socket.Dispose(); Socket.Dispose();
} }
} }

View File

@ -41,7 +41,7 @@ namespace CryptoExchange.Net.Sockets
/// <returns></returns> /// <returns></returns>
public async Task Close() public async Task Close()
{ {
await subscription.Close(); await subscription.Close().ConfigureAwait(false);
} }
} }
} }