rmqt_consumerconfig.h

// Copyright 2020-2023 Bloomberg Finance L.P.
// SPDX-License-Identifier: Apache-2.0
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
// rmqt_consumerconfig.h
#ifndef INCLUDED_RMQT_CONSUMERCONFIG
#define INCLUDED_RMQT_CONSUMERCONFIG
 
#include <rmqt_properties.h>
#include <rmqt_queue.h>
#include <rmqt_topology.h>
 
#include <bdlmt_threadpool.h>
 
#include <bsl_cstdint.h>
#include <bsl_memory.h>
#include <bsl_optional.h>
#include <bsl_vector.h>
 
namespace BloombergLP {
namespace rmqp {
class MessageTransformer; // forward declaration
}
 
namespace rmqt {
 
 
class ConsumerConfig {
  public:
    static const uint16_t s_defaultPrefetchCount;
 
    static bsl::string generateConsumerTag();
 
    explicit ConsumerConfig(
        const bsl::string& consumerTag          = generateConsumerTag(),
        uint16_t prefetchCount                  = s_defaultPrefetchCount,
        bdlmt::ThreadPool* threadpool           = 0,
        rmqt::Exclusive::Value exclusiveFlag    = rmqt::Exclusive::OFF,
        bsl::optional<int64_t> consumerPriority = bsl::optional<int64_t>());
 
    ~ConsumerConfig();
 
    // Getters
    const bsl::string& consumerTag() const { return d_consumerTag; }
 
    uint16_t prefetchCount() const { return d_prefetchCount; }
 
    bdlmt::ThreadPool* threadpool() const { return d_threadpool; }
 
    rmqt::Exclusive::Value exclusiveFlag() const { return d_exclusiveFlag; }
 
    bsl::optional<int64_t> consumerPriority() const
    {
        return d_consumerPriority;
    }
 
    const bsl::vector<bsl::shared_ptr<rmqp::MessageTransformer> >&
    transformers() const
    {
        return d_transformers;
    }
 
    // Setters
    ConsumerConfig& setConsumerTag(const bsl::string& consumerTag)
    {
        d_consumerTag = consumerTag;
        return *this;
    }
 
    ConsumerConfig& setPrefetchCount(uint16_t prefetchCount)
 
    {
        d_prefetchCount = prefetchCount;
        return *this;
    }
 
    ConsumerConfig& setThreadpool(bdlmt::ThreadPool* threadpool)
    {
        d_threadpool = threadpool;
        return *this;
    }
    ConsumerConfig&
    setExclusiveFlag(rmqt::Exclusive::Value exclusiveFlag = rmqt::Exclusive::ON)
    {
        d_exclusiveFlag = exclusiveFlag;
        return *this;
    }
 
    ConsumerConfig&
    setConsumerPriority(const bsl::optional<int64_t>& consumerPriority)
    {
        d_consumerPriority = consumerPriority;
        return *this;
    }
 
    ConsumerConfig&
    addTransformer(const bsl::shared_ptr<rmqp::MessageTransformer>& transformer)
    {
        d_transformers.push_back(transformer);
        return *this;
    }
 
  private:
    bsl::string d_consumerTag;
    uint16_t d_prefetchCount;
    bdlmt::ThreadPool* d_threadpool;
    rmqt::Exclusive::Value d_exclusiveFlag;
    bsl::optional<int64_t> d_consumerPriority;
    bsl::vector<bsl::shared_ptr<rmqp::MessageTransformer> > d_transformers;
};
 
} // namespace rmqt
} // namespace BloombergLP
 
#endif