content/child/quota_message_filter.h

   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.
   4
   5 #ifndef CONTENT_CHILD_QUOTA_MESSAGE_FILTER_H_
   6 #define CONTENT_CHILD_QUOTA_MESSAGE_FILTER_H_
   7
   8 #include <map>
   9
  10 #include "base/synchronization/lock.h"
  11 #include "content/child/worker_thread_message_filter.h"
  12
  13 namespace content {
  14
  15 class QuotaMessageFilter : public WorkerThreadMessageFilter {
  16  public:
  17   explicit QuotaMessageFilter(ThreadSafeSender* thread_safe_sender);
  18
  19   // Generates a new request_id, registers { request_id, thread_id } map to
  20   // the message filter and returns the request_id.
  21   // This method can be called on any thread.
  22   int GenerateRequestID(int thread_id);
  23
  24   // Clears all requests from the thread_id.
  25   void ClearThreadRequests(int thread_id);
  26
  27  protected:
  28   ~QuotaMessageFilter() override;
  29
  30  private:
  31   // WorkerThreadMessageFilter:
  32   bool ShouldHandleMessage(const IPC::Message& msg) const override;
  33   void OnFilteredMessageReceived(const IPC::Message& msg) override;
  34   bool GetWorkerThreadIdForMessage(const IPC::Message& msg,
  35                                    int* ipc_thread_id) override;
  36
  37   typedef std::map<int, int> RequestIdToThreadId;
  38
  39   base::Lock request_id_map_lock_;
  40   RequestIdToThreadId request_id_map_;
  41   int next_request_id_;
  42
  43   DISALLOW_COPY_AND_ASSIGN(QuotaMessageFilter);
  44 };
  45
  46 }  // namespace content
  47
  48 #endif  // CONTENT_CHILD_QUOTA_MESSAGE_FILTER_H_