mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-23 21:01:08 +00:00
21d7e0ba7e
This allows us to use a consistent size for the dnsFlags field. across different files (previously some would use uint16_t and some uint32_t). It also improves type safety - making sure we don't pass in the wrong value to DNSFlags. Depends on D164856 Differential Revision: https://phabricator.services.mozilla.com/D164857
73 lines
3.0 KiB
C++
73 lines
3.0 KiB
C++
/* vim:set ts=4 sw=2 sts=2 et cin: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef HostRecordQueue_h__
|
|
#define HostRecordQueue_h__
|
|
|
|
#include <functional>
|
|
#include "mozilla/Mutex.h"
|
|
#include "nsHostRecord.h"
|
|
#include "nsRefPtrHashtable.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
class HostRecordQueue final {
|
|
public:
|
|
HostRecordQueue() = default;
|
|
~HostRecordQueue() = default;
|
|
HostRecordQueue(const HostRecordQueue& aCopy) = delete;
|
|
HostRecordQueue& operator=(const HostRecordQueue& aCopy) = delete;
|
|
|
|
uint32_t PendingCount() const { return mPendingCount; }
|
|
uint32_t EvictionQSize() const { return mEvictionQSize; }
|
|
|
|
// Insert the record to mHighQ or mMediumQ or mLowQ based on the record's
|
|
// priority.
|
|
void InsertRecord(nsHostRecord* aRec, nsIDNSService::DNSFlags aFlags,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Insert the record to mEvictionQ. In theory, this function should be called
|
|
// when the record is not in any queue.
|
|
void AddToEvictionQ(
|
|
nsHostRecord* aRec, uint32_t aMaxCacheEntries,
|
|
nsRefPtrHashtable<nsGenericHashKey<nsHostKey>, nsHostRecord>& aDB,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Called for removing the record from mEvictionQ. When this function is
|
|
// called, the record should be either in mEvictionQ or not in any queue.
|
|
void MaybeRenewHostRecord(nsHostRecord* aRec,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Called for clearing mEvictionQ.
|
|
void FlushEvictionQ(
|
|
nsRefPtrHashtable<nsGenericHashKey<nsHostKey>, nsHostRecord>& aDB,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Remove the record from the queue that contains it.
|
|
void MaybeRemoveFromQ(nsHostRecord* aRec, const MutexAutoLock& aProofOfLock);
|
|
// When the record's priority changes, move the record between pending queues.
|
|
void MoveToAnotherPendingQ(nsHostRecord* aRec, nsIDNSService::DNSFlags aFlags,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Returning the first record from one of the pending queue. When |aHighQOnly|
|
|
// is true, returning the record from mHighQ only. When false, return the
|
|
// record from mMediumQ or mLowQ.
|
|
already_AddRefed<AddrHostRecord> Dequeue(bool aHighQOnly,
|
|
const MutexAutoLock& aProofOfLock);
|
|
// Clear all queues and is called only during shutdown. |aCallback| is invoked
|
|
// when a record is removed from a queue.
|
|
void ClearAll(const std::function<void(nsHostRecord*)>& aCallback,
|
|
const MutexAutoLock& aProofOfLock);
|
|
|
|
private:
|
|
Atomic<uint32_t> mPendingCount{0};
|
|
Atomic<uint32_t> mEvictionQSize{0};
|
|
LinkedList<RefPtr<nsHostRecord>> mHighQ;
|
|
LinkedList<RefPtr<nsHostRecord>> mMediumQ;
|
|
LinkedList<RefPtr<nsHostRecord>> mLowQ;
|
|
LinkedList<RefPtr<nsHostRecord>> mEvictionQ;
|
|
};
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|
|
|
|
#endif // HostRecordQueue_h__
|