PGenericSocket.h

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#ifndef __PGENERIC_SOCKET_H__
#define __PGENERIC_SOCKET_H__
 
#include <unistd.h>
 
#include "PSocketMode.h"
#include "PSocketFlag.h"
#include "phoenix_mock_socket.h"

template<typename _TBackend, typename _TMockBackend>
class PGenericSocket{
        public:
                PGenericSocket(PSocketMode::PSocketMode mode);
                virtual ~PGenericSocket();
        
                bool createClientSocket(_TBackend & backend, _TMockBackend & mockBackend, const PSocketParam & socketParam, const typename _TBackend::Param & param, const std::string & mockPrefix, const typename _TMockBackend::Param & mockParam);
                bool createServerSocket(_TBackend & backend, _TMockBackend & mockBackend, const PSocketParam & socketParam, const typename _TBackend::Param & param, const std::string & mockPrefix, const typename _TMockBackend::Param & mockParam);
                
                void setMode(PSocketMode::PSocketMode mode);
                

                template<typename U>
                PSendStatus::PSendStatus sendData(const U & data, PSendFlag::PSendFlag flag){
                        PSendStatus::PSendStatus sendStatus = PSendStatus::OK;
                        if(p_mode != PSocketMode::NO_MOCK){
                                sendStatus = p_mockSocket.sendData(data, flag);
                        }
                        // If we dont test if sendStatus is OK, we will always send the message even if the mock backend is not connected
                        // I don't know if it is a good idea to do so
                        // If we do that, we might have somme issues if the user tries to end a message. The mock may give an error that will not be catch 
                        // Because the boolean value is not checked. So the real backend could have sent corretly the message and the mock not....
 
                        if(p_mode != PSocketMode::MOCK && sendStatus == PSendStatus::OK){
                                sendStatus = p_socket.sendData(data, flag);
                        }
                        return sendStatus;
                }
                PSendStatus::PSendStatus sendMsg(typename _TBackend::Message & msg, PSendFlag::PSendFlag flag);
                

                template<typename U>
                PRecvStatus::PRecvStatus recvData(U & data, PRecvFlag::PRecvFlag flag){
                        PRecvStatus::PRecvStatus recvStatus = PRecvStatus::OK;
                        if(p_mode == PSocketMode::NO_MOCK){   //Normal mode
                                recvStatus = p_socket.recvData(data, flag);
                        }else if(p_mode == PSocketMode::MOCK){   //Mock mode
                                recvStatus = p_mockSocket.recvData(data, flag);
                        }else{                                  //Mock record mode
                                recvStatus = p_socket.recvData(data, flag);
                                //If no message was recieved, we record it anyway
                                //Maybe we do not want to erase the status of the real socket with the mock backend, so no recvStatus = p_mockSocket.recvData(data, flag);
                                p_mockSocket.recvData(data, flag);
                        }
                        return recvStatus;
                }
                
                PRecvStatus::PRecvStatus recvMsg(typename _TBackend::Message & msg, PRecvFlag::PRecvFlag flag);
                
                void close();
                bool isConnected() const;
                bool waitUntilConnection(uint64_t refreshTimer, size_t nbRetry) const;
        private:
                void initialisationPGenericSocket(PSocketMode::PSocketMode mode);
                
                PSocketMode::PSocketMode p_mode;
                
                typename _TBackend::Socket p_socket;
                typename _TMockBackend::Socket p_mockSocket;
};
 
#include "PGenericSocket_impl.h"
 
 
#endif