Interceptors.h

#pragma once
 
#include <kafka/Project.h>
 
#include <functional>
 
 
namespace KAFKA_API { namespace clients {
 
class Interceptors
{
public:
    using ThreadStartCb = std::function<void(const std::string&, const std::string&)>;
 
    using ThreadExitCb  = std::function<void(const std::string&, const std::string&)>;
 
    using BrokerStateChangeCb = std::function<void(int, const std::string&, const std::string&, int, const std::string&)>;
 
    Interceptors& onThreadStart(ThreadStartCb cb) { _valid = true; _threadStartCb = std::move(cb); return *this; }
 
    Interceptors& onThreadExit(ThreadExitCb cb)   { _valid = true; _threadExitCb = std::move(cb);  return *this; }
 
    Interceptors& onBrokerStateChange(BrokerStateChangeCb cb) { _valid = true; _brokerStateChangeCb = std::move(cb);  return *this; }
 
    ThreadStartCb onThreadStart() const { return _threadStartCb; }
 
    ThreadExitCb  onThreadExit()  const { return _threadExitCb; }
 
    BrokerStateChangeCb onBrokerStateChange() const { return _brokerStateChangeCb; }
 
    bool empty() const { return !_valid; }
 
private:
    ThreadStartCb       _threadStartCb;
    ThreadExitCb        _threadExitCb;
    BrokerStateChangeCb _brokerStateChangeCb;
 
    bool _valid  = false;
};
 
} } // end of KAFKA_API::clients