53 lines
1.5 KiB
C++
53 lines
1.5 KiB
C++
#include "etcd/v3/AsyncSetAction.hpp"
|
|
#include "etcd/v3/action_constants.hpp"
|
|
#include "etcd/v3/Transaction.hpp"
|
|
|
|
using etcdserverpb::Compare;
|
|
|
|
etcdv3::AsyncSetAction::AsyncSetAction(etcdv3::ActionParameters param, bool create)
|
|
: etcdv3::Action(param)
|
|
{
|
|
etcdv3::Transaction transaction(parameters.key);
|
|
isCreate = create;
|
|
transaction.init_compare(Compare::CompareResult::Compare_CompareResult_EQUAL,
|
|
Compare::CompareTarget::Compare_CompareTarget_VERSION);
|
|
|
|
transaction.setup_basic_create_sequence(parameters.key, parameters.value, parameters.lease_id);
|
|
|
|
if(isCreate)
|
|
{
|
|
transaction.setup_basic_failure_operation(parameters.key);
|
|
}
|
|
else
|
|
{
|
|
transaction.setup_set_failure_operation(parameters.key, parameters.value, parameters.lease_id);
|
|
}
|
|
response_reader = parameters.kv_stub->AsyncTxn(&context, transaction.txn_request, &cq_);
|
|
response_reader->Finish(&reply, &status, (void*)this);
|
|
}
|
|
|
|
etcdv3::AsyncTxnResponse etcdv3::AsyncSetAction::ParseResponse()
|
|
{
|
|
|
|
AsyncTxnResponse txn_resp;
|
|
|
|
if(!status.ok())
|
|
{
|
|
txn_resp.set_error_code(status.error_code());
|
|
txn_resp.set_error_message(status.error_message());
|
|
}
|
|
else
|
|
{
|
|
txn_resp.ParseResponse(parameters.key, parameters.withPrefix, reply);
|
|
std::string action = isCreate? etcdv3::CREATE_ACTION:etcdv3::SET_ACTION;
|
|
txn_resp.set_action(action);
|
|
|
|
if(!reply.succeeded() && action == etcdv3::CREATE_ACTION)
|
|
{
|
|
txn_resp.set_error_code(105);
|
|
txn_resp.set_error_message("Key already exists");
|
|
}
|
|
}
|
|
return txn_resp;
|
|
}
|