| |
| #pragma once |
| #include "moses/Util.h" |
| #include "moses/ContextScope.h" |
| #include "moses/parameters/AllOptions.h" |
| #include <sys/time.h> |
| #include <boost/unordered_map.hpp> |
|
|
| #ifdef WITH_THREADS |
| #include <boost/thread/shared_mutex.hpp> |
| #include <boost/thread/locks.hpp> |
| #endif |
| namespace MosesServer{ |
| |
| struct Session |
| { |
| uint64_t const id; |
| time_t start_time; |
| time_t last_access; |
| boost::shared_ptr<Moses::ContextScope> const scope; |
| SPTR<std::map<std::string,float> > m_context_weights; |
|
|
| |
| Session(uint64_t const session_id) |
| : id(session_id) |
| , scope(new Moses::ContextScope) |
| { |
| last_access = start_time = time(NULL); |
| } |
|
|
| bool is_new() const { return last_access == start_time; } |
|
|
| void setup(std::map<std::string, xmlrpc_c::value> const& params); |
| }; |
|
|
| class SessionCache |
| { |
| mutable boost::shared_mutex m_lock; |
| uint64_t m_session_counter; |
| boost::unordered_map<uint64_t,Session> m_cache; |
| public: |
|
|
| SessionCache() : m_session_counter(1) {} |
|
|
| Session const& |
| operator[](uint32_t id) |
| { |
| boost::upgrade_lock<boost::shared_mutex> lock(m_lock); |
| if (id > 1) |
| { |
| boost::unordered_map<uint64_t, Session>::iterator m = m_cache.find(id); |
| if (m != m_cache.end()) |
| { |
| m->second.last_access = time(NULL); |
| return m->second; |
| } |
| } |
| boost::upgrade_to_unique_lock<boost::shared_mutex> xlock(lock); |
| id = ++m_session_counter; |
| std::pair<uint64_t, Session> foo(id, Session(id)); |
| return m_cache.insert(foo).first->second; |
| } |
|
|
| void |
| erase(uint32_t const id) |
| { |
| boost::unique_lock<boost::shared_mutex> lock(m_lock); |
| m_cache.erase(id); |
| } |
|
|
|
|
| }; |
|
|
|
|
| } |
|
|
