write-copy.h 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. #ifndef _WRITE_COPY_HPP_
  2. #define _WRITE_COPY_HPP_
  3. #include <unordered_map>
  4. #include <algorithm>
  5. #include <memory>
  6. #include <mutex>
  7. #include <visit.h>
  8. template<typename T,typename K,typename V>
  9. struct write_copy_base:acceptor<V>
  10. {
  11. std::unordered_map<K,V> m_map;
  12. write_copy_base()
  13. {
  14. }
  15. std::shared_ptr<T> clone()const
  16. {
  17. std::shared_ptr<T> ret=std::make_shared<T>();
  18. ret->m_map.insert(m_map.begin(),m_map.end());
  19. return ret;
  20. }
  21. std::shared_ptr<T> clone_add(const std::unordered_map<K,V>&m)const
  22. {
  23. std::shared_ptr<T> ret=std::move(clone());
  24. ret->_add(m);
  25. return ret;
  26. }
  27. std::shared_ptr<T> clone_add(K k,V v)const
  28. {
  29. std::shared_ptr<T> ret=std::move(clone());
  30. ret->_add(k,v);
  31. return ret;
  32. }
  33. std::shared_ptr<T> clone_remove(K k)const
  34. {
  35. std::shared_ptr<T> ret=std::move(clone());
  36. ret->_remove(k);
  37. return ret;
  38. }
  39. std::shared_ptr<T> clone_remove(const std::vector<K>&list)const
  40. {
  41. std::shared_ptr<T> ret=std::move(clone());
  42. ret->_remove(list);
  43. return ret;
  44. }
  45. virtual ~write_copy_base()
  46. {
  47. // std::for_each(m_list.begin(),m_list.end(),[](K*it){ });
  48. }
  49. private:
  50. void _remove(K k)
  51. {
  52. m_map.erase(k);
  53. }
  54. void _remove(const std::vector<K>& v)
  55. {
  56. for(K&k:v)
  57. {
  58. m_map.erase(k);
  59. }
  60. }
  61. void _add(K k,V v)
  62. {
  63. m_map.insert(std::make_pair(k,v));
  64. }
  65. void _add(const std::unordered_map<K,V>&m)
  66. {
  67. m_map.insert(m.begin(),m.end());
  68. }
  69. };
  70. template<typename T,typename K,typename V>
  71. struct single_base:write_copy_base<T,K,V>
  72. {
  73. typedef write_copy_base<T,K,V> base;
  74. static std::shared_ptr<T> m_instance;
  75. static std::shared_ptr<T> instance()
  76. {
  77. return m_instance;
  78. }
  79. std::mutex m_mutex;
  80. void add(K k,V c)
  81. {
  82. std::lock_guard<std::mutex> lock(m_mutex);
  83. m_instance=std::move(base::clone_add(k,c));
  84. }
  85. void add(const std::unordered_map<K,V>&c)
  86. {
  87. std::lock_guard<std::mutex> lock(m_mutex);
  88. m_instance=std::move(base::clone_add(c));
  89. }
  90. void remove(K k,V c)
  91. {
  92. std::lock_guard<std::mutex> lock(m_mutex);
  93. m_instance=std::move(base::clone_remove(k,c));
  94. }
  95. void remove(const std::unordered_map<K,V>&c)
  96. {
  97. std::lock_guard<std::mutex> lock(m_mutex);
  98. m_instance=std::move(base::clone_remove(c));
  99. }
  100. };
  101. #endif