YiLRUCache.h

Go to the documentation of this file.

// © You i Labs Inc. 2000-2017. All rights reserved.
#ifndef _YI_LRU_CACHE_H_
#define _YI_LRU_CACHE_H_

#include "network/YiHTTPResponse.h"
#include "network/YiUrl.h"
#include "thread/YiMutex.h"
#include "utility/YiDateTime.h"
#include "utility/YiString.h"

#include <set>

class CYIHTTPRequest;

struct CYILRUData
{
    virtual ~CYILRUData()
    {}

    CYIUrl m_url;
};

struct CYILRUCacheItem;
struct CYILRUSortItem
{
    bool operator() (const CYILRUCacheItem *lhs, const CYILRUCacheItem *rhs) const;
};

class CYILRUCache
{
public:
    CYILRUCache();
    ~CYILRUCache();
    
    void Add(const CYIUrl &url,
             std::unique_ptr<const CYILRUData> pData,
             uint32_t uMaxAgeInSeconds = 0);

    bool Contains(const CYIUrl &url) const;

    int32_t GetSize() const;

    std::vector<CYIUrl> GetUrls() const;

    const CYILRUData *GetData(const CYIUrl &url) const;

    bool HasExpired(const CYIUrl &url) const;

    CYIDateTime GetExpiryDate(const CYIUrl &url) const;

    void Remove(const CYIUrl &url);

    const CYILRUData *Peek(const CYIUrl &url) const;

    const CYILRUData *Peek(int32_t nIndex) const;

    void SetExpireTime(uint32_t uMaxAgeInSeconds, const CYIUrl &url);

    void LogCacheItems() const;

private:

    void Remove(CYILRUCacheItem *pItem);

    mutable std::set<CYILRUCacheItem * /* pItem */, CYILRUSortItem> m_TimeCache;
    std::map<CYIUrl /* url */, CYILRUCacheItem * /* pItem */> m_UrlCache;

    mutable CYIMutex m_CacheMutex;

    YI_DISALLOW_COPY_AND_ASSIGN(CYILRUCache)
};

#endif // _YI_LRU_CACHE_H_