test/acl
1
0
Fork 0
mirror of https://gitee.com/acl-dev/acl.git synced 2024-12-15 09:20:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'gitee-master' into gitlab-upstream

Browse Source
This commit is contained in:
zhengshuxin 2023-06-12 11:56:09 +08:00
commit 7d2dd15b80
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib_fiber/samples-c++1x/fiber_pool/main.cpp
View File

@ -53,14 +53,16 @@ static void usage(const char* procname) {
" -s ip:port, default: 127.0.0.1:9001\r\n" " -s ip:port, default: 127.0.0.1:9001\r\n"
" -c fiber_pool_count [default: 100] \r\n" " -c fiber_pool_count [default: 100] \r\n"
" -r timeout\r\n" " -r timeout\r\n"
" -S [if in sync mode, default: false]\r\n"
, procname); , procname);
} }
int main(int argc, char* argv[]) { int main(int argc, char* argv[]) {
acl::string addr("127.0.0.1:9001"); acl::string addr("127.0.0.1:9001");
bool sync = false;
int ch, nfibers = 100; int ch, nfibers = 100;
while ((ch = getopt(argc, argv, "hs:c:r:")) > 0) { while ((ch = getopt(argc, argv, "hs:Sc:r:")) > 0) {
switch (ch) { switch (ch) {
case 'h': case 'h':
usage(argv[0]); usage(argv[0]);
@ -68,6 +70,9 @@ int main(int argc, char* argv[]) {
case 's': case 's':
addr = optarg; addr = optarg;
break; break;
case 'S':
sync = true;
break;
case 'c': case 'c':
nfibers = atoi(optarg); nfibers = atoi(optarg);
break; break;
@ -92,7 +97,11 @@ int main(int argc, char* argv[]) {
for (int i = 0; i < nfibers; i++) { for (int i = 0; i < nfibers; i++) {
go[&box] { go[&box] {
while (true) { while (true) {
shared_message msg = box.pop(); shared_message msg;
if (!box.pop(msg)) {
printf("POP end!\r\n");
break;
}
auto client = msg->get_client(); auto client = msg->get_client();
auto data = msg->get_data(); auto data = msg->get_data();
if (client->get_conn().write(data.c_str(), data.size()) == -1) { if (client->get_conn().write(data.c_str(), data.size()) == -1) {
@ -112,7 +121,7 @@ int main(int argc, char* argv[]) {
} }
}; };
go[&ss, &box, &nusers, &nmsgs] { go[&ss, &box, &nusers, &nmsgs, sync] {
while (true) { while (true) {
auto conn = ss.accept(); auto conn = ss.accept();
if (conn == NULL) { if (conn == NULL) {
@ -124,7 +133,7 @@ int main(int argc, char* argv[]) {
auto client = std::make_shared<client_socket>(conn, nusers); auto client = std::make_shared<client_socket>(conn, nusers);
go[&box, &nmsgs, client] { go[&box, &nmsgs, client, sync] {
char buf[4096]; char buf[4096];
while (true) { while (true) {
int ret = client->get_conn().read(buf, sizeof(buf), false); int ret = client->get_conn().read(buf, sizeof(buf), false);
@ -132,13 +141,14 @@ int main(int argc, char* argv[]) {
break; break;
} }
#if 0 if (sync) {
if (client->get_conn().write(buf, ret) != ret) { if (client->get_conn().write(buf, ret) != ret) {
break; break;
} else { } else {
continue; continue;
} }
#endif }
++nmsgs; ++nmsgs;
auto msg = std::make_shared<message>(client, nmsgs, buf, ret); auto msg = std::make_shared<message>(client, nmsgs, buf, ret);
box.push(msg); box.push(msg);