server_builder_impl.h

Go to the documentation of this file.

/*
 * Copyright 2023 MusicScience37 (Kenta Kabashima)
 *
 * 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.
 */
#pragma once
 
#include <memory>
#include <utility>
#include <vector>
 
#include "msgpack_rpc/addresses/uri.h"
#include "msgpack_rpc/common/msgpack_rpc_exception.h"
#include "msgpack_rpc/common/status_code.h"
#include "msgpack_rpc/executors/i_async_executor.h"
#include "msgpack_rpc/logging/logger.h"
#include "msgpack_rpc/methods/i_method.h"
#include "msgpack_rpc/methods/i_method_processor.h"
#include "msgpack_rpc/methods/method_processor.h"
#include "msgpack_rpc/servers/impl/i_server_builder_impl.h"
#include "msgpack_rpc/servers/impl/i_server_impl.h"
#include "msgpack_rpc/servers/impl/server_impl.h"
#include "msgpack_rpc/transport/backend_list.h"
#include "msgpack_rpc/transport/i_acceptor.h"
#include "msgpack_rpc/transport/i_acceptor_factory.h"
#include "msgpack_rpc/transport/i_backend.h"
 
namespace msgpack_rpc::servers::impl {

class ServerBuilderImpl final : public IServerBuilderImpl {
public:
    ServerBuilderImpl(std::shared_ptr<executors::IAsyncExecutor> executor,
        std::shared_ptr<logging::Logger> logger,
        transport::BackendList backends, std::vector<addresses::URI> uris)
        : executor_(std::move(executor)),
          logger_(std::move(logger)),
          backends_(std::move(backends)),
          uris_(std::move(uris)),
          processor_(methods::create_method_processor(logger_)) {}

    void register_protocol(
        std::shared_ptr<transport::IBackend> backend) override {
        backends_.append(backend);
    }

    void listen_to(addresses::URI uri) override {
        uris_.push_back(std::move(uri));
    }

    void add_method(std::unique_ptr<methods::IMethod> method) override {
        processor_->append(std::move(method));
    }

    [[nodiscard]] std::unique_ptr<IServerImpl> build() override {
        if (uris_.empty()) {
            throw MsgpackRPCException(StatusCode::PRECONDITION_NOT_MET,
                "No URI to listen to was given.");
        }
 
        std::vector<std::shared_ptr<transport::IAcceptor>> acceptors;
        for (const auto& uri : uris_) {
            const auto backend = backends_.find(uri.scheme());
 
            const auto added_acceptors =
                backend->create_acceptor_factory()->create(uri);
            for (const auto& acceptor : added_acceptors) {
                acceptors.push_back(acceptor);
            }
        }
 
        if (acceptors.empty()) {
            throw MsgpackRPCException(StatusCode::PRECONDITION_NOT_MET,
                "All URI set to listen to was unusable.");
        }
 
        auto server = std::make_unique<ServerImpl>(
            std::move(acceptors), std::move(processor_), executor_, logger_);
        server->start();
 
        return server;
    }

    [[nodiscard]] std::shared_ptr<logging::Logger> logger() override {
        return logger_;
    }
 
private:
    std::shared_ptr<executors::IAsyncExecutor> executor_;

    std::shared_ptr<logging::Logger> logger_;

    transport::BackendList backends_{};

    std::vector<addresses::URI> uris_{};

    std::unique_ptr<methods::IMethodProcessor> processor_;
};
 
}  // namespace msgpack_rpc::servers::impl