caliverse_server/GameServer/Contents/MyHome/DBquery/DBQMyhomeInitializeForTestUser.cs

using Amazon.S3.Model;
using ServerCommon;
using ServerCore; using ServerBase;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace GameServer
{
    internal class DBQMyhomeInitializeForTestUser : QueryExecutorBase
    {
        private string m_combination_key_for_pk = string.Empty;
        private string m_pk = string.Empty;

        MyhomeUgcInfo m_myhome_ugc_info;

        public DBQMyhomeInitializeForTestUser(string testUserGuid, MyhomeUgcInfo myhomeUgcInfo)
            : base(typeof(DBQMyhomeInitializeForTestUser).Name)
        {
            m_combination_key_for_pk = testUserGuid;
            m_myhome_ugc_info = myhomeUgcInfo;
        }

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

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

            var doc = new MyhomeDoc();
            doc.setCombinationKeyForPK(m_combination_key_for_pk);

            var error_code = doc.onApplyPKSK();
            if (error_code.isFail())
            {
                err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}, {owner.toBasicString()}";
                result.setFail(error_code, err_msg);
                Log.getLogger().error(result.toBasicString());

                return result;
            }

            m_pk = doc.getPK();

            return await Task.FromResult(result);
        }

        //=====================================================================================
        // onPrepareQuery()를 성공할 경우 호출된다.
        //=====================================================================================
        public override async Task<Result> onQuery()
        {
            var result = new Result();
            var err_msg = string.Empty;

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

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

            var query_runner = query_batch.getQueryRunner();
            NullReferenceCheckHelper.throwIfNull(query_runner, () => $"query_runner is null !!!");

            var db_connector = query_batch.getDynamoDbConnector();

            var table = db_connector.getTableByDoc<MyhomeDoc>();
            var query_config = db_connector.makeQueryConfigForReadByPKOnly(m_pk);

            (result, var read_documents) = await table.simpleQueryWithQueryOperationConfig(query_config, query_batch.getTransId());
            if (result.isFail())
            {
                return result;
            }

            foreach (var read_document in read_documents)
            {
                var myhome_doc = new MyhomeDoc();
                result = await myhome_doc.onCopyFromDocument(read_document);
                if (result.isFail())
                {
                    return result;
                }

                var myhome_attrib = myhome_doc.getAttrib<MyHomeAttrib>();
                NullReferenceCheckHelper.throwIfNull(myhome_attrib, () => $"myhome_attrib is null !!!");

                // 마이홈 아이템 제거
                result = await tryDeleteItemDocFromMyhome(myhome_attrib.MyhomeGuid);
                if (result.isFail())
                {
                    return result;
                }

                // 마이홈에 배치된 Npc 배치 초기화
                result = await tryDeleteNpcLocationInTargetDocFromMyhome(myhome_attrib.MyhomeGuid);
                if (result.isFail())
                {
                    return result;
                }
                
                // MyhomeUgcInfo S3 에서 제거
                result = await MyhomeHelper.deleteMyhomeUgcInfoFolderFile(myhome_attrib.MyhomeGuid);
                if (result.isFail())
                {
                    err_msg = $"Failed to deleteMyhomeUgcInfoFolderFile() !!! : {result.toBasicString()}";
                    Log.getLogger().error(err_msg);

                    return result;
                }
                
                // MyhomeDoc Delete
                var to_delete_doc = read_document.createDocumentQueryContext(table.TableName, QueryType.Delete);
                result = await query_runner.tryRegisterQueryContext(to_delete_doc);
                if (result.isFail())
                {
                    return result;
                }
            }

            // 마이홈 생성
            result = await createMyhomeFromMyhomeUgcInfo(owner, m_myhome_ugc_info);
            if (result.isFail())
            {
                err_msg = $"Failed to createMyhomeFromMyhomeUgcInfo() !!! : {result.toBasicString()}";
                Log.getLogger().error(err_msg);

                return result;
            }

            return await Task.FromResult(result);
        }

        //=====================================================================================
        // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
        //=====================================================================================
        public override async Task onQueryResponseCommit()
        {
            await Task.CompletedTask;
            return;
        }

        //=====================================================================================
        // DB 쿼리를 실패하고, doFnRollback()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
        //=====================================================================================
        public override async Task onQueryResponseRollback(Result errorResult)
        {
            await Task.CompletedTask;
            return;
        }

        async Task<Result> tryDeleteItemDocFromMyhome(string myhomeGuid)
        {
            var result = new Result();
            var err_msg = string.Empty;

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

            var query_runner = query_batch.getQueryRunner();
            NullReferenceCheckHelper.throwIfNull(query_runner, () => $"query_runner is null !!!");

            var doc = new ItemDoc();
            doc.setCombinationKeyForPK(myhomeGuid);

            var error_code = doc.onApplyPKSK();
            if (error_code.isFail())
            {
                err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}";
                result.setFail(error_code, err_msg);
                Log.getLogger().error(result.toBasicString());

                return result;
            }

            var db_connector = query_batch.getDynamoDbConnector();

            var table = db_connector.getTableByDoc<ItemDoc>();
            var query_config = db_connector.makeQueryConfigForReadByPKOnly(doc.getPK());

            (result, var read_documents) = await table.simpleQueryWithQueryOperationConfig(query_config, query_batch.getTransId());
            if (result.isFail())
            {
                return result;
            }

            foreach(var read_document in read_documents)
            {
                // ItemDoc Delete
                var to_delete_doc = read_document.createDocumentQueryContext(table.TableName, QueryType.Delete);
                result = await query_runner.tryRegisterQueryContext(to_delete_doc);
                if (result.isFail())
                {
                    return result;
                }
            }

            return result;
        }

        async Task<Result> tryDeleteNpcLocationInTargetDocFromMyhome(string myhomeGuid)
        {
            var result = new Result();
            var err_msg = string.Empty;

            var server_logic = GameServerApp.getServerLogic();
            NullReferenceCheckHelper.throwIfNull(server_logic, () => $"server_logic is null !!!");

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

            var query_runner = query_batch.getQueryRunner();
            NullReferenceCheckHelper.throwIfNull(query_runner, () => $"query_runner is null !!!");

            var location_unique_id = server_logic.getMap().makeLOCATION_UNIQUE_IDByMetaIdWithLocationTargetType(LocationTargetType.Instance, myhomeGuid);

            var doc = new NpcLocationInTargetDoc();
            doc.setCombinationKeyForPK(location_unique_id);

            var error_code = doc.onApplyPKSK();
            if (error_code.isFail())
            {
                err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}";
                result.setFail(error_code, err_msg);
                Log.getLogger().error(result.toBasicString());

                return result;
            }

            var db_connector = query_batch.getDynamoDbConnector();
            var table = db_connector.getTableByDoc<NpcLocationInTargetDoc>();
            var query_config = db_connector.makeQueryConfigForReadByPKOnly(doc.getPK());

            (result, var read_documents) = await table.simpleQueryWithQueryOperationConfig(query_config, query_batch.getTransId());
            if (result.isFail())
            {
                return result;
            }

            foreach (var read_document in read_documents)
            {
                // NpcLocationInTargetDoc Delete
                var to_delete_doc = read_document.createDocumentQueryContext(table.TableName, QueryType.Delete);
                result = await query_runner.tryRegisterQueryContext(to_delete_doc);
                if (result.isFail())
                {
                    return result;
                }
            }

            return result;
        }

        async Task<Result> createMyhomeFromMyhomeUgcInfo(Player player, MyhomeUgcInfo myhome_ugc_info)
        {
            var result = new Result();
            var err_msg = string.Empty;

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

            var query_runner = query_batch.getQueryRunner();
            NullReferenceCheckHelper.throwIfNull(query_runner, () => $"query_runner is null !!!");

            var user_guid = player.getUserGuid();

            var myhome = new Myhome(user_guid);
            await myhome.onInit();
            
            var myhome_action = myhome.getEntityAction<MyhomeAction>();
            NullReferenceCheckHelper.throwIfNull(myhome_action, () => $"myhome_action is null !!!");

            await myhome_action.tryMakeDefaultMyhome(myhome_ugc_info);
            if (result.isFail())
            {
                return result;
            }

            var myhome_attribute = myhome.getEntityAttribute<MyhomeAttribute>();
            NullReferenceCheckHelper.throwIfNull(myhome_attribute, () => $"myhome_attribute is null !!!");

            (result, var myhome_doc) = await myhome_attribute.toDocBase();
            if (result.isFail())
            {
                Log.getLogger().error($"Failed to toDocBase() by DefaultUserCreator !!! : {myhome.toBasicString()}");
                return result;
            }
            NullReferenceCheckHelper.throwIfNull(myhome_doc, () => $"item_docs is null !!!");

            (result, var myhome_document) = await myhome_doc.onCopyToDocument();
            if (result.isFail())
            {
                return result;
            }

            var to_insert_doc = myhome_document.createDocumentQueryContext(dynamo_db_connector.getTableFullName(myhome_doc.TableName), QueryType.Insert);
            result = await query_runner.tryRegisterQueryContext(to_insert_doc);
            if (result.isFail())
            {
                return result;
            }

            var myhome_inventory_action = myhome.getEntityAction<MyhomeInventoryAction>();
            NullReferenceCheckHelper.throwIfNull(myhome_inventory_action, () => $"myhome_inventory_action is null !!!");

            var my_home_bag_inven = myhome_inventory_action.getBagInven();
            var my_home_has_items = my_home_bag_inven.getHasItemBases();

            foreach (var item in my_home_has_items)
            {
                var new_item = item;

                var item_attribute = new_item.getEntityAttribute<ItemAttributeBase>();
                NullReferenceCheckHelper.throwIfNull(item_attribute, () => $"item_attribute is null !!!");

                (result, var item_doc) = await item_attribute.toDocBase();
                if (result.isFail())
                {
                    Log.getLogger().error($"Failed to toDocBase() by DefaultUserCreator !!! : {new_item.toBasicString()}");
                    continue;
                }
                NullReferenceCheckHelper.throwIfNull(item_doc, () => $"item_docs is null !!!");

                (result, var document) = await item_doc.onCopyToDocument();
                if (result.isFail())
                {
                    return result;
                }

                to_insert_doc = document.createDocumentQueryContext(dynamo_db_connector.getTableFullName(item_doc.TableName), QueryType.Insert);
                result = await query_runner.tryRegisterQueryContext(to_insert_doc);
                if (result.isFail())
                {
                    return result;
                }
            }

            result = await myhome_action.tryUploadMyhomeUgcInfoToS3();
            if (result.isFail())
            {
                err_msg = $"Failed to tryUploadMyhomeUgcInfoToS3() !!! : {result.toBasicString()}";
                Log.getLogger().error(err_msg);

                return result;
            }

            return result;
        }
    }
}