X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=src%2Fsocket_wrapper.hxx;h=ebb85f6a8ff6000e9ae99cc2e260fbca0f0dd16b;hp=51c0693d1c28bcbf5c4e406e15ba985e964fdaba;hb=2a956e6587338524eec8f66efe3a733a2116eb9b;hpb=e1614a6d1d9c022b83c0aa8cdb948bd7dc98ff23 diff --git a/src/socket_wrapper.hxx b/src/socket_wrapper.hxx index 51c0693..ebb85f6 100644 --- a/src/socket_wrapper.hxx +++ b/src/socket_wrapper.hxx @@ -62,6 +62,11 @@ class BasicSocketWrapper : public ConnectionWrapper { } client_connection* get_connection(void); + + bool connection_established(void) + { return (c.get() != NULL); } + + void set_logging(std::ostream *_logstream, log_level_values _log_level); }; class ReconnectSocketWrapper : public BasicSocketWrapper @@ -79,12 +84,29 @@ class ReconnectSocketWrapper : public BasicSocketWrapper : BasicSocketWrapper(_path,_connect_timeout_usec,_max_retries) { } - void handle(command_client* stubBase, boost::function< void() > f); + bool handle(command_client* stubBase, boost::function< void() > f); }; +class dummy_client_connection : public client_connection +{ + private: + void real_write(const std::string& data) + { } + + public: + dummy_client_connection() + : client_connection() + { close(); } + + bool fill_buffer(long long usec_timeout=-1, long long *usec_timeout_remaining=NULL) + { return false; } +}; class ReconnectIgnoreFailureSocketWrapper : public ReconnectSocketWrapper { + private: + dummy_client_connection dc; + public: ReconnectIgnoreFailureSocketWrapper(int _port, const std::string& _server="127.0.0.1", long long _connect_timeout_usec=socket_client_connection::connect_timeout_usec_default, @@ -99,11 +121,9 @@ class ReconnectIgnoreFailureSocketWrapper : public ReconnectSocketWrapper { } client_connection* get_connection(void); - void handle(command_client* stubBase, boost::function< void() > f); + bool handle(command_client* stubBase, boost::function< void() > f); }; - - } #endif