Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(101)

Unified Diff: net/udp/udp_socket_perftest.cc

Issue 2508593002: net: move udp directory into socket (Closed)
Patch Set: revert sys/socket.h change Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « net/udp/udp_socket.h ('k') | net/udp/udp_socket_posix.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/udp/udp_socket_perftest.cc
diff --git a/net/udp/udp_socket_perftest.cc b/net/udp/udp_socket_perftest.cc
deleted file mode 100644
index e89968ed1db20aa07b1a49b51fa551530cfe2ebf..0000000000000000000000000000000000000000
--- a/net/udp/udp_socket_perftest.cc
+++ /dev/null
@@ -1,137 +0,0 @@
-// Copyright 2015 The Chromium Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style license that can be
-// found in the LICENSE file.
-
-#include "base/bind.h"
-#include "base/memory/weak_ptr.h"
-#include "base/message_loop/message_loop.h"
-#include "base/run_loop.h"
-#include "base/test/perf_time_logger.h"
-#include "net/base/io_buffer.h"
-#include "net/base/ip_endpoint.h"
-#include "net/base/net_errors.h"
-#include "net/base/test_completion_callback.h"
-#include "net/log/net_log_source.h"
-#include "net/test/gtest_util.h"
-#include "net/test/net_test_suite.h"
-#include "net/udp/udp_client_socket.h"
-#include "net/udp/udp_server_socket.h"
-#include "net/udp/udp_socket.h"
-#include "testing/gmock/include/gmock/gmock.h"
-#include "testing/gtest/include/gtest/gtest.h"
-#include "testing/platform_test.h"
-
-using net::test::IsOk;
-
-namespace net {
-
-namespace {
-
-class UDPSocketPerfTest : public PlatformTest {
- public:
- UDPSocketPerfTest()
- : buffer_(new IOBufferWithSize(kPacketSize)), weak_factory_(this) {}
-
- void DoneWritePacketsToSocket(UDPClientSocket* socket,
- int num_of_packets,
- base::Closure done_callback,
- int error) {
- WritePacketsToSocket(socket, num_of_packets, done_callback);
- }
-
- // Send |num_of_packets| to |socket|. Invoke |done_callback| when done.
- void WritePacketsToSocket(UDPClientSocket* socket,
- int num_of_packets,
- base::Closure done_callback);
-
- // Use non-blocking IO if |use_nonblocking_io| is true. This variable only
- // has effect on Windows.
- void WriteBenchmark(bool use_nonblocking_io);
-
- protected:
- static const int kPacketSize = 1024;
- scoped_refptr<IOBufferWithSize> buffer_;
- base::WeakPtrFactory<UDPSocketPerfTest> weak_factory_;
-};
-
-// Creates and address from an ip/port and returns it in |address|.
-void CreateUDPAddress(const std::string& ip_str,
- uint16_t port,
- IPEndPoint* address) {
- IPAddress ip_address;
- if (!ip_address.AssignFromIPLiteral(ip_str))
- return;
- *address = IPEndPoint(ip_address, port);
-}
-
-void UDPSocketPerfTest::WritePacketsToSocket(UDPClientSocket* socket,
- int num_of_packets,
- base::Closure done_callback) {
- scoped_refptr<IOBufferWithSize> io_buffer(new IOBufferWithSize(kPacketSize));
- memset(io_buffer->data(), 'G', kPacketSize);
-
- while (num_of_packets) {
- int rv =
- socket->Write(io_buffer.get(), io_buffer->size(),
- base::Bind(&UDPSocketPerfTest::DoneWritePacketsToSocket,
- weak_factory_.GetWeakPtr(), socket,
- num_of_packets - 1, done_callback));
- if (rv == ERR_IO_PENDING)
- break;
- --num_of_packets;
- }
- if (!num_of_packets) {
- done_callback.Run();
- return;
- }
-}
-
-void UDPSocketPerfTest::WriteBenchmark(bool use_nonblocking_io) {
- base::MessageLoopForIO message_loop;
- const uint16_t kPort = 9999;
-
- // Setup the server to listen.
- IPEndPoint bind_address;
- CreateUDPAddress("127.0.0.1", kPort, &bind_address);
- std::unique_ptr<UDPServerSocket> server(
- new UDPServerSocket(nullptr, NetLogSource()));
- if (use_nonblocking_io)
- server->UseNonBlockingIO();
- int rv = server->Listen(bind_address);
- ASSERT_THAT(rv, IsOk());
-
- // Setup the client.
- IPEndPoint server_address;
- CreateUDPAddress("127.0.0.1", kPort, &server_address);
- std::unique_ptr<UDPClientSocket> client(
- new UDPClientSocket(DatagramSocket::DEFAULT_BIND, RandIntCallback(),
- nullptr, NetLogSource()));
- if (use_nonblocking_io)
- client->UseNonBlockingIO();
- rv = client->Connect(server_address);
- EXPECT_THAT(rv, IsOk());
-
- base::RunLoop run_loop;
- base::TimeTicks start_ticks = base::TimeTicks::Now();
- int packets = 100000;
- client->SetSendBufferSize(1024);
- WritePacketsToSocket(client.get(), packets, run_loop.QuitClosure());
- run_loop.Run();
-
- double elapsed = (base::TimeTicks::Now() - start_ticks).InSecondsF();
- LOG(INFO) << "Write speed: " << packets / 1024 / elapsed << " MB/s";
-}
-
-TEST_F(UDPSocketPerfTest, Write) {
- base::PerfTimeLogger timer("UDP_socket_write");
- WriteBenchmark(false);
-}
-
-TEST_F(UDPSocketPerfTest, WriteNonBlocking) {
- base::PerfTimeLogger timer("UDP_socket_write_nonblocking");
- WriteBenchmark(true);
-}
-
-} // namespace
-
-} // namespace net
« no previous file with comments | « net/udp/udp_socket.h ('k') | net/udp/udp_socket_posix.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698