rpc-server.cpp

#include "server.hpp"

class rpc_server : public server {
public:
    rpc_server(int port):
    server(port) {
        
    }
    virtual scheduler::task_t process_request(scheduler &sche, int fd, void *request, int32_t sz) override {
        int32_t call_id = *(int *)buffer;
        switch(call_id) {
            case co_rpc::CALL_ID::ADD_NUMBER:
                int32_t res = rpc_add_number_impl(buffer);
                co_await co_syscall::write_package(sche, fd, &res, sizeof(res));
                break;
        }
        co_return 0;
    }
};


int main(int argc, char **argv) {
    if (argc != 2) {
        std::cout << "Usage: server <port>" << std::endl;
        return 0;
    }
    int port = std::atoi(argv[1]);
    scheduler sche;
    rpc_server sv(port);

    sche.schedule(sv.co_listen(sche));
    sche.run();

    return 0;
}