#pragma once
#include <WebCore/ResourceRequest.h>
#include <WebCore/ResourceResponse.h>
#include <WebCore/SharedBuffer.h>
#include <WebCore/Timer.h>
#include <wtf/Deque.h>
#include <wtf/HashMap.h>
#include <wtf/URLHash.h>
#include <wtf/text/WTFString.h>
namespace WebKit {
class PrefetchCache {
WTF_MAKE_NONCOPYABLE(PrefetchCache);
WTF_MAKE_FAST_ALLOCATED;
public:
PrefetchCache();
~PrefetchCache();
void clear();
struct Entry {
WTF_MAKE_STRUCT_FAST_ALLOCATED;
Entry(WebCore::ResourceResponse&&, RefPtr<WebCore::SharedBuffer>&&);
Entry(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&);
Ref<WebCore::SharedBuffer> releaseBuffer() { return buffer.releaseNonNull(); }
WebCore::ResourceResponse response;
RefPtr<WebCore::SharedBuffer> buffer;
WebCore::ResourceRequest redirectRequest;
};
std::unique_ptr<Entry> take(const URL&);
void store(const URL&, WebCore::ResourceResponse&&, RefPtr<WebCore::SharedBuffer>&&);
void storeRedirect(const URL&, WebCore::ResourceResponse&&, WebCore::ResourceRequest&&);
private:
void clearExpiredEntries();
using PrefetchEntriesMap = HashMap<URL, std::unique_ptr<Entry>>;
std::unique_ptr<PrefetchEntriesMap> m_sessionPrefetches;
using SessionPrefetchExpirationList = Deque<std::tuple<URL, WallTime>>;
SessionPrefetchExpirationList m_sessionExpirationList;
WebCore::Timer m_expirationTimer;
};
}