zhipuzi_pos_windows/zhipuzi_pos_windows_server/network/CClientSession.cpp

#include "CClientSession.h"

#include "CServer.h"
#include <sqlite3/sqlite3.h>

#include "PosMessage.h"

CClientSession::~CClientSession()
{
	
}

void CClientSession::start()
{
	m_pos_message.clear();

	boost::asio::async_read(socket_, boost::asio::buffer(&m_pos_message, sizeof(PosMessage)),
		boost::bind(&CClientSession::handle_read, this,
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
}

/*
 *服务器端主动关闭，才能调用这个函数
 **/
void CClientSession::stop()
{
	//先给客户端发送一个下线通知
	rapidjson::Document doc;
	doc.SetObject();
	rapidjson::Document::AllocatorType& allocator = doc.GetAllocator(); //获取分配器

	std::string timestamp = to_string(time(NULL));

	doc.AddMember("msg_type", "0", allocator);
	doc.AddMember("timestamp", StringRef(timestamp.c_str(), timestamp.length()), allocator);

	rapidjson::StringBuffer buffer;
	rapidjson::Writer<StringBuffer> writer(buffer);
	doc.Accept(writer);

	//返回给接入层的消息
	std::string login_out_msg = buffer.GetString();

	bool ret = send_message(login_out_msg);

	if (ret)
	{
		//然后关闭socket
		//LOG_INFO("in stop, send_message success!");
		socket_.close();
	}
	else
	{
		//LOG_ERROR("in stop, send_message failed!");
		//如果同步写数据失败，说明socket已经失效了，这里不要调用关闭操作，等待异步读返回错误就好
	}

	//注意这里不能调用delete this，因为后面会异步返回，要判断状态，如果这里delete了，后面就会内存越界
}

/*
 *同步发送消息，返回true表示发送成功，返回false表示发送失败
 **/
bool CClientSession::send_message(std::string msg)
{
	try
	{
		PosMessage new_msg;
		new_msg.m_length = (int)msg.length();
		memcpy(new_msg.data, msg.c_str(), msg.length());
		new_msg.data[new_msg.m_length] = '\0';

		socket_.write_some(boost::asio::buffer(&new_msg, sizeof(PosMessage)));
	}
	catch (const std::exception& e)
	{
		std::string err = e.what();

		//LOG_ERROR("send_message error, message:" << err.c_str());

		//注意这里是同步发送消息，这里不要处理socket的关闭和delete操作，让后面的异步读错误处理
		return false;
	}

	return true;
}

void CClientSession::SetServer(CServer* server)
{
    m_server = server;
}

void CClientSession::handle_read(const boost::system::error_code& error,
                                 size_t bytes_transferred)
{
    if(!error)
    {
        //读到客户端发过来的内容，进行绑定
        std::string msg = m_pos_message.data;

        rapidjson::Document document;
        document.Parse(msg.c_str());
        if(!document.IsObject())
        {
            LOG_ERROR("message 非法！ msg:" << msg.c_str());
			LOG_ERROR("msg length:" << m_pos_message.m_length);
			LOG_ERROR("ip:" << socket_.remote_endpoint().address().to_string());

			//处理完了，重新读取
			m_pos_message.clear();

			boost::asio::async_read(socket_, boost::asio::buffer(&m_pos_message, sizeof(PosMessage)),
				boost::bind(&CClientSession::handle_read, this,
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred));

            return;
        }

        std::string username = document["username"].GetString();
		std::string timestamp = document["timestamp"].GetString();
		std::string is_login = document["is_login"].GetString();

		//这个用于判断是否成功绑定用户名了
		m_username = username;

		if (is_login == "1")
		{
			//LOG_INFO("get client login message:" << msg.c_str());

			//获取到用户名了，进行绑定
			m_server->BindUsername(username, this);

			//绑定完了检查这个用户有没有离线消息
			m_server->CheckOfflineMessage(username);
		}

        //处理完了，重新读取
		m_pos_message.clear();

		boost::asio::async_read(socket_, boost::asio::buffer(&m_pos_message, sizeof(PosMessage)),
			boost::bind(&CClientSession::handle_read, this,
				boost::asio::placeholders::error,
				boost::asio::placeholders::bytes_transferred));
    }
    else
    {
		std::string err = error.message();
		//LOG_ERROR("handle_read error, err message:" << err.c_str());

		if (socket_.is_open())
		{
			socket_.close();

			//这个是客户端断开了链接
			if (m_username == "")
			{
				//说明这个socket还没获取过用户名，直接销毁对象即可
				LOG_ERROR("111 handle_read error, err message:" << err.c_str());
				delete this;
			}
			else
			{
				//说明这个socket已经绑定过用户名了，先从map中删除，再销毁对象
				LOG_ERROR("222 handle_read error, err message:" << err.c_str());
				LOG_ERROR("222 username:" << m_username.c_str());
				m_server->DeleteClient(m_username);

				delete this;
			}
		}
		else
		{
			//这个是服务端断开了链接，已经调用过close了
			//LOG_INFO("handle_read error，服务端断开了链接");
			delete this;
		}
    }
}