mirror of
https://github.com/codership/wsrep-lib.git
synced 2025-07-30 07:23:07 +03:00
Use provider for replaying in mock client_context instead of
calling server context apply directly.
This commit is contained in:
Teemu Ollakka
205
test/mock_provider.hpp
Normal file
205
test/mock_provider.hpp
Normal file
@ -0,0 +1,205 @@
|
||||
//
|
||||
// Copyright (C) 2018 Codership Oy <info@codership.com>
|
||||
//
|
||||
|
||||
#ifndef WSREP_FAKE_PROVIDER_HPP
|
||||
#define WSREP_FAKE_PROVIDER_HPP
|
||||
|
||||
#include "wsrep/provider.hpp"
|
||||
#include "wsrep/logger.hpp"
|
||||
#include "wsrep/buffer.hpp"
|
||||
|
||||
#include <cstring>
|
||||
#include <map>
|
||||
#include <iostream> // todo: proper logging
|
||||
|
||||
namespace wsrep
|
||||
{
|
||||
class mock_provider : public wsrep::provider
|
||||
{
|
||||
public:
|
||||
typedef std::map<wsrep::transaction_id, wsrep::seqno> bf_abort_map;
|
||||
|
||||
mock_provider(wsrep::server_context& server_context)
|
||||
: provider(server_context)
|
||||
, certify_result_()
|
||||
, commit_order_enter_result_()
|
||||
, commit_order_leave_result_()
|
||||
, release_result_()
|
||||
, group_id_("1")
|
||||
, server_id_("1")
|
||||
, group_seqno_(0)
|
||||
, bf_abort_map_()
|
||||
, start_fragments_()
|
||||
, fragments_()
|
||||
, commit_fragments_()
|
||||
, rollback_fragments_()
|
||||
{ }
|
||||
|
||||
int connect(const std::string&, const std::string&, const std::string&,
|
||||
bool)
|
||||
{ return 0; }
|
||||
int disconnect() { return 0; }
|
||||
enum wsrep::provider::status run_applier(void*)
|
||||
{
|
||||
return wsrep::provider::success;
|
||||
}
|
||||
// Provider implemenatation interface
|
||||
int start_transaction(wsrep::ws_handle&) { return 0; }
|
||||
enum wsrep::provider::status
|
||||
certify(wsrep::client_id client_id,
|
||||
wsrep::ws_handle& ws_handle,
|
||||
int flags,
|
||||
wsrep::ws_meta& ws_meta)
|
||||
{
|
||||
ws_handle = wsrep::ws_handle(ws_handle.transaction_id(), (void*)1);
|
||||
wsrep::log_info() << "provider certify: "
|
||||
<< "client: " << client_id.get()
|
||||
<< " flags: " << std::hex << flags
|
||||
<< std::dec
|
||||
<< " certify_status: " << certify_result_;
|
||||
|
||||
if (certify_result_)
|
||||
{
|
||||
return certify_result_;
|
||||
}
|
||||
|
||||
++fragments_;
|
||||
if (starts_transaction(flags))
|
||||
{
|
||||
++start_fragments_;
|
||||
}
|
||||
if (commits_transaction(flags))
|
||||
{
|
||||
++commit_fragments_;
|
||||
}
|
||||
if (rolls_back_transaction(flags))
|
||||
{
|
||||
assert(0);
|
||||
++rollback_fragments_;
|
||||
}
|
||||
|
||||
wsrep::stid stid(server_id_,
|
||||
ws_handle.transaction_id(),
|
||||
client_id);
|
||||
bf_abort_map::iterator it(bf_abort_map_.find(
|
||||
ws_handle.transaction_id()));
|
||||
if (it == bf_abort_map_.end())
|
||||
{
|
||||
++group_seqno_;
|
||||
wsrep::gtid gtid(group_id_, wsrep::seqno(group_seqno_));
|
||||
ws_meta = wsrep::ws_meta(gtid, stid,
|
||||
wsrep::seqno(group_seqno_ - 1),
|
||||
flags);
|
||||
return wsrep::provider::success;
|
||||
}
|
||||
else
|
||||
{
|
||||
enum wsrep::provider::status ret;
|
||||
if (it->second.nil())
|
||||
{
|
||||
ws_meta = wsrep::ws_meta(wsrep::gtid(), wsrep::stid(),
|
||||
wsrep::seqno::undefined(), 0);
|
||||
ret = wsrep::provider::error_certification_failed;
|
||||
}
|
||||
else
|
||||
{
|
||||
++group_seqno_;
|
||||
wsrep::gtid gtid(group_id_, wsrep::seqno(group_seqno_));
|
||||
ws_meta = wsrep::ws_meta(gtid, stid,
|
||||
wsrep::seqno(group_seqno_ - 1),
|
||||
flags);
|
||||
ret = wsrep::provider::error_bf_abort;
|
||||
}
|
||||
bf_abort_map_.erase(it);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
||||
int append_key(wsrep::ws_handle&, const wsrep::key&)
|
||||
{ return 0; }
|
||||
int append_data(wsrep::ws_handle&, const wsrep::const_buffer&)
|
||||
{ return 0; }
|
||||
int rollback(const wsrep::transaction_id)
|
||||
{
|
||||
++fragments_;
|
||||
++rollback_fragments_;
|
||||
return 0;
|
||||
}
|
||||
enum wsrep::provider::status
|
||||
commit_order_enter(const wsrep::ws_handle&,
|
||||
const wsrep::ws_meta&)
|
||||
{ return commit_order_enter_result_; }
|
||||
|
||||
int commit_order_leave(const wsrep::ws_handle&,
|
||||
const wsrep::ws_meta&)
|
||||
{ return commit_order_leave_result_;}
|
||||
|
||||
int release(wsrep::ws_handle&)
|
||||
{ return release_result_; }
|
||||
|
||||
int replay(wsrep::ws_handle&, void* ctx)
|
||||
{
|
||||
wsrep::mock_client_context& cc(
|
||||
*static_cast<wsrep::mock_client_context*>(ctx));
|
||||
wsrep::client_applier_mode applier_mode(cc);
|
||||
const wsrep::transaction_context& tc(cc.transaction());
|
||||
return server_context_.on_apply(cc, tc.ws_handle(), tc.ws_meta(),
|
||||
wsrep::const_buffer());
|
||||
}
|
||||
|
||||
int sst_sent(const wsrep::gtid&, int) { return 0; }
|
||||
int sst_received(const wsrep::gtid&, int) { return 0; }
|
||||
|
||||
std::vector<status_variable> status() const
|
||||
{
|
||||
return std::vector<status_variable>();
|
||||
}
|
||||
|
||||
// Methods to modify mock state
|
||||
/*! Inject BF abort event into the provider.
|
||||
*
|
||||
* \param bf_seqno Aborter sequence number
|
||||
* \param trx_id Trx id to be aborted
|
||||
* \param[out] victim_seqno
|
||||
*/
|
||||
enum wsrep::provider::status
|
||||
bf_abort(wsrep::seqno bf_seqno,
|
||||
wsrep::transaction_id trx_id,
|
||||
wsrep::seqno& victim_seqno)
|
||||
{
|
||||
// std::cerr << "bf_abort: " << trx_id << "\n";
|
||||
bf_abort_map_.insert(std::make_pair(trx_id, bf_seqno));
|
||||
if (bf_seqno.nil() == false)
|
||||
{
|
||||
group_seqno_ = bf_seqno.get();
|
||||
}
|
||||
victim_seqno = wsrep::seqno::undefined();
|
||||
return wsrep::provider::success;
|
||||
}
|
||||
|
||||
// Parameters to control return value from the call
|
||||
enum wsrep::provider::status certify_result_;
|
||||
enum wsrep::provider::status commit_order_enter_result_;
|
||||
enum wsrep::provider::status commit_order_leave_result_;
|
||||
enum wsrep::provider::status release_result_;
|
||||
|
||||
size_t start_fragments() const { return start_fragments_; }
|
||||
size_t fragments() const { return fragments_; }
|
||||
size_t commit_fragments() const { return commit_fragments_; }
|
||||
size_t rollback_fragments() const { return rollback_fragments_; }
|
||||
|
||||
private:
|
||||
wsrep::id group_id_;
|
||||
wsrep::id server_id_;
|
||||
long long group_seqno_;
|
||||
bf_abort_map bf_abort_map_;
|
||||
size_t start_fragments_;
|
||||
size_t fragments_;
|
||||
size_t commit_fragments_;
|
||||
size_t rollback_fragments_;
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
#endif // WSREP_FAKE_PROVIDER_HPP
|
||||
Reference in New Issue
Block a user