uni_push_pull.cpp

#include <functional>
#include <string>

#include <spdlog/spdlog.h>

#include "fantasy.hpp"

inline std::string addr{"tcp://127.0.0.1:5878"};

void start(std::function<void()> func) {
    std::thread(std::move(func)).detach();
}

#ifdef __cpp_impl_coroutine
#ifdef _ASIO_
struct CoroHelloWorldReceiver final : public fantasy::AsioCoroHelloWorldReceiverHandler {
    virtual asio::awaitable<void> hello_world(std::string in) noexcept override {
        spdlog::info("CoroHelloWorldReceiver::hello_world: {}", in);
        co_return;
    }
    virtual asio::awaitable<void> notice(int32_t in, std::string info) noexcept override {
        spdlog::info("CoroHelloWorldReceiver::notice: {}: {}", in, info);
        co_return;
    }
};
#else
struct CoroHelloWorldReceiver final : public fantasy::FrpcCoroHelloWorldReceiverHandler {
    virtual frpc::Task<void> hello_world(std::string in) noexcept override {
        spdlog::info("Frpc CoroHelloWorldReceiver::hello_world: {}", in);
        co_return;
    }
    virtual frpc::Task<void> notice(int32_t in, std::string info) noexcept override {
        spdlog::info("Frpc CoroHelloWorldReceiver::notice: {}: {}", in, info);
        co_return;
    }
};
#endif
#else
struct HelloWorldReceiverHandler final : public fantasy::HelloWorldReceiverHandler {
    HelloWorldReceiverHandler() = default;
    HelloWorldReceiverHandler(const std::string& label)
        : label(label) {
    }

    virtual void hello_world(std::string in) noexcept override {
        spdlog::info("HelloWorldReceiverHandler::hello_world: {}, {}", label, in);
        return;
    }
    virtual void notice(int32_t in, std::string info) noexcept override {
        spdlog::info("HelloWorldReceiverHandler::notice: {}, {}: {}", label, in, info);
        return;
    }

    std::string label{"test"};
};
#endif

void start_push() {
    frpc::ChannelConfig pub_config{};
    pub_config.socktype = zmq::socket_type::push;
    pub_config.bind = true;
    pub_config.addr = addr;
    auto sender = fantasy::HelloWorldSender::create(pub_config);
    int i = 10;
    while (i--) {
        sender->hello_world(std::to_string(i) + "_frpc_push_01");
        sender->notice(i, "hello world");
        std::this_thread::sleep_for(std::chrono::seconds(1));
    }
}

int main() {
    start(start_push);

    frpc::ChannelConfig sub_config{};
    sub_config.addr = addr;
    sub_config.socktype = zmq::socket_type::pull;
    sub_config.bind = false;

    auto receiver = fantasy::HelloWorldReceiver::create(
        sub_config,
#ifdef __cpp_impl_coroutine
        std::make_shared<CoroHelloWorldReceiver>(),
#else
        std::make_shared<HelloWorldReceiverHandler>("pull-001"),
#endif
        [](auto error) {
            spdlog::error("{}", error);
        });
    receiver->start();

    auto receiver1 = fantasy::HelloWorldReceiver::create(
        sub_config,
#ifdef __cpp_impl_coroutine
        std::make_shared<CoroHelloWorldReceiver>(),
#else
        std::make_shared<HelloWorldReceiverHandler>("pull-002"),
#endif
        [](auto error) {
            spdlog::error("{}", error);
        });
    receiver1->start();
    std::this_thread::sleep_for(std::chrono::seconds(15));

    return 0;
}