dom/quota/QuotaObject.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef mozilla_dom_quota_quotaobject_h__
   8 #define mozilla_dom_quota_quotaobject_h__
   9
  10 // Local includes
  11 #include "Client.h"
  12
  13 // Global includes
  14 #include <cstdint>
  15 #include "mozilla/AlreadyAddRefed.h"
  16 #include "mozilla/RefPtr.h"
  17 #include "nsCOMPtr.h"
  18 #include "nsISupports.h"
  19 #include "nsStringFwd.h"
  20
  21 // XXX Avoid including this here by moving function bodies to the cpp file.
  22 #include "mozilla/dom/quota/QuotaCommon.h"
  23
  24 namespace mozilla::dom::quota {
  25
  26 class OriginInfo;
  27 class QuotaManager;
  28
  29 class QuotaObject {
  30   friend class OriginInfo;
  31   friend class QuotaManager;
  32
  33   class StoragePressureRunnable;
  34
  35  public:
  36   void AddRef();
  37
  38   void Release();
  39
  40   const nsAString& Path() const { return mPath; }
  41
  42   [[nodiscard]] bool MaybeUpdateSize(int64_t aSize, bool aTruncate);
  43
  44   bool IncreaseSize(int64_t aDelta);
  45
  46   void DisableQuotaCheck();
  47
  48   void EnableQuotaCheck();
  49
  50  private:
  51   QuotaObject(OriginInfo* aOriginInfo, Client::Type aClientType,
  52               const nsAString& aPath, int64_t aSize)
  53       : mOriginInfo(aOriginInfo),
  54         mPath(aPath),
  55         mSize(aSize),
  56         mClientType(aClientType),
  57         mQuotaCheckDisabled(false),
  58         mWritingDone(false) {
  59     MOZ_COUNT_CTOR(QuotaObject);
  60   }
  61
  62   MOZ_COUNTED_DTOR(QuotaObject)
  63
  64   already_AddRefed<QuotaObject> LockedAddRef() {
  65     AssertCurrentThreadOwnsQuotaMutex();
  66
  67     ++mRefCnt;
  68
  69     RefPtr<QuotaObject> result = dont_AddRef(this);
  70     return result.forget();
  71   }
  72
  73   bool LockedMaybeUpdateSize(int64_t aSize, bool aTruncate);
  74
  75   mozilla::ThreadSafeAutoRefCnt mRefCnt;
  76
  77   OriginInfo* mOriginInfo;
  78   nsString mPath;
  79   int64_t mSize;
  80   Client::Type mClientType;
  81   bool mQuotaCheckDisabled;
  82   bool mWritingDone;
  83 };
  84
  85 }  // namespace mozilla::dom::quota
  86
  87 #endif  // mozilla_dom_quota_quotaobject_h__