AccountRestrictions.h

Go to the documentation of this file.

 
#pragma once
#include "TypedAccountRestriction.h"
#include "catapult/types.h"
 
namespace catapult { namespace state {
 
    class AccountRestrictions {
    private:
        using RestrictionsMap = std::map<model::AccountRestrictionType, AccountRestriction>;
        using const_iterator = RestrictionsMap::const_iterator;
 
    public:
        explicit AccountRestrictions(const Address& address);
 
    public:
        const Address& address() const;
 
        size_t size() const;
 
        bool isEmpty() const;
 
    public:
        template<typename TRestrictionValue>
        TypedAccountRestriction<TRestrictionValue> restriction(model::AccountRestrictionType restrictionType) const {
            const auto& restriction = this->restriction<const RestrictionsMap, const AccountRestriction>(m_restrictions, restrictionType);
            return TypedAccountRestriction<TRestrictionValue>(restriction);
        }
 
        const AccountRestriction& restriction(model::AccountRestrictionType restrictionType) const;
 
        AccountRestriction& restriction(model::AccountRestrictionType restrictionType);
 
    public:
        const_iterator begin() const;
 
        const_iterator end() const;
 
    private:
        void addRestriction(model::AccountRestrictionType restrictionType, size_t restrictionValueSize);
 
    private:
        template<typename TRestrictionsMap, typename TAccountRestriction>
        static TAccountRestriction& restriction(TRestrictionsMap& restrictionsMap, model::AccountRestrictionType restrictionType) {
            auto iter = restrictionsMap.find(AccountRestrictionDescriptor(restrictionType).restrictionType());
            if (restrictionsMap.cend() == iter)
                CATAPULT_THROW_INVALID_ARGUMENT_1("unknown account restriction type", static_cast<uint16_t>(restrictionType));
 
            return iter->second;
        }
 
    private:
        Address m_address;
        RestrictionsMap m_restrictions;
    };
}}