acl/lib_acl_cpp/samples/master_aio_proxy/res_callback.cpp

#include "stdafx.h"
#include "req_callback.h"
#include "res_callback.h"

res_callback::res_callback(req_callback* req, acl::ofstream& res_fp)
: conn_(NULL)
, req_(req)
, res_fp_(res_fp)
{

}

res_callback::~res_callback()
{
	acl_assert(req_ == NULL);
}

bool res_callback::read_callback(char* data, int len)
{
	if (req_ == NULL)
	{
		logger_warn("client peer disconnected!");
		return false;
	}

	// 取得客户端连接流，并将从服务端发来的数据写入
	acl::aio_socket_stream& peer = req_->get_conn();
	peer.write(data, len);

	// 将数据同时写入本地响应数据文件
	res_fp_.write(data, len);

	return true;
}

void res_callback::close_callback()
{
	if (req_)
	{
		req_callback* req = req_;
		req_ = NULL;
		req->disconnect();
	}

	// 必须在此处删除该动态分配的回调类对象以防止内存泄露  
	delete this;  
}

bool res_callback::open_callback()
{
	// 连接成功，设置IO读写回调函数
	conn_->add_read_callback(this);
	conn_->add_write_callback(this);

	// 等待从服务器读取数据
	conn_->read();

	// 通知请求类已经与服务器建立连接
	acl_assert(req_);
	req_->on_connected();
	return true;
}

bool res_callback::start(acl::aio_handle& handle, const char* server_addr)
{
	conn_ = acl::aio_socket_stream::open(&handle, server_addr, 0);
	if (conn_ == NULL)
	{
		logger_error("connect server %s error %s",
			server_addr, acl::last_serror());
		return false;
	}

	conn_->add_open_callback(this);
	conn_->add_timeout_callback(this);
	conn_->add_close_callback(this);

	return true;
}

acl::aio_socket_stream& res_callback::get_conn()
{
	acl_assert(conn_);
	return *conn_;
}

void res_callback::disconnect()
{
	if (conn_)
	{
		req_ = NULL;
		conn_->close();
	}
	else
		delete this;
}