초기커밋

GameServer/Contents/Quest/DBQQuery/DBQQuestMailSend.cs

@@ -0,0 +1,127 @@
+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 DBQQuestMailSend : QueryExecutorBase
+{
+    private string m_combination_key_for_pk = string.Empty;
+    private string m_pk = string.Empty;
+
+
+    private readonly List<QuestMailDoc> m_quest_mail_docs = new();
+
+    public DBQQuestMailSend(string combinationKeyForPK, List<QuestMailDoc> docs) 
+        : base(typeof(DBQQuestMailSend).Name)
+    {
+        m_combination_key_for_pk = combinationKeyForPK;
+        m_quest_mail_docs = docs;
+    }
+
+
+    //===================================================================================================
+    // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
+    //===================================================================================================
+    public override Task<Result> onPrepareQuery()
+    {
+        var result = new Result();
+
+        var owner = getOwner();
+        NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!");
+
+        foreach (var quest_mail_doc in m_quest_mail_docs)
+        {
+            var mail_attrib = quest_mail_doc.getAttrib<QuestMailAttrib>();
+            NullReferenceCheckHelper.throwIfNull(mail_attrib, () => $"mail_attrib is null !!!");
+
+            quest_mail_doc.setCombinationKeyForPK(m_combination_key_for_pk);
+            quest_mail_doc.setCombinationKeyForSK(mail_attrib.QuestId.ToString());
+            
+            var error_code = quest_mail_doc.onApplyPKSK();
+            if (error_code.isFail())
+            {
+                var err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}, {owner.toBasicString()}";
+                Log.getLogger().error(err_msg);
+                continue;
+            }
+        }
+
+        return Task.FromResult(result);
+    }
+
+
+    //===================================================================================================
+    // onPrepareQuery()를 성공할 경우 호출된다.
+    //===================================================================================================
+    public override async Task<Result> onQuery()
+    {
+        var result = new Result();
+        var err_msg = string.Empty;
+
+        var query_batch = getQueryBatch();
+        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!!");
+
+        var db_connector = query_batch.getDynamoDbConnector();
+        var query_contexts = new List<DynamoDbDocumentQueryContext>();
+
+        foreach (var doc in m_quest_mail_docs)
+        {
+            (result, var quest_mail_document) = await doc.onCopyToDocument();
+            if(result.isFail())
+            {
+                err_msg = $"Failed to onCopyToDocument() !!! : {result.toBasicString()}, m_pk:{m_pk} - {toBasicString()}";
+                Log.getLogger().error(err_msg);
+                continue;
+            }
+            var docmument_ctx = new DynamoDbDocumentQueryContext(db_connector.getTableFullName(doc.TableName), quest_mail_document, QueryType.Insert);
+            query_contexts.Add(docmument_ctx);
+        }
+
+        var table = db_connector.getTableByDoc<QuestMailDoc>();
+        (result, _) = await table.simpleTransactWriteWithDocument(query_contexts, 2, query_batch.getTransId());
+        if (result.isFail())
+        {
+            err_msg = $"Failed to simpleTransactWrite !!! : {result.toBasicString()}, m_pk:{m_pk}, {toBasicString()}";
+            Log.getLogger().error(err_msg);
+            return result;
+        }
+
+        var owner = getOwner();
+        NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!");
+
+        var quest_mail_action = owner.getEntityAction<QuestMailAction>();
+        NullReferenceCheckHelper.throwIfNull(quest_mail_action, () => $"quest_mail_action is null !!!");
+
+        foreach (var doc in m_quest_mail_docs)
+        {
+            await quest_mail_action.setQuestMailFromDoc(doc);
+        }
+        
+        return result;
+    }
+
+
+    //===================================================================================================
+    // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
+    //===================================================================================================
+    public override Task onQueryResponseCommit()
+    {
+        return Task.CompletedTask;
+    }
+
+
+    //===================================================================================================
+    // DB 쿼리를 실패하고, doFnRollback()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
+    //===================================================================================================
+    public override Task onQueryResponseRollback(Result errorResult)
+    {
+        return Task.CompletedTask;
+    }
+
+    private new Player? getOwner() => getQueryBatch()?.getLogActor() as Player;
+}