ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-02-16 18:47:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

make the message processing truely async (#1596)

Browse Source
This commit is contained in:
Alan Ren
2022-07-21 16:58:03 -07:00
committed by GitHub
parent fc2d1300fd
commit 08c74be51f
2 changed files with 44 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/Microsoft.SqlTools.Hosting/Hosting/Protocol/MessageDispatcher.cs
View File

@@ -123,6 +123,8 @@ namespace Microsoft.SqlTools.Hosting.Protocol
this.requestHandlers.Add( this.requestHandlers.Add(
requestType.MethodName, requestType.MethodName,
(requestMessage, messageWriter) => (requestMessage, messageWriter) =>
{
return Task.Run(async () =>
{ {
var requestContext = var requestContext =
new RequestContext<TResult>( new RequestContext<TResult>(
@@ -138,11 +140,12 @@ namespace Microsoft.SqlTools.Hosting.Protocol
} }
catch (Exception ex) catch (Exception ex)
{ {
return requestContext.SendError(ex.Message); await requestContext.SendError(ex.Message);
} }
} }
return requestHandler(typedParams, requestContext); await requestHandler(typedParams, requestContext);
});
}); });
} }
@@ -170,6 +173,8 @@ namespace Microsoft.SqlTools.Hosting.Protocol
this.eventHandlers.Add( this.eventHandlers.Add(
eventType.MethodName, eventType.MethodName,
(eventMessage, messageWriter) => (eventMessage, messageWriter) =>
{
return Task.Run(async () =>
{ {
var eventContext = new EventContext(messageWriter); var eventContext = new EventContext(messageWriter);
@@ -186,7 +191,8 @@ namespace Microsoft.SqlTools.Hosting.Protocol
} }
} }
return eventHandler(typedParams, eventContext); await eventHandler(typedParams, eventContext);
});
}); });
} }
@@ -317,21 +323,23 @@ namespace Microsoft.SqlTools.Hosting.Protocol
// thread is not blocked. // thread is not blocked.
_ = Task.Run(() => _ = Task.Run(() =>
{ {
_ = RunTask(handlerToAwait); _ = RunTask(messageToDispatch, handlerToAwait);
}); });
} }
else else
{ {
await RunTask(handlerToAwait); await RunTask(messageToDispatch, handlerToAwait);
} }
} }
} }
private async Task RunTask(Task task) private async Task RunTask(Message message, Task task)
{ {
try try
{ {
Logger.Write(TraceEventType.Verbose, $"Processing message with id[{message.Id}], of type[{message.MessageType}] and method[{message.Method}]");
await task; await task;
Logger.Write(TraceEventType.Verbose, $"Finished processing message with id[{message.Id}], of type[{message.MessageType}] and method[{message.Method}]");
} }
catch (TaskCanceledException) catch (TaskCanceledException)
{ {

4
src/Microsoft.SqlTools.ServiceLayer/Agent/AgentService.cs
View File

@@ -1211,7 +1211,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Agent
parameters.OwnerUri, parameters.OwnerUri,
out connInfo); out connInfo);
result.Success = true; result.Success = true;
result.Notebooks = AgentNotebookHelper.GetAgentNotebooks(connInfo).Result; result.Notebooks = await AgentNotebookHelper.GetAgentNotebooks(connInfo);
} }
catch (Exception e) catch (Exception e)
{ {
@@ -1260,7 +1260,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Agent
ConnectionServiceInstance.TryFindConnection( ConnectionServiceInstance.TryFindConnection(
parameters.OwnerUri, parameters.OwnerUri,
out connInfo); out connInfo);
result.NotebookMaterialized = AgentNotebookHelper.GetMaterializedNotebook(connInfo, parameters.NotebookMaterializedId, parameters.TargetDatabase).Result; result.NotebookMaterialized = await AgentNotebookHelper.GetMaterializedNotebook(connInfo, parameters.NotebookMaterializedId, parameters.TargetDatabase);
result.Success = true; result.Success = true;
} }
catch (Exception e) catch (Exception e)