ZmqContextManager.h

/*
Copyright (c) 2017-2025,
Battelle Memorial Institute; Lawrence Livermore National Security, LLC; Alliance for Sustainable
Energy, LLC.  See the top-level NOTICE for additional details. All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
*/
 
/*
 * LLNS Copyright Start
 * Copyright (c) 2017, Lawrence Livermore National Security
 * This work was performed under the auspices of the U.S. Department
 * of Energy by Lawrence Livermore National Laboratory in part under
 * Contract W-7405-Eng-48 and in part under Contract DE-AC52-07NA27344.
 * Produced at the Lawrence Livermore National Laboratory.
 * All rights reserved.
 * For details, see the LICENSE file.
 * LLNS Copyright End
 */
#pragma once
 
#include <atomic>
#include <map>
#include <memory>
#include <string>
namespace zmq {
class context_t;
}  // namespace zmq
 
class ZmqContextManager {
  private:
    static std::map<std::string, std::shared_ptr<ZmqContextManager>>
        contexts;  
    std::string name;  
    std::unique_ptr<zmq::context_t> zcontext;  
    std::atomic<bool> leakOnDelete{true};  
 
    explicit ZmqContextManager(const std::string& contextName);
 
  public:
    static std::shared_ptr<ZmqContextManager>
        getContextPointer(const std::string& contextName = std::string{});
    static zmq::context_t& getContext(const std::string& contextName = std::string{});
    static void startContext(const std::string& contextName = std::string{});
    static void closeContext(const std::string& contextName = std::string{});
    static bool setContextToLeakOnDelete(const std::string& contextName = std::string{});
    static bool setContextToNotLeakOnDelete(const std::string& contextName = std::string{});
    ~ZmqContextManager();
    const std::string& getName() const { return name; }
    zmq::context_t& getBaseContext() const { return *zcontext; }
};