IAS/session.cpp at master · geozy/IAS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/*
 * File:   session_impl.cpp
 * Author: georgez
 *
 * Created on 23 October 2014, 14:09
 */

#include "session.h"
#include "task.h"
#include "task_impl.h"
#include "iasServer.h"


session::session(tcp::socket& socket,iasServer* serv)
    : socket_(std::move(socket))
  {
       _pServer=serv;
       std::cout << "Session (Default Type) Created" << std::endl;
  }

session::~session(){
       std::cout << "Session Destructed" << std::endl;
   }

  void session::start()
  {
        do_read();
  }

  void session::addTask(task* tp){
      _pServer->addTask(tp);
  }
   void session::recycleTask(task* tp){
      _pServer->recycleTask(tp);
  }

   task* session::fetchTask(std::shared_ptr<session> ps){
       return _pServer->fetchTask(ps);
   }


  void session::do_read()
  {
    auto self(shared_from_this());

    task* ptask=fetchTask(self);

    socket_.async_read_some(boost::asio::buffer(ptask->_pti->_data, max_length),
        [this, self,ptask](boost::system::error_code ec, std::size_t length)
        {
          bool killTask=true;
          if (!ec)
          {
             ptask->_pti->setLength(length);
             {
                    addTask(ptask);
                    killTask=false;

                    do_read();
             }
          }else{
              std::cout << ec.message() << std::endl;
          }

          // If there is a failure, the task is recycled
          if(killTask){
              recycleTask(ptask);
          }
        });
  }

  tcp::socket* session::getSocket(){
      return &socket_;
  }

  void session::writeResponse(task* ptask){
      // prep the header for listeners
      auto header = ptask->getImpl()->getHeader();
      header->resp=1;

      // send the data - blocking
      boost::asio::write(socket_,boost::asio::buffer(header,header->length));
  }

  void session::writeToListener(task* ptask){
      // prep the header for listeners
      auto header = ptask->getImpl()->getHeader();
      header->resp=0;

      // send the data - blocking
      boost::asio::write(socket_,boost::asio::buffer(header,header->length));
  }

  std::shared_ptr<iaService> session::getService(){
      return _pService;
  }