TCPSession.h

#ifndef __Session_h__
# define __Session_h__
 
#include "TCPSocket.h"
#include "SessionFilter.h"
 
namespace core {
 
   class Command;
   enum Mode {LINE, BLOCK, PACKET};
   
   class TCPServer;
   
   
   class TCPSession : public TCPSocket {
      
    public:
 
      
      TCPSession(EPoll &ePoll, TCPServer &server, std::string text = "");
 
      
      ~TCPSession(); 
      
      Command *grab = NULL;
      
      virtual void output(std::stringstream &data);
 
      
      std::stringstream out;
      
      
      void send();
      
   
      void sendToAll();
      
      
      void sendToAll(SessionFilter filter);
 
      
      void terminate();
      
      
      TCPServer &server;
 
      
//      enum Mode {LINE, BLOCK};
            
    protected:
 
 
      virtual void onRegistered() override;    
 
      
      virtual void onDataReceived(char *data, int len) override;      
      
      
      virtual void onLineReceived(std::string line); 
      
      
      virtual void onBlockReceived(std::string block); 
 
      
      virtual void onConnected();
      
      
      virtual void protocol(std::string data); 
 
      
      void setMode(core::Mode mode, int size = 0); 
      
    private:
      char *lineBuffer = NULL;
      int lineBufferSize = 0;
      int lineLength = 0;
      int blockLength = 0;
      std::mutex mtx;
      bool term = false;
      core::Mode mode = LINE;
      int blockSize;
      
   };
   
}
 
#endif