register cpc teardown functions for std::atexit

apache · Dec 21, 2023 · 59bc5fb · 59bc5fb
1 parent 04346bb
commit 59bc5fb
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/cpc/include/cpc_compressor.hpp b/cpc/include/cpc_compressor.hpp
@@ -44,6 +44,10 @@ template<typename A> class cpc_compressor;
 template<typename A>
 inline cpc_compressor<A>& get_compressor();
 
+// function called atexit to clean up compression tables
+template<typename A>
+void destroy_compressor();
+
 template<typename A>
 class cpc_compressor {
 public:
@@ -110,6 +114,8 @@ class cpc_compressor {
 
   cpc_compressor();
   template<typename T> friend cpc_compressor<T>& get_compressor();
+  friend void destroy_compressor<A>();
+
   ~cpc_compressor();
 
   void make_decoding_tables(); // call this at startup

diff --git a/cpc/include/cpc_compressor_impl.hpp b/cpc/include/cpc_compressor_impl.hpp
@@ -22,6 +22,7 @@
 #ifndef CPC_COMPRESSOR_IMPL_HPP_
 #define CPC_COMPRESSOR_IMPL_HPP_
 
+#include <cstdlib>
 #include <memory>
 #include <stdexcept>
 
@@ -35,10 +36,21 @@ namespace datasketches {
 // construct on first use
 template<typename A>
 cpc_compressor<A>& get_compressor() {
+  static bool do_init = true;
   static cpc_compressor<A>* instance = new cpc_compressor<A>(); // use new for global initialization
+  if (do_init) {
+    std::atexit(destroy_compressor<A>); // just to clean up a little more nicely; don't worry if it fails
+    do_init = false;
+  }
   return *instance;
 }
 
+// register to call compressor destructor at exit
+template<typename A>
+void destroy_compressor() {
+  delete std::addressof(get_compressor<A>());
+}
+
 template<typename A>
 cpc_compressor<A>::cpc_compressor() {
   make_decoding_tables();