library/cpp/messagebus/test/example/server/server.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

#include <library/cpp/messagebus/test/example/common/proto.h>

using namespace NBus;
using namespace NCalculator;

namespace NCalculator {
    struct TCalculatorServer: public IBusServerHandler {
        TCalculatorProtocol Proto;
        TBusMessageQueuePtr MessageQueue;
        TBusServerSessionPtr ServerSession;

        TCalculatorServer() {
            MessageQueue = CreateMessageQueue();
            TBusServerSessionConfig config;
            ServerSession = TBusServerSession::Create(&Proto, this, config, MessageQueue);
        }

        ~TCalculatorServer() override {
            MessageQueue->Stop();
        }

        void OnMessage(TOnMessageContext& request) override {
            if (request.GetMessage()->GetHeader()->Type == TRequestSum::MessageType) {
                TRequestSum* requestSum = VerifyDynamicCast<TRequestSum*>(request.GetMessage());
                int a = requestSum->Record.GetA();
                int b = requestSum->Record.GetB();
                int result = a + b;
                Cerr << "requested " << a << " + " << b << ", sending " << result << "\n";
                TAutoPtr<TResponse> response(new TResponse);
                response->Record.SetResult(result);
                request.SendReplyMove(response);
            } else if (request.GetMessage()->GetHeader()->Type == TRequestMul::MessageType) {
                TRequestMul* requestMul = VerifyDynamicCast<TRequestMul*>(request.GetMessage());
                int a = requestMul->Record.GetA();
                int b = requestMul->Record.GetB();
                int result = a * b;
                Cerr << "requested " << a << " * " << b << ", sending " << result << "\n";
                TAutoPtr<TResponse> response(new TResponse);
                response->Record.SetResult(result);
                request.SendReplyMove(response);
            } else {
                Y_ABORT("unknown request");
            }
        }
    };
}

int main(int, char**) {
    TCalculatorServer server;

    Cerr << "listening on port " << server.ServerSession->GetActualListenPort() << "\n";

    for (;;) {
        Sleep(TDuration::Seconds(1));
    }

    return 0;
}