write-copy.h 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. #ifndef _WRITE_COPY_HPP_
  2. #define _WRITE_COPY_HPP_
  3. //add之前如果有数据,需要把运行中更改的数据进行保存。
  4. #include <unordered_map>
  5. #include <algorithm>
  6. #include <memory>
  7. #include <mutex>
  8. #include <atomic>
  9. #include "visit.h"
  10. template<typename T,typename K,typename V>
  11. struct write_copy_base:acceptor<V>
  12. {
  13. void accept(visitor<V> &v)
  14. {
  15. for(std::pair<K,V> me:m_map)
  16. {
  17. v.visit(me.second);
  18. }
  19. }
  20. std::unordered_map<K,V> m_map;
  21. int m_version=0;
  22. write_copy_base()
  23. {}
  24. int version()const
  25. {
  26. return m_version;
  27. }
  28. int set_version(int v)
  29. {
  30. return m_version=++v;
  31. }
  32. V get(K k)const
  33. {
  34. return m_map[k];
  35. }
  36. V get(K k)
  37. {
  38. auto serch = m_map.find(k);
  39. if(serch != m_map.end())
  40. return serch->second;
  41. else
  42. return nullptr;
  43. }
  44. std::shared_ptr<T> clone()const
  45. {
  46. std::shared_ptr<T> ret=std::make_shared<T>();
  47. ret->m_map.insert(m_map.begin(),m_map.end());
  48. ret->set_version(m_version);
  49. return ret;
  50. }
  51. std::shared_ptr<T> clone_add(const std::unordered_map<K,V>&m)const
  52. {
  53. std::shared_ptr<T> ret=std::move(clone());
  54. ret->_add(m);
  55. return ret;
  56. }
  57. std::shared_ptr<T> clone_add(K k,V v)const
  58. {
  59. std::shared_ptr<T> ret=std::move(clone());
  60. ret->_add(k,v);
  61. return ret;
  62. }
  63. std::shared_ptr<T> clone_remove(K k)const
  64. {
  65. std::shared_ptr<T> ret=std::move(clone());
  66. ret->_remove(k);
  67. return ret;
  68. }
  69. std::shared_ptr<T> clone_remove(const std::vector<K>&list)const
  70. {
  71. std::shared_ptr<T> ret=std::move(clone());
  72. ret->_remove(list);
  73. return ret;
  74. }
  75. virtual ~write_copy_base()
  76. {
  77. // std::for_each(m_list.begin(),m_list.end(),[](K*it){ });
  78. }
  79. private:
  80. void _remove(K k)
  81. {
  82. m_map.erase(k);
  83. }
  84. void _remove(const std::vector<K>& v)
  85. {
  86. for(K k:v)
  87. {
  88. m_map.erase(k);
  89. }
  90. }
  91. void _add(K k,V v)
  92. {
  93. m_map.insert(std::make_pair(k,v));
  94. }
  95. void _add(const std::unordered_map<K,V>&m)
  96. {
  97. m_map.insert(m.begin(),m.end());
  98. }
  99. };
  100. template<typename T,typename K,typename V>
  101. struct single_base:write_copy_base<T,K,V>
  102. {
  103. typedef write_copy_base<T,K,V> base;
  104. static std::shared_ptr<T> m_instance;
  105. static std::shared_ptr<T> instance()
  106. {
  107. return m_instance;
  108. }
  109. std::mutex m_mutex;
  110. void add(K k,V c)
  111. {
  112. std::lock_guard<std::mutex> lock(m_mutex);
  113. m_instance=std::move(base::clone_add(k,c));
  114. }
  115. void add(const std::unordered_map<K,V>&c)
  116. {
  117. std::lock_guard<std::mutex> lock(m_mutex);
  118. m_instance=std::move(base::clone_add(c));
  119. }
  120. void remove(K k,V c)
  121. {
  122. std::lock_guard<std::mutex> lock(m_mutex);
  123. m_instance=std::move(base::clone_remove(k,c));
  124. }
  125. void remove(const std::unordered_map<K,V>&c)
  126. {
  127. std::lock_guard<std::mutex> lock(m_mutex);
  128. m_instance=std::move(base::clone_remove(c));
  129. }
  130. void remove(const std::vector<K>&list)
  131. {
  132. std::lock_guard<std::mutex> lock(m_mutex);
  133. m_instance=std::move(base::clone_remove(list));
  134. }
  135. void remove(K k)
  136. {
  137. std::lock_guard<std::mutex> lock(m_mutex);
  138. m_instance=std::move(base::clone_remove(k));
  139. }
  140. };
  141. #endif