//
// Copyright (C) 2018 Codership Oy <info@codership.com>
//

#ifndef WSREP_DB_CLIENT_SERVICE_HPP
#define WSREP_DB_CLIENT_SERVICE_HPP

#include "wsrep/client_service.hpp"
#include "wsrep/transaction.hpp"

#include "db_client_state.hpp"

namespace db
{
    class client_service : public wsrep::client_service
    {
    public:
        client_service(db::client_state& client_state)
            : wsrep::client_service()
            , client_state_(client_state)
        { }

        bool is_autocommit() const override
        {
            return client_state_.is_autocommit();
        }

        bool do_2pc() const override
        {
            return client_state_.do_2pc();
        }

        bool interrupted() const override
        {
            return false;
        }

        void reset_globals() override
        {
            client_state_.reset_globals();
        }

        void store_globals() override
        {
            client_state_.store_globals();
        }

        int prepare_data_for_replication() override
        {
            return 0;
        }

        void cleanup_transaction() override { }

        size_t bytes_generated() const override
        {
            return 0;
        }

        bool statement_allowed_for_streaming() const override
        {
            return true;
        }
        int prepare_fragment_for_replication(wsrep::mutable_buffer&) override
        {
            return 0;
        }

        void remove_fragments() override
        { }

        int apply_write_set(const wsrep::const_buffer&) override;

        int apply_toi(const wsrep::const_buffer&) override;

        int commit(const wsrep::ws_handle&, const wsrep::ws_meta&) override;

        int rollback() override;

        void will_replay() override
        { }

        void wait_for_replayers(wsrep::unique_lock<wsrep::mutex>&) override { }
        enum wsrep::provider::status replay()
            override;

        int append_fragment(const wsrep::transaction&, int,
                            const wsrep::const_buffer&) override
        {
            return 0;
        }

        void emergency_shutdown() { ::abort(); }
        void debug_sync(const char*) override { }
        void debug_crash(const char*) override { }
    private:
        db::client_state& client_state_;
    };
}

#endif // WSREP_DB_CLIENT_SERVICE_HPP