sqlite_persistent_cookie_store_perftest.cc revision 7dbb3d5cf0c15f500944d211057644d6a2f37371
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "content/browser/net/sqlite_persistent_cookie_store.h" 6c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/compiler_specific.h" 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/scoped_temp_dir.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/perftimer.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/sequenced_task_runner.h" 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/synchronization/waitable_event.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/test/sequenced_worker_pool_owner.h" 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/threading/sequenced_worker_pool.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/cookies/canonical_cookie.h" 17c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/cookies/cookie_constants.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 197dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 21c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)namespace content { 22c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 23c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)namespace { 24c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 25c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const base::FilePath::CharType cookie_filename[] = FILE_PATH_LITERAL("Cookies"); 26c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 27c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} // namespace 28c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SQLitePersistentCookieStorePerfTest : public testing::Test { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SQLitePersistentCookieStorePerfTest() 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : pool_owner_(new base::SequencedWorkerPoolOwner(1, "Background Pool")), 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_(false, false), 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_(false, false) { 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnLoaded(const std::vector<net::CanonicalCookie*>& cookies) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cookies_ = cookies; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_.Signal(); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnKeyLoaded(const std::vector<net::CanonicalCookie*>& cookies) { 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cookies_ = cookies; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_.Signal(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Load() { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->Load(base::Bind(&SQLitePersistentCookieStorePerfTest::OnLoaded, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_.Wait(); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::SequencedTaskRunner> background_task_runner() { 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return pool_owner_->pool()->GetSequencedTaskRunner( 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->GetNamedSequenceToken("background")); 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::SequencedTaskRunner> client_task_runner() { 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return pool_owner_->pool()->GetSequencedTaskRunner( 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->GetNamedSequenceToken("client")); 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void SetUp() OVERRIDE { 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new SQLitePersistentCookieStore( 66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) temp_dir_.path().Append(cookie_filename), 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) client_task_runner(), 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) background_task_runner(), 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) false, NULL); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<net::CanonicalCookie*> cookies; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Load(); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(0u, cookies_.size()); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates 15000 cookies from 300 eTLD+1s. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time t = base::Time::Now(); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int domain_num = 0; domain_num < 300; domain_num++) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string domain_name(base::StringPrintf(".domain_%d.com", domain_num)); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GURL gurl("www" + domain_name); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int cookie_num = 0; cookie_num < 50; ++cookie_num) { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) t += base::TimeDelta::FromInternalValue(10); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->AddCookie( 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::CanonicalCookie(gurl, 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringPrintf("Cookie_%d", cookie_num), "1", 83c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) domain_name, "/", t, t, t, false, false, 84c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::COOKIE_PRIORITY_DEFAULT)); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Replace the store effectively destroying the current one and forcing it 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to write its data to disk. 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = NULL; 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Shut down the pool, causing deferred (no-op) commits to be discarded. 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->Shutdown(); 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ~SequencedWorkerPoolOwner blocks on pool shutdown. 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_.reset(new base::SequencedWorkerPoolOwner(1, "pool")); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new SQLitePersistentCookieStore( 97c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) temp_dir_.path().Append(cookie_filename), 98c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) client_task_runner(), 99c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) background_task_runner(), 100c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) false, NULL); 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void TearDown() OVERRIDE { 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) store_ = NULL; 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->Shutdown(); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<base::SequencedWorkerPoolOwner> pool_owner_; 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WaitableEvent loaded_event_; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WaitableEvent key_loaded_event_; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<net::CanonicalCookie*> cookies_; 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir_; 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SQLitePersistentCookieStore> store_; 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the performance of priority load of cookies for a specfic domain key 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SQLitePersistentCookieStorePerfTest, TestLoadForKeyPerformance) { 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int domain_num = 0; domain_num < 3; ++domain_num) { 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string domain_name(base::StringPrintf("domain_%d.com", domain_num)); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PerfTimeLogger timer( 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ("Load cookies for the eTLD+1 " + domain_name).c_str()); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->LoadCookiesForKey(domain_name, 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&SQLitePersistentCookieStorePerfTest::OnKeyLoaded, 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_.Wait(); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer.Done(); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(50U, cookies_.size()); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the performance of load 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SQLitePersistentCookieStorePerfTest, TestLoadPerformance) { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PerfTimeLogger timer("Load all cookies"); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Load(); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer.Done(); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(15000U, cookies_.size()); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 141c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 142c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} // namespace content 143