/TGame/TCommon/Service/Session.cpp

http://awoe.googlecode.com/ · C++ · 118 lines · 97 code · 19 blank · 2 comment · 5 complexity · 2594de30fdfa3b67ff67c2429c7668c1 MD5 · raw file 

    

  1. #include "stdafx.h"

    2. 

  2. #include "Session.h"

    3. 

  3. #include <iostream>

    4. 

  4. #include "boost/bind.hpp"

    5. 

  5. 

    6. 

  6. Session::Session( boost::asio::io_service &svcIO)

    7. 

  7. :m_theSvcIO(svcIO), 

    8. 

  8. m_theSocket(svcIO), 

    9. 

  9. m_wpMsgBufferFactory(NULL), 

    10. 

  10. m_nSesId(0), 

    11. 

  11. m_nEttId(0)

    12. 

  12. {

    13. 

  13. }

    14. 

  14. 

    15. 

  15. int

    16. 

  16. Session::getID()const

    17. 

  17. {

    18. 

  18. 	return m_nSesId;

    19. 

  19. }

    20. 

  20. 

    21. 

  21. void

    22. 

  22. Session::setID(int nID)

    23. 

  23. {

    24. 

  24. 	m_nSesId = nID;

    25. 

  25. }

    26. 

  26. 

    27. 

  27. int

    28. 

  28. Session::getSID()const

    29. 

  29. {

    30. 

  30. 	return m_nEttId;

    31. 

  31. }

    32. 

  32. 

    33. 

  33. void

    34. 

  34. Session::setSID(int nID)

    35. 

  35. {

    36. 

  36. 	m_nEttId = nID;

    37. 

  37. }

    38. 

  38. 

    39. 

  39. boost::asio::ip::tcp::socket&

    40. 

  40. Session::getSocket()

    41. 

  41. {

    42. 

  42. 	return m_theSocket;

    43. 

  43. }

    44. 

  44. 

    45. 

  45. bool

    46. 

  46. Session::read(MsgBuffer* pBuffer)

    47. 

  47. {

    48. 

  48. 	if (pBuffer==NULL)

    49. 

  49. 	{

    50. 

  50. 		pBuffer = new MsgBuffer;

    51. 

  51. 	}

    52. 

  52. 

    53. 

  53. 	if (pBuffer)

    54. 

  54. 	{

    55. 

  55. 		m_theSocket.async_read_some(pBuffer->prepare(),

    56. 

  56. 			boost::bind(&Session::onInput, this, boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred, pBuffer));

    57. 

  57. 	}

    58. 

  58. 

    59. 

  59. 	return true;

    60. 

  60. }

    61. 

  61. 

    62. 

  62. 

    63. 

  63. bool

    64. 

  64. Session::write(MsgBuffer* pBuffer)

    65. 

  65. {

    66. 

  66. 	if (pBuffer)

    67. 

  67. 	{

    68. 

  68. 		pBuffer->update_buffer_length4();

    69. 

  69. 

    70. 

  70. 		m_theSocket.async_write_some(pBuffer->data(),

    71. 

  71. 			boost::bind(&Session::onOutput, this, boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred, pBuffer));

    72. 

  72. 	}

    73. 

  73. 

    74. 

  74. 	return true;

    75. 

  75. }

    76. 

  76. 

    77. 

  77. bool

    78. 

  78. Session::write(IMsg* pMsg)

    79. 

  79. {

    80. 

  80. 	return false;

    81. 

  81. }

    82. 

  82. 

    83. 

  83. void

    84. 

  84. Session::setDataStreamPool(MsgBufferFactory* pMsgBuffFact)

    85. 

  85. {

    86. 

  86. 	m_wpMsgBufferFactory = pMsgBuffFact;

    87. 

  87. }

    88. 

  88. 

    89. 

  89. bool

    90. 

  90. Session::onUpdate(const int& nElapse)

    91. 

  91. {

    92. 

  92. 	return true;

    93. 

  93. }

    94. 

  94. 

    95. 

  95. void 

    96. 

  96. Session::onInput(const boost::system::error_code& error, size_t bytes_transferred, MsgBuffer* pBuffer )

    97. 

  97. {

    98. 

  98. 	if (!error)

    99. 

  99. 	{

    100. 

  100. 		read();

    101. 

  101. 	}

    102. 

  102. 	else

    103. 

  103. 	{

    104. 

  104. 		std::cout<<"Error!"<<std::endl;

    105. 

  105. 	}

    106. 

  106. }

    107. 

  107. 

    108. 

  108. void

    109. 

  109. Session::onOutput(const boost::system::error_code& error, size_t bytes_transferred, MsgBuffer* pBuffer )

    110. 

  110. {

    111. 

  111. 	if (pBuffer)

    112. 

  112. 	{

    113. 

  113. 		//	recycle

    114. 

  114. 		//m_wpMsgBufferFactory->free(buffer);

    115. 

  115. 

    116. 

  116. 		delete pBuffer;

    117. 

  117. 	}

    118. 

  118. }
    119.