environment.h 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #pragma once
  2. #include <map>
  3. #include <string>
  4. #include <mutex>
  5. #include "base/define.h"
  6. #include "base/log4.h"
  7. namespace ylib
  8. {
  9. class environment
  10. {
  11. public:
  12. template<typename T>
  13. T* to(const std::string& name)
  14. {
  15. std::unique_lock<std::mutex> auto_lock(m_mutex);
  16. return (T*)m_map[name];
  17. }
  18. template<typename T>
  19. void set(const std::string name,T* value)
  20. {
  21. std::unique_lock<std::mutex> auto_lock(m_mutex);
  22. m_map[name] = value;
  23. }
  24. void del(const std::string& name)
  25. {
  26. std::unique_lock<std::mutex> auto_lock(m_mutex);
  27. auto iter = m_map.find(name);
  28. if (iter != m_map.end())
  29. m_map.erase(iter);
  30. }
  31. private:
  32. std::mutex m_mutex;
  33. std::map<std::string,void*> m_map;
  34. };
  35. namespace network
  36. {
  37. namespace http
  38. {
  39. class center;
  40. }
  41. }
  42. namespace mysql_plus
  43. {
  44. class pool;
  45. }
  46. }
  47. namespace ylib
  48. {
  49. extern ylib::environment *env;
  50. extern ylib::log4 *log;
  51. }