using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Collections.Concurrent;



using Newtonsoft.Json;
using Amazon.DynamoDBv2;
using Amazon.DynamoDBv2.Model;
using Amazon.DynamoDBv2.DocumentModel;
using Amazon.Runtime;


using ServerCore;
using ServerBase;


using SESSION_ID = System.Int32;
using WORLD_ID = System.UInt32;
using META_ID = System.UInt32;
using ENTITY_GUID = System.String;
using ACCOUNT_ID = System.String;
using OWNER_GUID = System.String;
using USER_GUID = System.String;
using CHARACTER_GUID = System.String;
using ITEM_GUID = System.String;



namespace ServerCommon;

public class DBQWriteToAttributeAllWithTransactionRunner : QueryExecutorBase
{
    private readonly List<DynamoDbDocBase> m_to_write_docs = new();
    private readonly List<IEntityAttributeTransactor> m_transactors = new();

    private readonly Dictionary<string, EntityCommonResult> m_entity_common_results = new();

    public DBQWriteToAttributeAllWithTransactionRunner()
        : base(typeof(DBQWriteToAttributeAllWithTransactionRunner).Name)
    {
    }

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

        var result = new Result();
        var err_msg = string.Empty;

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

        var to_find_transaction_type = TransactionIdType.PrivateContents;
        var transaction_runner = owner.findTransactionRunner(to_find_transaction_type);
        if(null == transaction_runner)
        {
            err_msg = $"Not found TransactionRunner !!! : TransactionType:{to_find_transaction_type} - {owner.toBasicString()}";
            result.setFail(ServerErrorCode.TransactionRunnerNotFound, err_msg);
            Log.getLogger().error(result.toBasicString());

            return result;
        }

        // 자기 자신의 Transactor들을 얻어 온다 !!!
        var transactors = transaction_runner.getEntityAttributeTransactorAll4DbQuery();
        foreach(var transactor in transactors)
        {
            if(true == transactor.isReadOnly())
            {
                continue;
            }

            (result, var to_write_doc_nullable) = await transactor.makeDocBase();
            if(result.isFail())
            {
                return result;
            }
            NullReferenceCheckHelper.throwIfNull(to_write_doc_nullable, () => $"to_write_doc_nullable is null !!!");

            m_to_write_docs.Add(to_write_doc_nullable);
            m_transactors.Add(transactor);
        }

        return result;
    }

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

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

        var query_batch = getQueryBatch();
        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!! - {owner.toBasicString()}");

        var query_runner = query_batch.getQueryRunner();
        NullReferenceCheckHelper.throwIfNull(query_runner, () => $"query_runner is null !!! - {owner.toBasicString()}");

        foreach (var to_write_doc in m_to_write_docs)
        {
            (result, var query_context) = await query_runner.tryCreateQueryContext(to_write_doc);
            if (result.isFail())
            {
                return result;
            }
            NullReferenceCheckHelper.throwIfNull(query_context, () => $"query_context is null !!! - {owner.toBasicString()}");

            result = await query_runner.tryRegisterQueryContext(query_context);
            if (result.isFail())
            {
                return result;
            }
        }

        var write_msg = $"{this.getTypeName()} : tryWriteCount:{m_to_write_docs.Count}, isUseTransact:{query_batch.isUseTransact()} - TransId:{query_batch.getTransId()}";
        Log.getLogger().info(write_msg);

        return result;
    }

    //=====================================================================================
    // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
    //=====================================================================================
    public override async Task onQueryResponseCommit()
    {
        var result = new Result();
        var err_msg = string.Empty;

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

        var to_find_transaction_id_type = TransactionIdType.PrivateContents;
        var transaction_runner = owner.findTransactionRunner(to_find_transaction_id_type);
        if (null == transaction_runner)
        {
            err_msg = $"transaction_runner is null !!! : TransactionIdType:{to_find_transaction_id_type} - {owner.toBasicString()}";
            Log.getLogger().error(err_msg);

            return;
        }

        var query_batch = getQueryBatch();
        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!! - {owner.toBasicString()}");

        result = await transaction_runner.onCommitResults4DbQuery(m_transactors, query_batch);
        if (result.isFail())
        {
            err_msg = $"Failed to onCommitResults4DbQuery() !!! : {result.toBasicString()} - {owner.toBasicString()}";
            Log.getLogger().error(err_msg);

            return;
        }        
    }

    //=====================================================================================
    // DB 쿼리를 성공한 후에 EntityAttribute 변경 사항이 있는 경우 호출해 준다.
    //=====================================================================================
    public virtual async Task onMergeAfterQueryCommit()
    {
        await Task.CompletedTask;

        var result = new Result();
        var err_msg = string.Empty;

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

        var to_find_transaction_id_type = TransactionIdType.PrivateContents;
        var transaction_runner = owner.findTransactionRunner(to_find_transaction_id_type);
        if (null == transaction_runner)
        {
            err_msg = $"transaction_runner is null !!! : TransactionIdType:{to_find_transaction_id_type} - {owner.toBasicString()}";
            Log.getLogger().error(err_msg);

            return;
        }

        var query_batch = getQueryBatch();
        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!! - {owner.toBasicString()}");

        result = await transaction_runner.onCommitResults(query_batch);
        if (result.isFail())
        {
            err_msg = $"Failed to onCommitResults() !!! : {result.toBasicString()} - {owner.toBasicString()}";
            Log.getLogger().error(err_msg);

            return;
        }
    }

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