NetClientComponentSystem.cs 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. using System.Net;
  2. using System.Net.Sockets;
  3. namespace ET.Client
  4. {
  5. [FriendOf(typeof(NetClientComponent))]
  6. public static class NetClientComponentSystem
  7. {
  8. [ObjectSystem]
  9. public class AwakeSystem: AwakeSystem<NetClientComponent, AddressFamily, NetworkProtocol>
  10. {
  11. protected override void Awake(NetClientComponent self, AddressFamily addressFamily, NetworkProtocol protocol)
  12. {
  13. switch (protocol)
  14. {
  15. case NetworkProtocol.KCP:
  16. {
  17. self.ServiceId = NetServices.Instance.AddService(new KService(addressFamily, ServiceType.Outer));
  18. break;
  19. }
  20. case NetworkProtocol.TCP:
  21. {
  22. self.ServiceId = NetServices.Instance.AddService(new TService(addressFamily, ServiceType.Outer));
  23. break;
  24. }
  25. }
  26. NetServices.Instance.RegisterReadCallback(self.ServiceId, self.OnRead);
  27. NetServices.Instance.RegisterErrorCallback(self.ServiceId, self.OnError);
  28. }
  29. }
  30. [ObjectSystem]
  31. public class DestroySystem: DestroySystem<NetClientComponent>
  32. {
  33. protected override void Destroy(NetClientComponent self)
  34. {
  35. NetServices.Instance.RemoveService(self.ServiceId);
  36. }
  37. }
  38. private static void OnRead(this NetClientComponent self, long channelId, long actorId, object message)
  39. {
  40. Session session = self.GetChild<Session>(channelId);
  41. if (session == null)
  42. {
  43. return;
  44. }
  45. session.LastRecvTime = TimeHelper.ClientNow();
  46. OpcodeHelper.LogMsg(self.DomainZone(), message);
  47. EventSystem.Instance.Publish(NetClientComponentOnRead.Clone(session, message));
  48. }
  49. private static void OnError(this NetClientComponent self, long channelId, int error)
  50. {
  51. Session session = self.GetChild<Session>(channelId);
  52. if (session == null)
  53. {
  54. return;
  55. }
  56. session.Error = error;
  57. session.Dispose();
  58. EventSystem.Instance.Publish<EventType.NetClosed>();
  59. }
  60. public static Session Create(this NetClientComponent self, IPEndPoint realIPEndPoint)
  61. {
  62. long channelId = NetServices.Instance.CreateConnectChannelId();
  63. Session session = self.AddChildWithId<Session, int>(channelId, self.ServiceId);
  64. session.RemoteAddress = realIPEndPoint;
  65. #if !UNITY
  66. if (self.DomainScene().SceneType != SceneType.Benchmark)
  67. {
  68. session.AddComponent<SessionIdleCheckerComponent>();
  69. }
  70. #endif
  71. NetServices.Instance.CreateChannel(self.ServiceId, session.Id, realIPEndPoint);
  72. return session;
  73. }
  74. public static Session Create(this NetClientComponent self, IPEndPoint routerIPEndPoint, IPEndPoint realIPEndPoint, uint localConn)
  75. {
  76. long channelId = localConn;
  77. Session session = self.AddChildWithId<Session, int>(channelId, self.ServiceId);
  78. session.RemoteAddress = realIPEndPoint;
  79. #if !UNITY
  80. if (self.DomainScene().SceneType != SceneType.Benchmark)
  81. {
  82. session.AddComponent<SessionIdleCheckerComponent>();
  83. }
  84. #endif
  85. NetServices.Instance.CreateChannel(self.ServiceId, session.Id, routerIPEndPoint);
  86. return session;
  87. }
  88. }
  89. }