#include "CClientSession.h" #include "CServer.h" #include #include "PosMessage.h" CClientSession::~CClientSession() { } void CClientSession::start() { 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() { socket_.close(); //注意这里不能调用delete this,因为后面会异步返回,要判断状态,如果这里delete了,后面就会内存越界 } /* *同步发送消息,返回true表示发送成功,返回false表示发送失败 **/ bool CClientSession::send_message(std::string msg) { try { PosMessage new_msg; new_msg.m_length = 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(); //注意这里是同步发送消息,这里不要处理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_INFO("message 非法!"); 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") { //获取到用户名了,进行绑定 m_server->BindUsername(username, this); //绑定完了检查这个用户有没有离线消息 m_server->CheckOfflineMessage(username); } //处理完了,重新读取 socket_.async_read_some(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(); if (socket_.is_open()) { //这个是客户端断开了链接 if (m_username == "") { //说明这个socket还没获取过用户名 delete this; } else { m_server->DeleteClient(m_username); delete this; } } else { //这个是服务端断开了链接 delete this; } } }