Use WTF::Function instead of std::function in NetworkProcess code
[WebKit-https.git] / Source / WebKit2 / NetworkProcess / cache / NetworkCacheStatistics.h
1 /*
2  * Copyright (C) 2015 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #ifndef NetworkCacheStatistics_h
27 #define NetworkCacheStatistics_h
28
29 #if ENABLE(NETWORK_CACHE)
30
31 #include "NetworkCache.h"
32 #include "NetworkCacheKey.h"
33 #include <WebCore/SQLiteDatabase.h>
34 #include <WebCore/Timer.h>
35 #include <wtf/WorkQueue.h>
36
37 namespace WebCore {
38 class ResourceRequest;
39 }
40
41 namespace WebKit {
42 namespace NetworkCache {
43
44 class Statistics {
45 public:
46     static std::unique_ptr<Statistics> open(const String& cachePath);
47     explicit Statistics(const String& databasePath);
48
49     void clear();
50
51     void recordRetrievalRequest(uint64_t webPageID);
52     void recordNotCachingResponse(const Key&, StoreDecision);
53     void recordNotUsingCacheForRequest(uint64_t webPageID, const Key&, const WebCore::ResourceRequest&, RetrieveDecision);
54     void recordRetrievalFailure(uint64_t webPageID, const Key&, const WebCore::ResourceRequest&);
55     void recordRetrievedCachedEntry(uint64_t webPageID, const Key&, const WebCore::ResourceRequest&, UseDecision);
56     void recordRevalidationSuccess(uint64_t webPageID, const Key&, const WebCore::ResourceRequest&);
57
58 private:
59     WorkQueue& serialBackgroundIOQueue() { return m_serialBackgroundIOQueue.get(); }
60
61     void initialize(const String& databasePath);
62     void bootstrapFromNetworkCache(const String& networkCachePath);
63     void shrinkIfNeeded();
64
65     void addHashesToDatabase(const HashSet<String>& hashes);
66     void addStoreDecisionsToDatabase(const HashMap<String, NetworkCache::StoreDecision>&);
67     void writeTimerFired();
68
69     typedef Function<void (bool wasEverRequested, const std::optional<StoreDecision>&)> RequestedCompletionHandler;
70     enum class NeedUncachedReason { No, Yes };
71     void queryWasEverRequested(const String&, NeedUncachedReason, RequestedCompletionHandler&&);
72     void markAsRequested(const String& hash);
73
74     struct EverRequestedQuery {
75         String hash;
76         bool needUncachedReason;
77         RequestedCompletionHandler completionHandler;
78     };
79
80     std::atomic<size_t> m_approximateEntryCount { 0 };
81
82     mutable Ref<WorkQueue> m_serialBackgroundIOQueue;
83     mutable HashSet<std::unique_ptr<const EverRequestedQuery>> m_activeQueries;
84     WebCore::SQLiteDatabase m_database;
85     HashSet<String> m_hashesToAdd;
86     HashMap<String, NetworkCache::StoreDecision> m_storeDecisionsToAdd;
87     WebCore::Timer m_writeTimer;
88 };
89
90 }
91 }
92
93 #endif // ENABLE(NETWORK_CACHE)
94
95 #endif // NetworkCacheStatistics_h