2 * Copyright 2017-present Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include <folly/Benchmark.h>
18 #include <folly/io/async/test/ZeroCopy.h>
19 #include <folly/portability/GFlags.h>
21 using namespace folly;
24 const std::string& host,
29 LOG(INFO) << "Running client. host = " << host << " port = " << port
30 << " numLoops = " << numLoops << " zeroCopy = " << zeroCopy
31 << " bufferSize = " << bufferSize;
34 std::unique_ptr<ZeroCopyTestAsyncSocket> client(
35 new ZeroCopyTestAsyncSocket(&evb, numLoops, bufferSize, zeroCopy));
36 SocketAddress addr(host, port);
37 evb.runInEventBaseThread([&]() { client->connect(addr); });
42 void runServer(uint16_t port, int numLoops, bool zeroCopy, size_t bufferSize) {
43 LOG(INFO) << "Running server. port = " << port << " numLoops = " << numLoops
44 << " zeroCopy = " << zeroCopy << " bufferSize = " << bufferSize;
47 folly::AsyncServerSocket::UniquePtr listenSock(
48 new folly::AsyncServerSocket(&evb));
49 ZeroCopyTestServer server(&evb, numLoops, bufferSize, zeroCopy);
51 server.addCallbackToServerSocket(*listenSock);
53 evb.runInEventBaseThread([&]() {
54 listenSock->bind(port);
55 listenSock->setZeroCopy(zeroCopy);
56 listenSock->listen(10);
57 listenSock->startAccepting();
64 static auto constexpr kMaxLoops = 20000;
66 void zeroCopyOn(unsigned /* unused */, size_t bufferSize) {
67 ZeroCopyTest test(kMaxLoops, true, bufferSize);
71 void zeroCopyOff(unsigned /* unused */, size_t bufferSize) {
72 ZeroCopyTest test(kMaxLoops, false, bufferSize);
76 BENCHMARK_PARAM(zeroCopyOn, 4096);
77 BENCHMARK_PARAM(zeroCopyOff, 4096);
79 BENCHMARK_PARAM(zeroCopyOn, 8192);
80 BENCHMARK_PARAM(zeroCopyOff, 8192);
82 BENCHMARK_PARAM(zeroCopyOn, 16384);
83 BENCHMARK_PARAM(zeroCopyOff, 16384);
85 BENCHMARK_PARAM(zeroCopyOn, 32768);
86 BENCHMARK_PARAM(zeroCopyOff, 32768);
88 BENCHMARK_PARAM(zeroCopyOn, 65536);
89 BENCHMARK_PARAM(zeroCopyOff, 65536);
91 BENCHMARK_PARAM(zeroCopyOn, 131072);
92 BENCHMARK_PARAM(zeroCopyOff, 131072);
94 BENCHMARK_PARAM(zeroCopyOn, 262144);
95 BENCHMARK_PARAM(zeroCopyOff, 262144);
97 BENCHMARK_PARAM(zeroCopyOn, 524288);
98 BENCHMARK_PARAM(zeroCopyOff, 524288);
100 BENCHMARK_PARAM(zeroCopyOn, 1048576);
101 BENCHMARK_PARAM(zeroCopyOff, 1048576);
102 BENCHMARK_DRAW_LINE()
104 DEFINE_bool(client, false, "client mode");
105 DEFINE_bool(server, false, "server mode");
106 DEFINE_bool(zeroCopy, false, "use zerocopy");
107 DEFINE_int32(numLoops, kMaxLoops, "number of loops");
108 DEFINE_int32(bufferSize, 524288, "buffer size");
109 DEFINE_int32(port, 33130, "port");
110 DEFINE_string(host, "::1", "host");
112 int main(int argc, char** argv) {
113 gflags::ParseCommandLineFlags(&argc, &argv, true);
122 } else if (FLAGS_server) {
123 runServer(FLAGS_port, FLAGS_numLoops, FLAGS_zeroCopy, FLAGS_bufferSize);