added pre-finalize callback

Author: mdorier

src/Provider.cpp

@@ -17,26 +17,31 @@ Provider::Provider(const tl::engine& engine, ssg_group_id_t gid, bool must_join,
                    uint16_t provider_id, const std::string& config, const tl::pool& p)
 : self(std::make_shared<ProviderImpl>(engine, gid, must_join, mona, provider_id, p)) {
     self->get_engine().push_finalize_callback(this, [p=this]() { p->self.reset(); });
+    self->get_engine().push_prefinalize_callback(this, [p=this]() { p->self->onPreFinalize(); });
     self->processConfig(config);
 }
 
 Provider::Provider(margo_instance_id mid, ssg_group_id_t gid, bool must_join, mona_instance_t mona,
                    uint16_t provider_id, const std::string& config, const tl::pool& p)
 : self(std::make_shared<ProviderImpl>(mid, gid, must_join, mona, provider_id, p)) {
     self->get_engine().push_finalize_callback(this, [p=this]() { p->self.reset(); });
+    self->get_engine().push_prefinalize_callback(this, [p=this]() { p->self->onPreFinalize(); });
     self->processConfig(config);
 }
 
 Provider::Provider(Provider&& other) {
     if(!other.self) return;
     other.self->get_engine().pop_finalize_callback(this);
+    other.self->get_engine().pop_prefinalize_callback(this);
     self = std::move(other.self);
     self->get_engine().push_finalize_callback(this, [p=this]() { p->self.reset(); });
+    self->get_engine().push_prefinalize_callback(this, [p=this]() { p->self->onPreFinalize(); });
 }
 
 Provider::~Provider() {
     if(self) {
         self->get_engine().pop_finalize_callback(this);
+        self->get_engine().pop_prefinalize_callback(this);
     }
     spdlog::trace("Colza provider destroyed");
 }

src/ProviderImpl.hpp

@@ -151,8 +151,8 @@ class ProviderImpl : public tl::provider<ProviderImpl> {
         spdlog::trace("[colza:{0}] Registered provider with id {0}", id());
     }
 
-    ~ProviderImpl() {
-        spdlog::trace("[colza:{}] Deregistering provider", id());
+    void onPreFinalize() {
+        spdlog::trace("[colza:{}] Pre-finalizing provider", id());
         m_create_pipeline.deregister();
         m_destroy_pipeline.deregister();
         m_check_pipeline.deregister();
@@ -174,6 +174,10 @@ class ProviderImpl : public tl::provider<ProviderImpl> {
         spdlog::trace("[colza:{}]    => done!", id());
     }
 
+    ~ProviderImpl() {
+        spdlog::trace("[colza:{}] Deregistering provider", id());
+    }
+
     void processConfig(const std::string& config) {
         spdlog::trace("[colza:{}] Processing Colza configuration", id());
         if(config.empty()) {