Add IO benchmark test demo.

This commit is contained in:
shuxin   zheng 2024-10-30 16:25:35 +08:00
parent 1fc4f9a676
commit 8464223071
4 changed files with 157 additions and 0 deletions

View File

@ -60,8 +60,24 @@ long long get_curr_stamp(void)
double stamp_sub(const struct timeval& from, const struct timeval& sub)
{
#if 0
struct timeval res;
memcpy(&res, &from, sizeof(struct timeval));
res.tv_usec -= sub.tv_usec;
if (res.tv_usec < 0) {
--res.tv_sec;
res.tv_usec += 1000000;
}
res.tv_sec -= sub.tv_sec;
return res.tv_sec * 1000.0 + res.tv_usec / 1000.0;
#else
return (from.tv_sec - sub.tv_sec) * 1000
+ (from.tv_usec - sub.tv_usec) / 1000;
#endif
}
} // namespace acl

View File

@ -0,0 +1,3 @@
include ../Makefile_cpp.in
CFLAGS += -std=c++11
PROG = client

View File

@ -0,0 +1,111 @@
#include "stdafx.h"
static bool use_sockopt = false;
static std::atomic<long long> total_count(0);
static void fiber_main(const char *addr, int count, int timeo) {
acl::socket_stream conn;
if (!conn.open(addr, timeo, timeo)) {
printf("Connect %s error %s\r\n", addr, acl::last_serror());
return;
}
conn.set_rw_timeout(timeo, use_sockopt);
printf("Connect %s ok, fd=%d\r\n", addr, conn.sock_handle());
const char s[] = "hello world!\r\n";
char buf[1024];
int i;
for (i = 0; i < count; i++) {
if (conn.write(s, sizeof(s) - 1) == -1) {
printf("write error %s\r\n", acl::last_serror());
break;
}
if (conn.read(buf, sizeof(buf) -1, false) == -1) {
printf("read error %s\r\n", acl::last_serror());
break;
}
}
total_count += i;
}
static void thread_main(const char *addr, int nfibers, int count , int timeo) {
for (int j = 0; j < nfibers; j++) {
go[=] {
fiber_main(addr, count, timeo);
};
}
acl::fiber::schedule_with(acl::FIBER_EVENT_T_KERNEL);
}
static void usage(const char *procname) {
printf("usage: %s -h [help]\r\n"
" -s server_addr[default: 127.0.0.1:9001]\r\n"
" -k cpus[default: 1]\r\n"
" -c fibers count[default: 100]\r\n"
" -o io timeout[default: -1]\r\n"
" -O [if use setsockopt to set IO timeout, default: false]\r\n"
, procname);
}
int main(int argc, char *argv[]) {
int ch, cpus = 1, count = 10000, nfibers = 100, timeo = -1;
std::string addr("127.0.0.1:9001");
while ((ch = getopt(argc, argv, "hs:k:c:o:O")) > 0) {
switch (ch) {
case 'h':
usage(argv[0]);
return 0;
case 's':
addr = optarg;
break;
case 'k':
cpus = atoi(optarg);
break;
case 'c':
nfibers = atoi(optarg);
break;
case 'o':
timeo = atoi(optarg);
break;
case 'O':
use_sockopt = true;
break;
default:
usage(argv[0]);
return 1;
}
}
struct timeval begin;
gettimeofday(&begin, nullptr);
std::vector<std::thread*> threads;
for (int i = 0; i < cpus; i++) {
std::thread *thr = new std::thread([&]{
thread_main(addr.c_str(), nfibers, count, timeo);
});
threads.push_back(thr);
}
for (auto thr : threads) {
thr->join();
delete thr;
}
struct timeval end;
gettimeofday(&end, nullptr);
double tc = acl::stamp_sub(end, begin);
double speed = (total_count * 1000) / (tc > 0 ? tc : 0.0001);
printf("Total count=%lld, tc=%.2f ms, speed=%.2f\r\n",
total_count.load(), tc, speed);
return 0;
}

View File

@ -0,0 +1,27 @@
/**
* Copyright (C) 2015-2018 IQIYI
* All rights reserved.
*
* AUTHOR(S)
* Zheng Shuxin
* E-mail: zhengshuxin@qiyi.com
*
* VERSION
* 10/30 15:30:05 2024
*/
#pragma once
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <sys/time.h>
#include <string>
#include <atomic>
#include <vector>
#include <thread>
#include "lib_acl.h"
#include "acl_cpp/lib_acl.hpp"
#include "fiber/fiber.hpp"
#include "fiber/go_fiber.hpp"