tests/loadable_storage.cpp - master - zOMGdev/cppcms

tests/loadable_storage.cpp
107 строк3 KB
Artyom Beilis
- Update copyright notice,
03 янв 2012, 17:58
03 янв 2012, 17:58068d3f9
///////////////////////////////////////////////////////////////////////////////
//                                                                             
//  Copyright (C) 2008-2012  Artyom Beilis (Tonkikh) <artyomtnk@yahoo.com>     
//                                                                             
//  See accompanying file COPYING.TXT file for licensing details.
//
///////////////////////////////////////////////////////////////////////////////
#include <cppcms/defs.h>
#include <cppcms/session_storage.h>
#include <cppcms/json.h>
#include <time.h>
#include <map>

namespace {

class test_storage  : public cppcms::sessions::session_storage {
public:
	///
	/// Save session with end of life time at \a timeout using session id \a sid and content \a in
	///

	virtual void save(std::string const &sid,time_t timeout,std::string const &in)
	{
		data_[sid]=std::pair<time_t,std::string>(timeout,in);
	}

	///
	/// Load session with \a sid, put its end of life time to \a timeout and return its
	/// value to \a out
	///
	virtual bool load(std::string const &sid,time_t &timeout,std::string &out)
	{
		data_type::iterator p=data_.find(sid);
		if(p==data_.end())
			return false;
		if(p->second.first < time(0)) {
			data_.erase(p);
			return false;
		}
		out=p->second.second;
		timeout = p->second.first;
		return true;
	}

	///
	/// Remove a session with id \a sid  from the storage
	///

	virtual void remove(std::string const &sid)
	{
		data_.erase(sid);
	}

	///
	/// Return true of the save or load operations can be blocking
	///
	virtual bool is_blocking() 
	{
		return false;
	}
private:
	typedef std::map<std::string,std::pair<time_t,std::string> > data_type;
	data_type data_;
};

class test_fact : public cppcms::sessions::session_storage_factory {
public:
	///
	/// Get a pointer to session_storage. Note if the returned pointer is same for different calls
	/// session_storage implementation should be thread safe.
	///
	virtual booster::shared_ptr<cppcms::sessions::session_storage> get() 
	{
		return storage_;
	}

	///
	/// Return true if session_storage requires garbage collection - removal of expired session time-to-time
	///
	virtual bool requires_gc() { return false; };
	///
	/// Delete the object, cleanup
	///
	virtual ~test_fact() {}
	test_fact() : 
		storage_(new test_storage())
	{
	}
private:
	booster::shared_ptr<cppcms::sessions::session_storage> storage_;
};

} // anon

#if defined(CPPCMS_WIN32)
# define STORAGE_API __declspec(dllexport)
#else
# define STORAGE_API
#endif

extern "C" {
	STORAGE_API cppcms::sessions::session_storage_factory *my_sessions_generator(cppcms::json::value const &v)
	{
		v.get<bool>("must_be_set");
		return new test_fact();
	}
}