RPCClientPipeTransport.cs 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. using System;
  2. using System.Threading;
  3. using H.Pipes;
  4. using InABox.Core;
  5. namespace InABox.Rpc
  6. {
  7. public class RpcClientPipeTransport : RpcClientTransport, IDisposable
  8. {
  9. private PipeClient<RpcMessage> _pipe;
  10. private string _name;
  11. public RpcClientPipeTransport(string name)
  12. {
  13. _name = name;
  14. _pipe = new PipeClient<RpcMessage>(_name);
  15. _pipe.Connected += PipeConnected;
  16. _pipe.Disconnected += PipeDisconnected;
  17. _pipe.MessageReceived += PipeMessageReceived;
  18. _pipe.ExceptionOccurred += PipeExceptionOccurred;
  19. }
  20. public override bool Connect(CancellationToken ct = default)
  21. {
  22. _pipe.ConnectAsync(ct).Wait();
  23. return _pipe.IsConnected;
  24. }
  25. public override bool IsConnected() => _pipe?.IsConnected == true;
  26. public override bool IsSecure() => false;
  27. public override string? ServerName() => _name;
  28. public override void Disconnect()
  29. {
  30. _pipe.DisconnectAsync();
  31. }
  32. public override void Send(RpcMessage message)
  33. {
  34. _pipe.WriteAsync(message);
  35. }
  36. private void PipeConnected(object? sender, H.Pipes.Args.ConnectionEventArgs<RpcMessage> e)
  37. {
  38. DoOpen();
  39. }
  40. private void PipeDisconnected(object? sender, H.Pipes.Args.ConnectionEventArgs<RpcMessage> e)
  41. {
  42. DoClose(RpcTransportCloseEventType.Closed);
  43. }
  44. private void PipeExceptionOccurred(object? sender, H.Pipes.Args.ExceptionEventArgs e)
  45. {
  46. Logger.Send(LogType.Error, "", $"Exception occured: {e.Exception.Message}");
  47. }
  48. private void PipeMessageReceived(object? sender, H.Pipes.Args.ConnectionMessageEventArgs<RpcMessage?> e)
  49. {
  50. Accept(e.Message);
  51. }
  52. public void Dispose()
  53. {
  54. _pipe.DisposeAsync().AsTask().Wait();
  55. }
  56. ~RpcClientPipeTransport()
  57. {
  58. Dispose();
  59. }
  60. protected override RpcClientTransport Clone() => new RpcClientPipeTransport(_name);
  61. }
  62. }