using ServerCommon; using ServerCore; using ServerBase; using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; namespace GameServer; public class DBQDeleteFriend : QueryExecutorBase { private string m_my_combination_key_for_pk = string.Empty; private string m_my_combination_key_for_sk = string.Empty; private string m_friend_combination_key_for_pk = string.Empty; private string m_friend_combination_key_for_sk = string.Empty; private FriendDoc m_my_doc = new(); private string m_my_pk = string.Empty; private FriendDoc m_friend_doc = new(); private string m_friend_pk = string.Empty; public DBQDeleteFriend(string myGuid, string friendGuid) : base(typeof(DBQDeleteFriend).Name) { m_my_combination_key_for_pk = myGuid; m_my_combination_key_for_sk = friendGuid; m_friend_combination_key_for_pk = friendGuid; m_friend_combination_key_for_sk = myGuid; } //=================================================================================================== // DB 쿼리 직전에 준비해야 할 로직들을 작성한다. //=================================================================================================== public override Task onPrepareQuery() { var result = new Result(); var owner = getOwner(); NullReferenceCheckHelper.throwIfNull(owner, () => "owner is null !!!"); m_my_doc = new FriendDoc(); m_friend_doc = new FriendDoc(); m_my_doc.setCombinationKeyForPKSK(m_my_combination_key_for_pk, m_my_combination_key_for_sk); m_friend_doc.setCombinationKeyForPKSK(m_friend_combination_key_for_pk, m_friend_combination_key_for_sk); var my_error_code = m_my_doc.onApplyPKSK(); var friend_error_code = m_friend_doc.onApplyPKSK(); if (my_error_code.isFail() || friend_error_code.isFail()) { var err_msg = $"Failed to onApplyPKSK() !!! my_error_code : {my_error_code.toBasicString()}, friend_error_code : {friend_error_code.toBasicString()} - {owner.toBasicString()}"; result.setFail(my_error_code, err_msg); Log.getLogger().error(err_msg); return Task.FromResult(result); } m_my_pk = m_my_doc.getPK(); m_friend_pk = m_friend_doc.getPK(); return Task.FromResult(result); } //=================================================================================================== // onPrepareQuery()를 성공할 경우 호출된다. //=================================================================================================== public override async Task onQuery() { var result = new Result(); var query_batch = getQueryBatch(); NullReferenceCheckHelper.throwIfNull(query_batch, () => "query_batch is null !!!"); var db_connector = query_batch.getDynamoDbConnector(); (result, var my_document) = await m_my_doc.onCopyToDocument(); if (result.isFail()) { var err_msg = $"Failed to onCopyToDocument() !!! : {result.toBasicString()}, m_my_pk:{m_my_pk}, m_friend_pk:{m_friend_pk} - {toBasicString()}"; Log.getLogger().error(err_msg); return result; } (result, var friend_document) = await m_friend_doc.onCopyToDocument(); if (result.isFail()) { var err_msg = $"Failed to onCopyToDocument() !!! : {result.toBasicString()}, m_my_pk:{m_my_pk}, m_friend_pk:{m_friend_pk} - {toBasicString()}"; Log.getLogger().error(err_msg); return result; } var table = db_connector.getTableByDoc(m_friend_doc); var my_docmument_ctx = new DynamoDbDocumentQueryContext(table.TableName, my_document, QueryType.Delete); var my_friend_ctx = new DynamoDbDocumentQueryContext(table.TableName, friend_document, QueryType.Delete); (result, _) = await table.simpleTransactWriteWithDocument(new List() { my_docmument_ctx, my_friend_ctx }, 2, query_batch.getTransId()); if (result.isFail()) { var err_msg = $"Failed to simpleTransactWrite !!! : {result.toBasicString()}, m_my_pk:{m_my_pk}, m_friend_pk:{m_friend_pk} - {toBasicString()}"; Log.getLogger().error(err_msg); return result; } return result; } //=================================================================================================== // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다. //=================================================================================================== public override Task onQueryResponseCommit() { //여기서 메모리 업데이트 var owner = getOwner(); NullReferenceCheckHelper.throwIfNull(owner, () => "owner is null !!!"); var friend_agent_action = owner.getEntityAction(); NullReferenceCheckHelper.throwIfNull(friend_agent_action, () => "friend_agent_action is null !!!"); var friends = friend_agent_action.getFriends(); if (false == friends.TryRemove(m_friend_combination_key_for_pk, out var friend)) { var err_msg = $"Delete Friend : Failed to TryRemove !!! m_my_combination_key_for_sk = {m_my_combination_key_for_sk}, m_friend_combination_key_for_sk = {m_friend_combination_key_for_sk} - {toBasicString()}"; Log.getLogger().error(err_msg); } return Task.CompletedTask; } //=================================================================================================== // DB 쿼리를 실패하고, doFnRollback()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다. //=================================================================================================== public override Task onQueryResponseRollback(Result errorResult) { return Task.CompletedTask; } private new Player? getOwner() => getQueryBatch()?.getLogActor() as Player; }