70 lines
2.0 KiB
C#
70 lines
2.0 KiB
C#
using Fantasy;
|
|
using Fantasy.Async;
|
|
using Fantasy.Entitas;
|
|
using Fantasy.Network;
|
|
using Fantasy.Platform.Net;
|
|
using NB.Game;
|
|
|
|
namespace NB.Gate;
|
|
|
|
public static class GateUnitManageComponentSystem
|
|
{
|
|
public static GateUnit Add(this GateUnitManageComponent self, Session session, long accountId)
|
|
{
|
|
if (self.Units.TryGetValue(accountId, out var unit))
|
|
{
|
|
unit.Session = session;
|
|
return unit;
|
|
}
|
|
|
|
unit = Entity.Create<GateUnit>(self.Scene, accountId, true, true);
|
|
unit.AccountID = accountId;
|
|
unit.Session = session;
|
|
self.Units.Add(accountId, unit);
|
|
return unit;
|
|
}
|
|
|
|
public static GateUnit? Get(this GateUnitManageComponent self, long accountId)
|
|
{
|
|
return self.Units.GetValueOrDefault(accountId);
|
|
}
|
|
|
|
public static bool TryGet(this GateUnitManageComponent self, long accountId, out GateUnit? unit)
|
|
{
|
|
return self.Units.TryGetValue(accountId, out unit);
|
|
}
|
|
|
|
public static async FTask Remove(this GateUnitManageComponent self, long accountId, bool isDispose = true)
|
|
{
|
|
if (!self.Units.TryGetValue(accountId, out var unit)) return;
|
|
//通知其他服务器下线
|
|
var result = await GateLoginHelper.Offline(unit);
|
|
if (result != 0)
|
|
{
|
|
Log.Error($"通知其他服务器下线失败，错误码：{result}");
|
|
return;
|
|
}
|
|
|
|
Log.Info($"accountId:{accountId} 下线成功");
|
|
self.Units.Remove(accountId);
|
|
if (isDispose)
|
|
{
|
|
unit.Dispose();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 回话迭代器
|
|
/// </summary>
|
|
/// <param name="self"></param>
|
|
/// <returns></returns>
|
|
public static IEnumerable<Session> ForEachUnitSession(this GateUnitManageComponent self)
|
|
{
|
|
foreach (var (_, gateUnit) in self.Units)
|
|
{
|
|
Session gateSession = gateUnit.Session;
|
|
if (gateSession == null) continue;
|
|
yield return gateSession;
|
|
}
|
|
}
|
|
} |