AMActorHandler.cs 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. using System;
  2. using System.Threading.Tasks;
  3. namespace ETModel
  4. {
  5. public abstract class AMActorHandler<E, Message>: IMActorHandler where E: Entity where Message : class
  6. {
  7. protected abstract Task Run(E entity, Message message);
  8. public async Task Handle(Session session, Entity entity, IActorMessage actorRequest)
  9. {
  10. Message msg = actorRequest as Message;
  11. if (msg == null)
  12. {
  13. Log.Error($"消息类型转换错误: {actorRequest.GetType().FullName} to {typeof (Message).Name}");
  14. return;
  15. }
  16. E e = entity as E;
  17. if (e == null)
  18. {
  19. Log.Error($"Actor类型转换错误: {entity.GetType().Name} to {typeof(E).Name}");
  20. return;
  21. }
  22. int rpcId = actorRequest.RpcId;
  23. ActorResponse response = new ActorResponse
  24. {
  25. RpcId = rpcId
  26. };
  27. session.Reply(response);
  28. await this.Run(e, msg);
  29. }
  30. public Type GetMessageType()
  31. {
  32. return typeof (Message);
  33. }
  34. }
  35. public abstract class AMActorRpcHandler<E, Request, Response>: IMActorHandler where E: Entity where Request: class, IActorRequest where Response : class, IActorResponse
  36. {
  37. protected static void ReplyError(Response response, Exception e, Action<Response> reply)
  38. {
  39. Log.Error(e);
  40. response.Error = ErrorCode.ERR_RpcFail;
  41. response.Message = e.ToString();
  42. reply(response);
  43. }
  44. protected abstract Task Run(E unit, Request message, Action<Response> reply);
  45. public async Task Handle(Session session, Entity entity, IActorMessage actorRequest)
  46. {
  47. try
  48. {
  49. Request request = actorRequest as Request;
  50. if (request == null)
  51. {
  52. Log.Error($"消息类型转换错误: {actorRequest.GetType().FullName} to {typeof (Request).Name}");
  53. return;
  54. }
  55. E e = entity as E;
  56. if (e == null)
  57. {
  58. Log.Error($"Actor类型转换错误: {entity.GetType().Name} to {typeof(E).Name}");
  59. return;
  60. }
  61. int rpcId = request.RpcId;
  62. await this.Run(e, request, response =>
  63. {
  64. // 等回调回来,session可以已经断开了
  65. if (session.IsDisposed)
  66. {
  67. return;
  68. }
  69. response.RpcId = rpcId;
  70. session.Reply(response);
  71. });
  72. }
  73. catch (Exception e)
  74. {
  75. throw new Exception($"解释消息失败: {actorRequest.GetType().FullName}", e);
  76. }
  77. }
  78. public Type GetMessageType()
  79. {
  80. return typeof (Request);
  81. }
  82. }
  83. }