caliverse_server/GameServer/Contents/UserBlock/DbQuery/DBQBlockUser.cs

using Google.Protobuf;
using Google.Protobuf.WellKnownTypes;


using ServerCore;
using ServerBase;
using ServerCommon;
using ServerCommon.BusinessLogDomain;
using MetaAssets;


namespace GameServer;

public class DBQBlockUser : QueryExecutorBase
{
    private string m_my_combination_key_for_pk = string.Empty;
    private string m_my_combination_key_for_sk = string.Empty;

    BlockUser? m_block_user_nullable = null;
    private string m_pk = string.Empty;
    private string m_sk = string.Empty;

    public DBQBlockUser(string myGuid, string blockGuid)
        : base(typeof(DBQBlockUser).Name)
    {
        m_my_combination_key_for_pk = myGuid;
        m_my_combination_key_for_sk = blockGuid;
    }

    //===================================================================================================
    // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
    //===================================================================================================
    public override async Task<Result> onPrepareQuery()
    {
        var result = new Result();

        var server_logic = GameServerApp.getServerLogic();
        var db_client = server_logic.getDynamoDbClient();

        var owner = getOwner();
        NullReferenceCheckHelper.throwIfNull(owner, () => "owner is null !!!");

        m_block_user_nullable = new BlockUser(owner);

        var user_guid = m_my_combination_key_for_pk;
        var block_user_guid = m_my_combination_key_for_sk;

        (result, var make_primary_key) = await DynamoDBDocBaseHelper.makePrimaryKey<BlockUserDoc>(user_guid, block_user_guid);
        if (result.isFail())
        {
            return result;
        }
        NullReferenceCheckHelper.throwIfNull(make_primary_key, () => $"make_primary_key is null !!! : PK:{user_guid}, SK:{block_user_guid} - {owner.toBasicString()}");

        var attribute = m_block_user_nullable.getEntityAttribute<BlockUserAttribute>();
        NullReferenceCheckHelper.throwIfNull(attribute, () => "attribute is null !!!");

        attribute.BlockGuid = m_my_combination_key_for_sk;
        attribute.IsNew = 1;
        attribute.CreateTime = DateTimeHelper.Current;

        m_pk = make_primary_key.PK;
        m_sk = make_primary_key.SK;

        return result;
    }

    //===================================================================================================
    // onPrepareQuery()를 성공할 경우 호출된다.
    //===================================================================================================
    public override async Task<Result> onQuery()
    {
        NullReferenceCheckHelper.throwIfNull(m_block_user_nullable, () => "m_block_user_nullable is null !!!");

        var result = new Result();

        var query_batch = getQueryBatch();
        NullReferenceCheckHelper.throwIfNull(query_batch, () => "query_batch is null !!!");

        var db_connector = query_batch.getDynamoDbConnector();

        var block_user_attribute = m_block_user_nullable.getEntityAttribute<BlockUserAttribute>();
        NullReferenceCheckHelper.throwIfNull(block_user_attribute, () => "block_user_attribute is null !!!");

        //block_user_attribute.newEntityAttribute();

        var block_user_doc = new BlockUserDoc(m_my_combination_key_for_pk, block_user_attribute.BlockGuid);
        block_user_doc.setQueryType(QueryType.Insert);
        var block_user_attib = block_user_doc.getAttrib<BlockUserAttrib>();
        NullReferenceCheckHelper.throwIfNull(block_user_attib, () => "block_user_attib is null !!!");
        block_user_attib.BlockGuid = block_user_attribute.BlockGuid;
        block_user_attib.IsNew = block_user_attribute.IsNew;
        

        (result, Amazon.DynamoDBv2.DocumentModel.Document? block_document) = await block_user_doc.onCopyToDocument(); ;
        if (result.isFail())
        {
            var err_msg = $"Failed to onCopyToDocument() !!!, m_pk:{m_pk}, m_sk:{m_sk}, {result.toBasicString()} - {toBasicString()}";
            Log.getLogger().error(err_msg);

            return result;
        }

        var table = db_connector.getTableByName(block_user_doc.TableName);
        var query_context = block_document.createDocumentQueryContext(table.TableName, block_user_doc.getQueryType());
        (result, _) = await table.simpleTransactWriteWithDocument(new List<DynamoDbDocumentQueryContext>() { query_context }, 2, query_batch.getTransId());
        if (result.isFail())
        {
            var err_msg = $"Failed to simpleTransactWriteWithDocument !!!, m_pk:{m_pk}, m_sk:{m_sk}, {result.toBasicString()} - {toBasicString()}";
            Log.getLogger().error(err_msg);

            return result;
        }

        return result;
    }

    //===================================================================================================
    // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
    //===================================================================================================
    public override Task onQueryResponseCommit()
    {
        NullReferenceCheckHelper.throwIfNull(m_block_user_nullable, () => "m_block_user_nullable is null !!!");

        //여기서 메모리 업데이트
        var owner = getOwner();
        NullReferenceCheckHelper.throwIfNull(owner, () => "owner is null !!!");

        var block_user_agent_action = owner.getEntityAction<BlockUserAgentAction>();
        NullReferenceCheckHelper.throwIfNull(block_user_agent_action, () => "block_user_agent_action is null !!!");

        var block_users = block_user_agent_action.getBlockUsers();

        block_users.AddOrUpdate(m_my_combination_key_for_sk, m_block_user_nullable, (key, oldValue) => m_block_user_nullable);

        return Task.CompletedTask;
    }

    //===================================================================================================
    // DB 쿼리를 실패하고, doFnRollback()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
    //===================================================================================================
    public override Task onQueryResponseRollback(Result errorResult)
    {
        return Task.CompletedTask;
    }
    private new Player? getOwner() => getQueryBatch()?.getLogActor() as Player;
}