초기커밋

GameServer/Entity/CaliumStorage/DbQuery/DBQStorageCaliumSyncDateUpdate.cs

@@ -0,0 +1,133 @@
+using Amazon.DynamoDBv2;
+using Amazon.DynamoDBv2.Model;
+
+
+using ServerCore;
+using ServerBase;
+using ServerCommon;
+
+
+
+namespace GameServer;
+
+public class DBQStorageCaliumSyncDateUpdate : QueryExecutorBase
+{
+    private PrimaryKey m_primary_key = new();
+    private DateTime m_sync_date { get; set; }
+    
+    public DBQStorageCaliumSyncDateUpdate(DateTime syncDate) : base(nameof(DBQStorageCaliumSyncDateUpdate))
+    {
+        m_sync_date = syncDate;
+    }
+    
+    //=====================================================================================
+    // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
+    //=====================================================================================
+    public override async Task<Result> onPrepareQuery()
+    {
+        var owner = getOwner();
+
+        var (result, make_primary_key) = await DynamoDBDocBaseHelper.makePrimaryKey<CaliumStorageDoc>(string.Empty);
+        if (result.isFail())
+        {
+            return result;
+        }
+        NullReferenceCheckHelper.throwIfNull(make_primary_key, () => $"make_primary_key is null !!! - {owner.toBasicString()}");
+
+        m_primary_key = make_primary_key;
+
+        return result;
+    }
+    
+    //=====================================================================================
+    // onPrepareQuery()를 성공할 경우 호출된다.
+    //=====================================================================================
+    public override async Task<Result> onQuery()
+    {
+        var owner = getOwner();
+
+        var query_batch = getQueryBatch() as QueryBatch<QueryRunnerWithItemRequest>;
+        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!! - {owner.toBasicString()}");
+
+        var query_runner_with_item_request = query_batch.getQueryRunner();
+        NullReferenceCheckHelper.throwIfNull(query_runner_with_item_request, () => $"query_runner_with_item_request is null !!! - {owner.toBasicString()}");
+
+        var db_connector = query_batch.getDynamoDbConnector();
+        var target = JsonHelper.getJsonPropertyName<CaliumStorageAttrib>(nameof(CaliumStorageAttrib.DailyRollUpCheckDate));
+        var update = makeUpdateItemRequest(db_connector, m_primary_key.toKeyWithAttributeValue(), target);
+        if (update.result.isFail()) return update.result;
+        NullReferenceCheckHelper.throwIfNull(update.updateRequest, () => $"update_request is null !!! - {owner.toBasicString()}");
+
+        var exception_handler = new DynamoDbQueryExceptionNotifier.ExceptionHandler(DynamoDbQueryExceptionNotifier.ConditionalCheckFailed, ServerErrorCode.LackOfTotalCalium);
+        
+        var query_context = update.updateRequest.createItemRequestQueryContext(QueryType.Update, exception_handler);
+        var result = await query_runner_with_item_request.tryRegisterQueryContext(query_context);
+
+        return 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;
+    }
+
+    private (Result result, UpdateItemRequest? updateRequest) makeUpdateItemRequest(DynamoDbClient dbClient, Dictionary<string, AttributeValue> attributeValueWithPrimaryKey, string targetAttribName)
+    {
+        var result = new Result();
+        string err_msg;
+
+        if (string.IsNullOrEmpty(targetAttribName))
+        {
+            err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : targetAttribName - {targetAttribName}";
+            result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg);
+            Log.getLogger().error(result.toBasicString());
+
+            return (result, null);
+        }
+
+        var target_doc = new CaliumStorageDoc();
+
+        var query_builder = new DynamoDbItemRequestHelper.UpdateItemRequestBuilder(dbClient.getTableFullName(target_doc.TableName));
+        query_builder.withKeys(attributeValueWithPrimaryKey);
+
+        var attrib_path_json_string = target_doc.toJsonStringOfAttribs();
+        (var is_success, var attribute_expression) = DynamoDbClientHelper.toAttributeExpressionFromJson(attrib_path_json_string, targetAttribName);
+        if (false == is_success)
+        {
+            err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : attribPath:{attrib_path_json_string}, targetKey:{targetAttribName}";
+            result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg);
+            Log.getLogger().error(result.toBasicString());
+
+            return (result, null);
+        }
+
+        var attribute_names = DynamoDbClientHelper.toExpressionAttributeNamesFromJson(attrib_path_json_string, targetAttribName);
+        query_builder.withExpressionAttributeNames(attribute_names);
+
+        var update_expression = $"SET {attribute_expression} = :changeValue";
+        query_builder.withUpdateExpression(update_expression);
+        
+        var expression_attribute_values = new Dictionary<string, AttributeValue>
+        {
+            { ":changeValue", new AttributeValue { S = m_sync_date.toString("yyyy-MM-ddTHH:mm:ss") } }
+        };
+        query_builder.withExpressionAttributeValues(expression_attribute_values);
+
+        query_builder.withReturnValues(ReturnValue.ALL_NEW);
+
+        return query_builder.build();
+    }
+}