src/third_party/WebKit/Source/modules/webdatabase/QuotaTracker.cpp

   1 /*
   2  * Copyright (C) 2011 Google 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 are
   6  * met:
   7  *
   8  *     * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *     * Redistributions in binary form must reproduce the above
  11  * copyright notice, this list of conditions and the following disclaimer
  12  * in the documentation and/or other materials provided with the
  13  * distribution.
  14  *     * Neither the name of Google Inc. nor the names of its
  15  * contributors may be used to endorse or promote products derived from
  16  * this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include "config.h"
  32 #include "modules/webdatabase/QuotaTracker.h"
  33
  34 #include "public/platform/Platform.h"
  35 #include "wtf/StdLibExtras.h"
  36 #include "wtf/Threading.h"
  37
  38 namespace blink {
  39
  40 QuotaTracker& QuotaTracker::instance()
  41 {
  42     AtomicallyInitializedStatic(QuotaTracker&, tracker = *new QuotaTracker);
  43     return tracker;
  44 }
  45
  46 void QuotaTracker::getDatabaseSizeAndSpaceAvailableToOrigin(
  47     const String& originIdentifier, const String& databaseName,
  48     unsigned long long* databaseSize, unsigned long long* spaceAvailable)
  49 {
  50     // Extra scope to unlock prior to potentially calling Platform.
  51     {
  52         MutexLocker lockData(m_dataGuard);
  53         ASSERT(m_databaseSizes.contains(originIdentifier));
  54         HashMap<String, SizeMap>::const_iterator it = m_databaseSizes.find(originIdentifier);
  55         ASSERT(it->value.contains(databaseName));
  56         *databaseSize = it->value.get(databaseName);
  57
  58         if (m_spaceAvailableToOrigins.contains(originIdentifier)) {
  59             *spaceAvailable = m_spaceAvailableToOrigins.get(originIdentifier);
  60             return;
  61         }
  62     }
  63
  64     // The embedder hasn't pushed this value to us, so we pull it as needed.
  65     *spaceAvailable = Platform::current()->databaseGetSpaceAvailableForOrigin(originIdentifier);
  66 }
  67
  68 void QuotaTracker::updateDatabaseSize(
  69     const String& originIdentifier, const String& databaseName,
  70     unsigned long long databaseSize)
  71 {
  72     MutexLocker lockData(m_dataGuard);
  73     HashMap<String, SizeMap>::ValueType* it = m_databaseSizes.add(originIdentifier, SizeMap()).storedValue;
  74     it->value.set(databaseName, databaseSize);
  75 }
  76
  77 void QuotaTracker::updateSpaceAvailableToOrigin(const String& originIdentifier, unsigned long long spaceAvailable)
  78 {
  79     MutexLocker lockData(m_dataGuard);
  80     m_spaceAvailableToOrigins.set(originIdentifier, spaceAvailable);
  81 }
  82
  83 void QuotaTracker::resetSpaceAvailableToOrigin(const String& originIdentifier)
  84 {
  85     MutexLocker lockData(m_dataGuard);
  86     m_spaceAvailableToOrigins.remove(originIdentifier);
  87 }
  88
  89 } // namespace blink