ha_relationship_mapper.h

Go to the documentation of this file.

// Copyright (C) 2023-2024 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef HA_RELATIONSHIP_MAPPER_H
#define HA_RELATIONSHIP_MAPPER_H
 
#include <config.h>
 
#include <exceptions/exceptions.h>
#include <boost/shared_ptr.hpp>
#include <unordered_map>
#include <vector>
 
namespace isc {
namespace ha {
 
template<typename MappedType>
class HARelationshipMapper {
public:
 
    typedef boost::shared_ptr<MappedType> MappedTypePtr;
 
    void map(const std::string& key, MappedTypePtr obj) {
        if (mapping_.count(key) > 0) {
            isc_throw(InvalidOperation, "a relationship '" << key << "' already exists");
        }
        mapping_[key] = obj;
 
        auto found = false;
        for (auto const& o : vector_) {
            if (o == obj) {
                found = true;
                break;
            }
        }
        if (!found) {
            vector_.push_back(obj);
        }
    }
 
    MappedTypePtr get(const std::string& key) const {
        auto obj = mapping_.find(key);
        if (obj == mapping_.end()) {
            return (MappedTypePtr());
        }
        return (obj->second);
    }
 
    MappedTypePtr get() const {
        if (vector_.empty()) {
            isc_throw(InvalidOperation, "expected one relationship to be configured");
        }
        return (vector_[0]);
    }
 
    const std::vector<MappedTypePtr>& getAll() const {
        return (vector_);
    }
 
    bool hasMultiple() const {
        return (vector_.size() > 1);
    }
 
private:
 
    std::unordered_map<std::string, MappedTypePtr> mapping_;
 
    std::vector<MappedTypePtr> vector_;
};
 
} // end of namespace isc::ha
} // end of namespace isc
 
#endif // HA_RELATIONSHIP_MAPPER_H