1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "content/child/quota_message_filter.h"
8 #include "base/location.h"
9 #include "base/message_loop/message_loop_proxy.h"
10 #include "base/pickle.h"
11 #include "content/child/quota_dispatcher.h"
12 #include "content/child/thread_safe_sender.h"
13 #include "content/common/quota_messages.h"
14 #include "webkit/child/worker_task_runner.h"
16 using webkit_glue::WorkerTaskRunner;
20 QuotaMessageFilter::QuotaMessageFilter(
21 ThreadSafeSender* thread_safe_sender)
22 : main_thread_loop_proxy_(base::MessageLoopProxy::current()),
23 thread_safe_sender_(thread_safe_sender),
27 bool QuotaMessageFilter::OnMessageReceived(const IPC::Message& msg) {
28 if (IPC_MESSAGE_CLASS(msg) != QuotaMsgStart)
31 bool result = PickleIterator(msg).ReadInt(&request_id);
33 base::Closure closure = base::Bind(
34 &QuotaMessageFilter::DispatchMessage, this, msg);
37 base::AutoLock lock(request_id_map_lock_);
38 RequestIdToThreadId::iterator found = request_id_map_.find(request_id);
39 if (found != request_id_map_.end()) {
40 thread_id = found->second;
41 request_id_map_.erase(found);
45 main_thread_loop_proxy_->PostTask(FROM_HERE, closure);
48 WorkerTaskRunner::Instance()->PostTask(thread_id, closure);
52 int QuotaMessageFilter::GenerateRequestID(int thread_id) {
53 base::AutoLock lock(request_id_map_lock_);
54 request_id_map_[next_request_id_] = thread_id;
55 return next_request_id_++;
58 void QuotaMessageFilter::ClearThreadRequests(int thread_id) {
59 base::AutoLock lock(request_id_map_lock_);
60 for (RequestIdToThreadId::iterator iter = request_id_map_.begin();
61 iter != request_id_map_.end();) {
62 if (iter->second == thread_id)
63 request_id_map_.erase(iter++);
69 QuotaMessageFilter::~QuotaMessageFilter() {}
71 void QuotaMessageFilter::DispatchMessage(const IPC::Message& msg) {
72 QuotaDispatcher::ThreadSpecificInstance(thread_safe_sender_.get(), this)
73 ->OnMessageReceived(msg);
76 } // namespace content