Inna/CypherCore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Core/Misc: Replace database query in WorldSession::HandleAddFriendOpcode with async version

Browse Source 
Port From (https://github.com/TrinityCore/TrinityCore/commit/54a6e603ffc8b4913669cf0f189a966d25b620d8)
This commit is contained in:
hondacrx
2022-03-02 09:35:51 -05:00
parent c26134cc27
commit 6e6ee2d694
4 changed files with 94 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/Game/Accounts/AccountManager.cs
+12 -9
View File
@@ -247,14 +247,6 @@ namespace Game
return !result.IsEmpty() ? result.Read<uint>(0) : 0;
}
public AccountTypes GetSecurity(uint accountId)
{
PreparedStatement stmt = DB.Login.GetPreparedStatement(LoginStatements.GET_ACCOUNT_ACCESS_GMLEVEL);
stmt.AddValue(0, accountId);
SQLResult result = DB.Login.Query(stmt);
return !result.IsEmpty() ? (AccountTypes)result.Read<byte>(0) : AccountTypes.Player;
}
public AccountTypes GetSecurity(uint accountId, int realmId)
{
PreparedStatement stmt = DB.Login.GetPreparedStatement(LoginStatements.GET_GMLEVEL_BY_REALMID);
@@ -264,6 +256,17 @@ namespace Game
return !result.IsEmpty() ? (AccountTypes)result.Read<uint>(0) : AccountTypes.Player;
}
public QueryCallback GetSecurityAsync(uint accountId, int realmId, Action<uint> callback)
{
PreparedStatement stmt = DB.Login.GetPreparedStatement(LoginStatements.GET_GMLEVEL_BY_REALMID);
stmt.AddValue(0, accountId);
stmt.AddValue(1, realmId);
return DB.Login.AsyncQuery(stmt).WithCallback(result =>
{
callback(!result.IsEmpty() ? result.Read<byte>(0) : (uint)AccountTypes.Player);
});
}
public bool GetName(uint accountId, out string name)
{
name = "";
@@ -497,7 +500,7 @@ namespace Game
return false;
}
RBACData rbac = new(accountId, "", (int)realmId);
RBACData rbac = new(accountId, "", (int)realmId, (byte)GetSecurity(accountId, (int)realmId));
rbac.LoadFromDB();
bool hasPermission = rbac.HasPermission(permissionId);
Source/Game/Entities/Player/SocialMgr.cs
+2 -2
View File
@@ -34,7 +34,7 @@ namespace Game.Entities
public const int FriendLimit = 50;
public const int IgnoreLimit = 50;
public void GetFriendInfo(Player player, ObjectGuid friendGUID, FriendInfo friendInfo)
public static void GetFriendInfo(Player player, ObjectGuid friendGUID, FriendInfo friendInfo)
{
if (!player)
return;
@@ -283,7 +283,7 @@ namespace Game.Entities
if (++ignoredCount > SocialManager.IgnoreLimit)
continue;
Global.SocialMgr.GetFriendInfo(player, v.Key, v.Value);
SocialManager.GetFriendInfo(player, v.Key, v.Value);
contactList.Contacts.Add(new ContactInfo(v.Key, v.Value));
}
Source/Game/Handlers/SocialHandler.cs
+79 -33
View File
@@ -222,42 +222,88 @@ namespace Game
if (!ObjectManager.NormalizePlayerName(ref packet.Name))
return;
FriendsResult friendResult = FriendsResult.NotFound;
ObjectGuid friendGuid = ObjectGuid.Empty;
CharacterCacheEntry characterInfo = Global.CharacterCacheStorage.GetCharacterCacheByName(packet.Name);
if (characterInfo != null)
CharacterCacheEntry friendCharacterInfo = Global.CharacterCacheStorage.GetCharacterCacheByName(packet.Name);
if (friendCharacterInfo == null)
{
friendGuid = characterInfo.Guid;
ObjectGuid friendAccountGuid = ObjectGuid.Create(HighGuid.WowAccount, characterInfo.AccountId);
Team team = Player.TeamForRace(characterInfo.RaceId);
uint friendAccountId = characterInfo.AccountId;
if (HasPermission(RBACPermissions.AllowGmFriend) || Global.AccountMgr.IsPlayerAccount(Global.AccountMgr.GetSecurity(friendAccountId, (int)Global.WorldMgr.GetRealm().Id.Index)))
{
if (friendGuid == GetPlayer().GetGUID())
friendResult = FriendsResult.Self;
else if (GetPlayer().GetTeam() != team && !HasPermission(RBACPermissions.TwoSideAddFriend))
friendResult = FriendsResult.Enemy;
else if (GetPlayer().GetSocial().HasFriend(friendGuid))
friendResult = FriendsResult.Already;
else
{
Player playerFriend = Global.ObjAccessor.FindPlayer(friendGuid);
if (playerFriend && playerFriend.IsVisibleGloballyFor(GetPlayer()))
friendResult = FriendsResult.AddedOnline;
else
friendResult = FriendsResult.AddedOffline;
if (GetPlayer().GetSocial().AddToSocialList(friendGuid, friendAccountGuid, SocialFlag.Friend))
GetPlayer().GetSocial().SetFriendNote(friendGuid, packet.Notes);
else
friendResult = FriendsResult.ListFull;
}
}
Global.SocialMgr.SendFriendStatus(GetPlayer(), FriendsResult.NotFound, ObjectGuid.Empty);
return;
}
Global.SocialMgr.SendFriendStatus(GetPlayer(), friendResult, friendGuid);
void processFriendRequest()
{
var playerGuid = _player.GetGUID();
var friendGuid = friendCharacterInfo.Guid;
var friendAccountGuid = ObjectGuid.Create(HighGuid.WowAccount, friendCharacterInfo.AccountId);
var team = Player.TeamForRace(friendCharacterInfo.RaceId);
var friendNote = packet.Notes;
if (playerGuid.GetCounter() != m_GUIDLow)
return; // not the player initiating request, do nothing
FriendsResult friendResult = FriendsResult.NotFound;
if (friendGuid == GetPlayer().GetGUID())
friendResult = FriendsResult.Self;
else if (GetPlayer().GetTeam() != team && !HasPermission(RBACPermissions.TwoSideAddFriend))
friendResult = FriendsResult.Enemy;
else if (GetPlayer().GetSocial().HasFriend(friendGuid))
friendResult = FriendsResult.Already;
else
{
Player pFriend = Global.ObjAccessor.FindPlayer(friendGuid);
if (pFriend != null && pFriend.IsVisibleGloballyFor(GetPlayer()))
friendResult = FriendsResult.Online;
else
friendResult = FriendsResult.AddedOnline;
if (GetPlayer().GetSocial().AddToSocialList(friendGuid, friendAccountGuid, SocialFlag.Friend))
GetPlayer().GetSocial().SetFriendNote(friendGuid, friendNote);
else
friendResult = FriendsResult.ListFull;
}
Global.SocialMgr.SendFriendStatus(GetPlayer(), friendResult, friendGuid);
}
if (HasPermission(RBACPermissions.AllowGmFriend))
{
processFriendRequest();
return;
}
// First try looking up friend candidate security from online object
Player friendPlayer = Global.ObjAccessor.FindPlayer(friendCharacterInfo.Guid);
if (friendPlayer != null)
{
if (!Global.AccountMgr.IsPlayerAccount(friendPlayer.GetSession().GetSecurity()))
{
Global.SocialMgr.SendFriendStatus(GetPlayer(), FriendsResult.NotFound, ObjectGuid.Empty);
return;
}
processFriendRequest();
return;
}
// When not found, consult database
GetQueryProcessor().AddCallback(Global.AccountMgr.GetSecurityAsync(friendCharacterInfo.AccountId, (int)Global.WorldMgr.GetRealmId().Index, friendSecurity =>
{
if (!Global.AccountMgr.IsPlayerAccount((AccountTypes)friendSecurity))
{
Global.SocialMgr.SendFriendStatus(GetPlayer(), FriendsResult.NotFound, ObjectGuid.Empty);
return;
}
processFriendRequest();
}));
}
[WorldPacketHandler(ClientOpcodes.DelFriend)]