#pragma once
#include <drogon/Session.h>
#include <drogon/drogon_callbacks.h>
#include <drogon/CacheMap.h>
#include <trantor/utils/NonCopyable.h>
#include <trantor/net/EventLoop.h>
#include <functional>
#include <memory>
#include <string>
#include <mutex>
#include <vector>
namespace drogon
{
class SessionManager : public trantor::NonCopyable
{
  public:
    using IdGeneratorCallback = std::function<std::string()>;
    SessionManager(
        trantor::EventLoop *loop,
        size_t timeout,
        const std::vector<AdviceStartSessionCallback> &startAdvices,
        const std::vector<AdviceDestroySessionCallback> &destroyAdvices,
        IdGeneratorCallback idGeneratorCallback);
    ~SessionManager()
    {
        sessionMapPtr_.reset();
    }
    SessionPtr getSession(const std::string &sessionID, bool needToSet);
    void changeSessionId(const SessionPtr &sessionPtr);
  private:
    std::unique_ptr<CacheMap<std::string, SessionPtr>> sessionMapPtr_;
    trantor::EventLoop *loop_;
    size_t timeout_;
    const std::vector<AdviceStartSessionCallback> &sessionStartAdvices_;
    const std::vector<AdviceDestroySessionCallback> &sessionDestroyAdvices_;
    IdGeneratorCallback idGeneratorCallback_;
};
}