rmqa_rabbitcontextoptions.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.
 
// rmqa_rabbitcontextoptions.h
#ifndef INCLUDED_RMQA_RABBITCONTEXTOPTIONS
#define INCLUDED_RMQA_RABBITCONTEXTOPTIONS
 
#include <rmqp_metricpublisher.h>
 
#include <rmqp_consumertracing.h>
#include <rmqp_producertracing.h>
#include <rmqt_fieldvalue.h>
#include <rmqt_properties.h>
#include <rmqt_result.h>
 
#include <bdlmt_threadpool.h>
#include <bsl_memory.h>
#include <bsl_set.h>
#include <bsls_timeinterval.h>
 
namespace BloombergLP {
namespace rmqa {
 
 
class RabbitContextOptions {
  public:
    typedef bsl::set<bsl::string> Tunables;
 
    RabbitContextOptions();
 
    RabbitContextOptions& setThreadpool(bdlmt::ThreadPool* threadpool);
 
    RabbitContextOptions& setMetricPublisher(
        const bsl::shared_ptr<rmqp::MetricPublisher>& metricPublisher);
 
    RabbitContextOptions&
    setErrorCallback(const rmqt::ErrorCallback& errorCallback);
 
    RabbitContextOptions& setClientProperty(const bsl::string& name,
                                            const rmqt::FieldValue& value);
 
    RabbitContextOptions&
    setMessageProcessingTimeout(const bsls::TimeInterval& timeout);
 
    RabbitContextOptions& setConnectionErrorThreshold(
        const bsl::optional<bsls::TimeInterval>& timeout);
 
    RabbitContextOptions& setConsumerTracing(
        const bsl::shared_ptr<rmqp::ConsumerTracing>& consumerTracing);
 
    RabbitContextOptions& setProducerTracing(
        const bsl::shared_ptr<rmqp::ProducerTracing>& producerTracing);
 
    RabbitContextOptions& useRabbitMQFieldValueEncoding(bool rabbitEncoding);
 
    RabbitContextOptions&
    setShuffleConnectionEndpoints(bool shuffleConnectionEndpoints);
 
    bdlmt::ThreadPool* threadpool() const { return d_threadpool; }
 
    const bsl::shared_ptr<rmqp::MetricPublisher>& metricPublisher() const
    {
        return d_metricPublisher;
    }
 
    const rmqt::ErrorCallback& errorCallback() const { return d_onError; }
 
    const rmqt::FieldTable& clientProperties() const
    {
        return d_clientProperties;
    }
 
    const bsls::TimeInterval& messageProcessingTimeout() const
    {
        return d_messageProcessingTimeout;
    }
 
    const bsl::optional<bsls::TimeInterval>& connectionErrorThreshold() const
    {
        return d_connectionErrorThreshold;
    }
 
    const rmqt::Tunables& tunables() const { return d_tunables; }
 
    const bsl::shared_ptr<rmqp::ConsumerTracing>& consumerTracing() const
    {
        return d_consumerTracing;
    }
 
    const bsl::shared_ptr<rmqp::ProducerTracing>& producerTracing() const
    {
        return d_producerTracing;
    }
 
    const bsl::optional<bool>& shuffleConnectionEndpoints() const
    {
        return d_shuffleConnectionEndpoints;
    }
 
#ifdef USES_LIBRMQ_EXPERIMENTAL_FEATURES
    RabbitContextOptions& setTunable(const bsl::string& tunable);
#endif
 
  private:
    static const int DEFAULT_MESSAGE_PROCESSING_TIMEOUT = 60;
    bdlmt::ThreadPool* d_threadpool;
    rmqt::ErrorCallback d_onError;
    bsl::shared_ptr<rmqp::MetricPublisher> d_metricPublisher;
    rmqt::FieldTable d_clientProperties;
    bsls::TimeInterval d_messageProcessingTimeout;
    rmqt::Tunables d_tunables;
    bsl::optional<bsls::TimeInterval> d_connectionErrorThreshold;
    bsl::shared_ptr<rmqp::ConsumerTracing> d_consumerTracing;
    bsl::shared_ptr<rmqp::ProducerTracing> d_producerTracing;
    bsl::optional<bool> d_shuffleConnectionEndpoints;
};
 
} // namespace rmqa
} // namespace BloombergLP
 
#endif