using System; using System.Collections.Generic; using System.Globalization; using System.Linq; using System.Text; using System.Threading.Tasks; using Newtonsoft.Json; using Newtonsoft.Json.Linq; using Amazon.DynamoDBv2.Model; using Google.Protobuf.WellKnownTypes; using ServerCore; using ServerBase; using ServerCommon; using DYNAMO_DB_TABLE_FULL_NAME = System.String; using SESSION_ID = System.Int32; 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 UGC_NPC_META_GUID = System.String; using ITEM_GUID = System.String; namespace GameServer { public class DBQUgcNpcLikeSelectedFlagUpdate : QueryExecutorBase { private string m_combination_key_for_pk = string.Empty; private string m_combination_key_for_sk = string.Empty; private bool m_is_update_last_date = false; private PrimaryKey m_primary_key = new(); //===================================================================================== // 0 < deltaCount => 증가 // 0 > deltaCount => 감소 //===================================================================================== public DBQUgcNpcLikeSelectedFlagUpdate( string combinationKeyForPK , string combinationKeyForSK , bool isUpdateLastDate = false) : base(typeof(DBQUgcNpcLikeSelectedFlagUpdate).Name) { m_combination_key_for_pk = combinationKeyForPK; m_combination_key_for_sk = combinationKeyForSK; m_is_update_last_date = isUpdateLastDate; } //===================================================================================== // DB 쿼리 직전에 준비해야 할 로직들을 작성한다. //===================================================================================== public override async Task onPrepareQuery() { await Task.CompletedTask; var result = new Result(); var err_msg = string.Empty; var owner = getOwner(); (result, var make_primary_key) = await DynamoDBDocBaseHelper.makePrimaryKey(m_combination_key_for_pk, m_combination_key_for_sk); 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 onQuery() { var result = new Result(); var err_msg = string.Empty; var owner = getOwner(); var query_batch = getQueryBatch(); 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 selected_flag_query_config = db_connector.makeQueryConfigForReadByPKSK(m_primary_key.PK, m_primary_key.SK); (result, var found_selected_flag_docs) = await db_connector.simpleQueryDocTypesWithQueryOperationConfig(selected_flag_query_config, eventTid: query_batch.getTransId()); if (result.isFail()) { return result; } if (0 == found_selected_flag_docs.Count) { // UgcNpcLikeSelectedFlagDoc 없다면 PutItemRequest를 처리 한다. var doc = new UgcNpcLikeSelectedFlagDoc(OwnerEntityType.User, m_combination_key_for_pk, m_combination_key_for_sk); var attrib = doc.getAttrib(); NullReferenceCheckHelper.throwIfNull(attrib, () => $"attrib is null !!!"); attrib.IsSelectedFlag = true; if (m_is_update_last_date) attrib.LastCheckDate = DateTimeHelper.Current; (result, var result_document) = await doc.onCopyToDocument(); if (result.isFail()) { err_msg = $"Failed to onCopyToDocument() !!! : {result.toBasicString()}, {doc.toDocTypePKSK()} - {owner.toBasicString()}"; Log.getLogger().error(err_msg); return result; } // PutItemRequest 생성하는 simple 계열의 함수를 DynamoDbClientHelper에 추가한다. var put_item_request = new PutItemRequest(); put_item_request.TableName = db_connector.getTableFullName(doc.TableName); put_item_request.Item = result_document.ToAttributeMap(); var query_context = put_item_request.createItemRequestQueryContext(QueryType.Insert); result = await query_runner_with_item_request.tryRegisterQueryContext(query_context); } else { // UgcNpcLikeSelectedFlagDoc 있다면 UpdateItemRequest를 처리 한다. var selected_flag_attrib = found_selected_flag_docs[0].getAttrib(); NullReferenceCheckHelper.throwIfNull(selected_flag_attrib, () => $"selected_flag_attrib is null !!! - {owner.toBasicString()}"); var to_update_flag = (true == selected_flag_attrib.IsSelectedFlag ? false : true); var change_attrib_data = new Dictionary(); change_attrib_data.Add(nameof(UgcNpcLikeSelectedFlagAttrib.IsSelectedFlag), to_update_flag.ToString()); if(m_is_update_last_date) change_attrib_data.Add(nameof(UgcNpcLikeSelectedFlagAttrib.LastCheckDate), DateTimeHelper.Current.ToString(CultureInfo.InvariantCulture)); // UpdateItemRequest를 생성하는 simple 계열의 함수를 DynamoDbClientHelper에 추가한다. (result, var update_item_request) = makeUpdateItemRequest( db_connector , db_connector.getTableFullName(found_selected_flag_docs[0].TableName) , m_primary_key.toKeyWithAttributeValue() , change_attrib_data ); if (result.isFail()) { return result; } NullReferenceCheckHelper.throwIfNull(update_item_request, () => $"update_item_request is null !!! - {owner.toBasicString()}"); var query_context = update_item_request.createItemRequestQueryContext(QueryType.Upsert); result = await query_runner_with_item_request.tryRegisterQueryContext(query_context); } return result; } private (Result, UpdateItemRequest?) makeUpdateItemRequest( DynamoDbClient dbClient , DYNAMO_DB_TABLE_FULL_NAME tableFullName , Dictionary attributeValueWithPrimaryKey , Dictionary changeAttribs) { var result = new Result(); var owner = getOwner(); NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!"); var update_item_request = new UpdateItemRequest(); update_item_request.TableName = tableFullName; update_item_request.Key = attributeValueWithPrimaryKey; update_item_request.ExpressionAttributeNames = new Dictionary(); var target_doc = new UgcNpcLikeSelectedFlagDoc(); var attrib_path_json_string = target_doc.toJsonStringOfAttribs(); var expressionAttributeValues = new Dictionary(); var update_expression = "SET "; var expression_idx = 0; if (changeAttribs.Count <= 0) { var err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : attribPath:{attrib_path_json_string} - {owner.toBasicString()}"; result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg); Log.getLogger().error(result.toBasicString()); return (result, null); } foreach (var data in changeAttribs) { var target_key = JsonHelper.getJsonPropertyName(data.Key); (var is_success, var attribute_expression) = DynamoDbClientHelper.toAttributeExpressionFromJson(attrib_path_json_string, target_key); if(false == is_success) { var err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : attribPath:{attrib_path_json_string}, targetKey:{target_key} - {owner.toBasicString()}"; result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg); Log.getLogger().error(result.toBasicString()); continue; } var expressionAttributeNames = DynamoDbClientHelper.toExpressionAttributeNamesFromJson(attrib_path_json_string, target_key); foreach (var name in expressionAttributeNames) { update_item_request.ExpressionAttributeNames.TryAdd(name.Key, name.Value); } if (expression_idx > 0) update_expression += ","; update_expression += $" {attribute_expression} = :newValue_{expression_idx}"; expressionAttributeValues.Add($":newValue_{expression_idx}", new AttributeValue { S = data.Value }); expression_idx++; } update_item_request.UpdateExpression = update_expression; update_item_request.ExpressionAttributeValues = expressionAttributeValues; update_item_request.ReturnValues = "ALL_NEW"; return (result, update_item_request); } //===================================================================================== // 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; } public override Player getOwner() { var owner = base.getOwner() as Player; NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!"); return owner; } } }