using ServerCore; 
using ServerBase;
using ServerCommon;


namespace GameServer;

public class DBQCaliumStorageLoad : QueryExecutorBase
{
    private string m_pk = string.Empty;
    
    public DBQCaliumStorageLoad() : base(nameof(DBQCaliumStorageLoad)) {}
    
    //===================================================================================================
    // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
    //===================================================================================================
    public override Task<Result> onPrepareQuery()
    {
        var result = new Result();

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

        var doc = new CaliumStorageDoc();
        doc.setCombinationKeyForPK(string.Empty);

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

        m_pk = doc.getPK();

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

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

        var db_connector = query_batch.getDynamoDbConnector();
        var query_config = db_connector.makeQueryConfigForReadByPKOnly(m_pk);
        
        (result, var read_docs) = await db_connector.simpleQueryDocTypesWithQueryOperationConfig<CaliumStorageDoc>(query_config, eventTid: query_batch.getTransId());
        if (result.isFail()) return result;

        if (read_docs.Count <= 0)
        {
            result.setFail(ServerErrorCode.DynamoDbQueryNoMatchAttribute, $"fail to find calium storage doc");
            return result;
        }
        
        var calium_storage_doc = read_docs.FirstOrDefault();
        NullReferenceCheckHelper.throwIfNull(calium_storage_doc, () => $"calium_storage_doc is null !!! - {owner.toBasicString()}");

        var calium_storage_attribute = owner.getEntityAttribute<CaliumStorageAttribute>();
        NullReferenceCheckHelper.throwIfNull(calium_storage_attribute, () => $"calium_storage_attribute is null !!! - {owner.toBasicString()}");

        if (false == calium_storage_attribute.copyEntityAttributeFromDoc(calium_storage_doc))
        {
            var err_msg = $"failed to load Calium Storage !!! - {owner.toBasicString()}";
            result.setFail(ServerErrorCode.FailToLoadCalium, err_msg);
            Log.getLogger().error(result.toBasicString());
        }
        
        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 CaliumStorageEntity? getOwner() => getQueryBatch()?.getLogActor() as CaliumStorageEntity;
}