Home | History | Annotate | Download | only in storage 
      1 /*
      2  * Copyright (C) 2008 Apple 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
      6  * are met:
      7  *
      8  * 1.  Redistributions of source code must retain the above copyright
      9  *     notice, this list of conditions and the following disclaimer.
     10  * 2.  Redistributions in binary form must reproduce the above copyright
     11  *     notice, this list of conditions and the following disclaimer in the
     12  *     documentation and/or other materials provided with the distribution.
     13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
     14  *     its contributors may be used to endorse or promote products derived
     15  *     from this software without specific prior written permission.
     16  *
     17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
     18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
     21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
     22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
     24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 #ifndef OriginUsageRecord_h
     29 #define OriginUsageRecord_h
     30 
     31 #if ENABLE(DATABASE)
     32 
     33 #include "PlatformString.h"
     34 #include <wtf/HashMap.h>
     35 #include <wtf/HashSet.h>
     36 #include <wtf/text/StringHash.h>
     37 
     38 namespace WebCore {
     39 
     40 // Objects of this class can be used from multiple threads with external synchronization.
     41 // String arguments are also supposed to be deeply copied by the caller when necessary.
     42 class OriginUsageRecord {
     43     WTF_MAKE_NONCOPYABLE(OriginUsageRecord); WTF_MAKE_FAST_ALLOCATED;
     44 public:
     45     OriginUsageRecord();
     46 
     47     void addDatabase(const String& identifier, const String& fullPath);
     48     void removeDatabase(const String& identifier);
     49     void markDatabase(const String& identifier); // Size may have changed, and will need to be recalculated.
     50     unsigned long long diskUsage();
     51 
     52 private:
     53     struct DatabaseEntry {
     54         DatabaseEntry() { }
     55         DatabaseEntry(const String& filename) : filename(filename) { }
     56         DatabaseEntry(const String& filename, unsigned long long size) : filename(filename), size(size) { }
     57         String filename;
     58         unsigned long long size; // This member remains uninitialized until disk usage is calculated.
     59     };
     60     HashMap<String, DatabaseEntry> m_databaseMap;
     61     HashSet<String> m_unknownSet;
     62 
     63     unsigned long long m_cachedDiskUsage;
     64     bool m_cachedDiskUsageIsValid;
     65 };
     66 
     67 } // namespace WebCore
     68 
     69 #endif
     70 
     71 #endif
     72