/TGame/TServerGate/Network/DataSession.cpp

http://awoe.googlecode.com/ · C++ · 96 lines · 79 code · 16 blank · 1 comment · 5 complexity · 73736711fa7eb5b785cb5d448bcd380a MD5 · raw file 

    

  1. #include "stdafx.h"

    2. 

  2. #include "DataSession.h"

    3. 

  3. #include <iostream>

    4. 

  4. #include "boost/bind.hpp"

    5. 

  5. 

    6. 

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

    7. 

  7. :m_theSvcIO(svcIO), m_theSocket(svcIO), m_wpMsgBufferFactory(NULL)

    8. 

  8. {

    9. 

  9. }

    10. 

  10. 

    11. 

  11. int

    12. 

  12. DataSession::getID()const

    13. 

  13. {

    14. 

  14. 	return m_nSesId;

    15. 

  15. }

    16. 

  16. 

    17. 

  17. void

    18. 

  18. DataSession::setID(int nID)

    19. 

  19. {

    20. 

  20. 	m_nSesId = nID;

    21. 

  21. }

    22. 

  22. 

    23. 

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

    24. 

  24. DataSession::getSocket()

    25. 

  25. {

    26. 

  26. 	return m_theSocket;

    27. 

  27. }

    28. 

  28. 

    29. 

  29. bool

    30. 

  30. DataSession::read(MsgBuffer* pStream)

    31. 

  31. {

    32. 

  32. 	if (pStream==NULL)

    33. 

  33. 	{

    34. 

  34. 		pStream = m_wpMsgBufferFactory->malloc();

    35. 

  35. 	}

    36. 

  36. 

    37. 

  37. 	if (pStream)

    38. 

  38. 	{

    39. 

  39. 		boost::asio::async_read(m_theSocket,

    40. 

  40. 			pStream->prepare(1024),

    41. 

  41. 			boost::bind(&DataSession::onInput, this, boost::asio::placeholders::error, pStream));

    42. 

  42. 	}

    43. 

  43. 

    44. 

  44. 	return true;

    45. 

  45. }

    46. 

  46. 

    47. 

  47. 

    48. 

  48. bool

    49. 

  49. DataSession::write(MsgBuffer* pStream)

    50. 

  50. {

    51. 

  51. 	if (pStream)

    52. 

  52. 	{

    53. 

  53. 		boost::asio::async_write(m_theSocket,

    54. 

  54. 			pStream->data(),

    55. 

  55. 			boost::bind(&DataSession::onOutput, this, boost::asio::placeholders::error, pStream));

    56. 

  56. 	}

    57. 

  57. 

    58. 

  58. 	return true;

    59. 

  59. }

    60. 

  60. 

    61. 

  61. void

    62. 

  62. DataSession::setDataStreamPool(MsgBufferFactory* pMsgBuffFact)

    63. 

  63. {

    64. 

  64. 	m_wpMsgBufferFactory = pMsgBuffFact;

    65. 

  65. }

    66. 

  66. 

    67. 

  67. bool

    68. 

  68. DataSession::onUpdate(const int& nElapse)

    69. 

  69. {

    70. 

  70. 	return true;

    71. 

  71. }

    72. 

  72. 

    73. 

  73. void 

    74. 

  74. DataSession::onInput(const boost::system::error_code& error, MsgBuffer* pStream )

    75. 

  75. {

    76. 

  76. 	if (!error)

    77. 

  77. 	{

    78. 

  78. 		

    79. 

  79. 

    80. 

  80. 		read();

    81. 

  81. 	}

    82. 

  82. 	else

    83. 

  83. 	{

    84. 

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

    85. 

  85. 	}

    86. 

  86. }

    87. 

  87. 

    88. 

  88. void

    89. 

  89. DataSession::onOutput(const boost::system::error_code& error, MsgBuffer* pStream )

    90. 

  90. {

    91. 

  91. 	if (pStream)

    92. 

  92. 	{

    93. 

  93. 		//	recycle

    94. 

  94. 		m_wpMsgBufferFactory->free(pStream);

    95. 

  95. 	}

    96. 

  96. }
    97.