blob: 09fa36bbe8bd603594dd56cbe3651013161d5352 [file] [log] [blame]
// Copyright (c) 2011 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.
#ifndef CHROME_BROWSER_BROWSING_DATA_QUOTA_HELPER_IMPL_H_
#define CHROME_BROWSER_BROWSING_DATA_QUOTA_HELPER_IMPL_H_
#pragma once
#include <map>
#include <set>
#include <string>
#include <utility>
#include "base/callback_old.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_callback_factory.h"
#include "base/memory/scoped_ptr.h"
#include "base/time.h"
#include "chrome/browser/browsing_data_quota_helper.h"
#include "content/browser/browser_thread.h"
#include "webkit/quota/quota_types.h"
namespace quota {
class QuotaManager;
}
// Implementation of BrowsingDataQuotaHelper. Since a client of
// BrowsingDataQuotaHelper should live in UI thread and QuotaManager lives in
// IO thread, we have to communicate over thread using PostTask.
class BrowsingDataQuotaHelperImpl : public BrowsingDataQuotaHelper {
public:
virtual void StartFetching(FetchResultCallback* callback) OVERRIDE;
virtual void CancelNotification() OVERRIDE;
private:
void FetchQuotaInfo();
void OnComplete();
void GetHostUsage(const std::string& host, quota::StorageType type);
void ProcessPendingHosts();
// Callback function for GetOriginModifiedSince.
void GotOrigins(const std::set<GURL>& origins, quota::StorageType type);
// Callback function for GetHostUsage.
void GotHostUsage(const std::string& host,
quota::StorageType type,
int64 usage);
explicit BrowsingDataQuotaHelperImpl(base::MessageLoopProxy* ui_thread,
base::MessageLoopProxy* io_thread,
quota::QuotaManager* quota_manager);
virtual ~BrowsingDataQuotaHelperImpl();
scoped_refptr<quota::QuotaManager> quota_manager_;
scoped_ptr<FetchResultCallback> callback_;
typedef std::set<std::pair<std::string, quota::StorageType> > PendingHosts;
PendingHosts pending_hosts_;
std::map<std::string, QuotaInfo> quota_info_;
bool is_fetching_;
scoped_refptr<base::MessageLoopProxy> ui_thread_;
scoped_refptr<base::MessageLoopProxy> io_thread_;
base::ScopedCallbackFactory<BrowsingDataQuotaHelperImpl> callback_factory_;
friend class BrowsingDataQuotaHelper;
friend class BrowsingDataQuotaHelperTest;
DISALLOW_COPY_AND_ASSIGN(BrowsingDataQuotaHelperImpl);
};
#endif // CHROME_BROWSER_BROWSING_DATA_QUOTA_HELPER_IMPL_H_