call_future.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 <chrono>
#include <memory>
#include <utility>
 
#include "msgpack_rpc/clients/impl/i_call_future_impl.h"
#include "msgpack_rpc/clients/server_exception.h"
#include "msgpack_rpc/messages/call_result.h"
 
namespace msgpack_rpc::clients {

template <typename Result>
class CallFuture {
public:
    explicit CallFuture(std::shared_ptr<impl::ICallFutureImpl> impl)
        : impl_(std::move(impl)) {}

    [[nodiscard]] Result get_result() {
        const auto call_result = impl_->get_result();
        return get_from_call_result(call_result);
    }

    [[nodiscard]] Result get_result_within(std::chrono::nanoseconds timeout) {
        const auto call_result = impl_->get_result_within(timeout);
        return get_from_call_result(call_result);
    }
 
private:
    [[nodiscard]] static Result get_from_call_result(
        const messages::CallResult& call_result) {
        if (call_result.is_success()) {
            return call_result.result_as<Result>();
        }
        throw ServerException(call_result.object(), call_result.zone());
    }

    std::shared_ptr<impl::ICallFutureImpl> impl_;
};

template <>
class CallFuture<void> {
public:
    explicit CallFuture(std::shared_ptr<impl::ICallFutureImpl> impl)
        : impl_(std::move(impl)) {}

    void get_result() {
        const auto call_result = impl_->get_result();
        get_from_call_result(call_result);
    }

    void get_result_within(std::chrono::nanoseconds timeout) {
        const auto call_result = impl_->get_result_within(timeout);
        get_from_call_result(call_result);
    }
 
private:
    static void get_from_call_result(const messages::CallResult& call_result) {
        if (call_result.is_success()) {
            return;
        }
        throw ServerException(call_result.object(), call_result.zone());
    }

    std::shared_ptr<impl::ICallFutureImpl> impl_;
};
 
}  // namespace msgpack_rpc::clients