258 lines
12 KiB
C#
258 lines
12 KiB
C#
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<Result> onPrepareQuery()
|
|
{
|
|
await Task.CompletedTask;
|
|
|
|
var result = new Result();
|
|
var err_msg = string.Empty;
|
|
|
|
var owner = getOwner();
|
|
|
|
(result, var make_primary_key) = await DynamoDBDocBaseHelper.makePrimaryKey<UgcNpcLikeSelectedFlagDoc>(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<Result> 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<UgcNpcLikeSelectedFlagDoc>(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<UgcNpcLikeSelectedFlagAttrib>();
|
|
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<UgcNpcLikeSelectedFlagAttrib>();
|
|
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<string, string>();
|
|
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<string, AttributeValue> attributeValueWithPrimaryKey
|
|
, Dictionary<string, string> 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<string, string>();
|
|
|
|
var target_doc = new UgcNpcLikeSelectedFlagDoc();
|
|
var attrib_path_json_string = target_doc.toJsonStringOfAttribs();
|
|
|
|
var expressionAttributeValues = new Dictionary<string, AttributeValue>();
|
|
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<UgcNpcLikeSelectedFlagAttrib>(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;
|
|
}
|
|
}
|
|
}
|