add one async http client sample

This commit is contained in:
zhengshuxin 2019-08-08 17:40:16 +08:00
parent e94f0f8a76
commit 7667d4b031
4 changed files with 263 additions and 0 deletions

View File

@ -0,0 +1,5 @@
base_path = ../../..
include ../../Makefile.in
PROG = http_client
#EXTLIBS += -lpolarssl -lz
EXTLIBS += -lz

View File

@ -0,0 +1,243 @@
#include <iostream>
#include <assert.h>
#include <getopt.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include "lib_acl.h"
#include "acl_cpp/lib_acl.hpp"
static acl::atomic_long __aio_refer = 0;
//////////////////////////////////////////////////////////////////////////////
class http_client : public acl::http_aclient
{
public:
http_client(acl::aio_handle& handle, const char* host)
: http_aclient(handle, NULL)
, host_(host)
, debug_(false)
, compressed_(false)
{
++__aio_refer;
}
~http_client(void)
{
printf("delete http_client!\r\n");
if (--__aio_refer == 0) {
printf("%s: stop aio engine now!\r\n", __FUNCTION__);
handle_.stop();
}
}
http_client& enable_debug(bool on)
{
debug_ = on;
return *this;
}
protected:
// @override
void destroy(void)
{
printf("http_client will be deleted!\r\n");
fflush(stdout);
delete this;
}
// @override
bool on_connect(void)
{
printf("--------------- connect server ok ------------\r\n");
fflush(stdout);
printf(">>> begin send_request\r\n");
//this->ws_handshake();
this->send_request(NULL, 0);
return true;
}
// @override
void on_disconnect(void)
{
printf("disconnect from server\r\n");
fflush(stdout);
}
// @override
void on_connect_timeout(void)
{
printf("connect timeout\r\n");
fflush(stdout);
}
// @override
void on_connect_failed(void)
{
printf("connect failed\r\n");
fflush(stdout);
}
// @override
bool on_read_timeout(void)
{
printf("read timeout\r\n");
return true;
}
// @override
bool on_http_res_hdr(const acl::http_header& header)
{
acl::string buf;
header.build_response(buf);
compressed_ = header.is_transfer_gzip();
printf("-----------%s: response header----\r\n", __FUNCTION__);
printf("[%s]\r\n", buf.c_str());
fflush(stdout);
return true;
}
// @override
bool on_http_res_body(char* data, size_t dlen)
{
if (debug_ && (!compressed_ || this->is_unzip_body())) {
(void) write(1, data, dlen);
} else {
printf(">>>read body: %ld\r\n", dlen);
}
return true;
}
// @override
bool on_http_res_finish(bool success)
{
printf("---------------response over-------------------\r\n");
printf("http finish: keep_alive=%s, success=%s\r\n",
keep_alive_ ? "true" : "false",
success ? "ok" : "failed");
fflush(stdout);
return keep_alive_;
}
private:
acl::string host_;
bool debug_;
bool compressed_;
};
static void usage(const char* procname)
{
printf("usage: %s -h[help]\r\n"
" -s server_addr\r\n"
" -D [if in debug mode, default: false]\r\n"
" -c cocorrent\r\n"
" -t connect_timeout[default: 5]\r\n"
" -i rw_timeout[default: 5]\r\n"
" -U url\r\n"
" -H host\r\n"
" -N name_server[default: 8.8.8.8:53]\r\n"
, procname);
}
int main(int argc, char* argv[])
{
int ch, conn_timeout = 5, rw_timeout = 5, cocurrent = 1;
acl::string addr("127.0.0.1:80"), name_server("8.8.8.8:53");
acl::string host("www.baidu.com"), url("/20160528212429_c2HAm.jpeg");
bool debug = false;
while ((ch = getopt(argc, argv, "hc:s:N:U:H:t:i:D")) > 0) {
switch (ch) {
case 'h':
usage(argv[0]);
return (0);
case 'c':
cocurrent = atoi(optarg);
break;
case 's':
addr = optarg;
break;
case 'N':
name_server = optarg;
break;
case 'U':
url = optarg;
break;
case 'H':
host = optarg;
break;
case 't':
conn_timeout = atoi(optarg);
break;
case 'i':
rw_timeout = atoi(optarg);
break;
case 'D':
debug = true;
break;
default:
break;
}
}
acl::acl_cpp_init();
acl::log::stdout_open(true);
// 定义 AIO 事件引擎
acl::aio_handle handle(acl::ENGINE_KERNEL);
//////////////////////////////////////////////////////////////////////
// 设置 DNS 域名服务器地址
handle.set_dns(name_server.c_str(), 5);
// 开始异步连接远程 WEB 服务器
for (int i = 0; i < cocurrent; i++) {
http_client* conn = new http_client(handle, host);
if (!conn->open(addr, conn_timeout, rw_timeout)) {
printf("connect %s error\r\n", addr.c_str());
fflush(stdout);
delete conn;
return 1;
}
(*conn).enable_debug(debug); // 是否启用调试方式
conn->unzip_body(true); // 针对 HTTP 自动解压
// 设置 HTTP 请求头,也可将此过程放在 conn->on_connect() 里
acl::http_header& head = conn->request_header();
head.set_url(url)
.set_host(host)
.accept_gzip(true)
.set_keep_alive(false);
if (i > 0) {
continue;
}
acl::string buf;
head.build_request(buf);
printf("---------------request header-----------------\r\n");
printf("[%s]\r\n", buf.c_str());
fflush(stdout);
}
// 开始 AIO 事件循环过程
while (true) {
// 如果返回 false 则表示不再继续,需要退出
if (!handle.check()) {
break;
}
}
handle.check();
return 0;
}

View File

@ -0,0 +1,12 @@
// stdafx.h : 标准系统包含文件的包含文件,
// 或是常用但不常更改的项目特定的包含文件
//
#pragma once
//#include <iostream>
//#include <tchar.h>
// TODO: 在此处引用程序要求的附加头文件
#include "acl_cpp/lib_acl.hpp"

View File

@ -0,0 +1,3 @@
#!/bin/sh
./http_client -s "pvwu8bubc.bkt.clouddn.com:80" -H "pvwu8bubc.bkt.clouddn.com" -U "/20160528212429_c2HAm.jpeg" -c 10