library/cpp/messagebus/test/example/client/client.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

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

#include <util/random/random.h> 
 
using namespace NBus; 
using namespace NCalculator; 
 
namespace NCalculator { 
    struct TCalculatorClient: public IBusClientHandler {
        TCalculatorProtocol Proto; 
        TBusMessageQueuePtr MessageQueue; 
        TBusClientSessionPtr ClientSession; 
 
        TCalculatorClient() { 
            MessageQueue = CreateMessageQueue(); 
            TBusClientSessionConfig config; 
            config.TotalTimeout = 2 * 1000; 
            ClientSession = TBusClientSession::Create(&Proto, this, config, MessageQueue); 
        } 
 
        ~TCalculatorClient() override {
            MessageQueue->Stop(); 
        } 
 
        void OnReply(TAutoPtr<TBusMessage> request, TAutoPtr<TBusMessage> response0) override {
            Y_VERIFY(response0->GetHeader()->Type == TResponse::MessageType, "wrong response");
            TResponse* response = VerifyDynamicCast<TResponse*>(response0.Get()); 
            if (request->GetHeader()->Type == TRequestSum::MessageType) { 
                TRequestSum* requestSum = VerifyDynamicCast<TRequestSum*>(request.Get()); 
                int a = requestSum->Record.GetA(); 
                int b = requestSum->Record.GetB(); 
                Cerr << a << " + " << b << " = " << response->Record.GetResult() << "\n"; 
            } else if (request->GetHeader()->Type == TRequestMul::MessageType) { 
                TRequestMul* requestMul = VerifyDynamicCast<TRequestMul*>(request.Get()); 
                int a = requestMul->Record.GetA(); 
                int b = requestMul->Record.GetB(); 
                Cerr << a << " * " << b << " = " << response->Record.GetResult() << "\n"; 
            } else { 
                Y_FAIL("unknown request");
            } 
        } 
 
        void OnError(TAutoPtr<TBusMessage>, EMessageStatus status) override {
            Cerr << "got error " << status << "\n"; 
        } 
    }; 
 
} 
 
int main(int, char**) { 
    TCalculatorClient client; 
 
    for (;;) { 
        TNetAddr addr(TNetAddr("127.0.0.1", TCalculatorProtocol().GetPort())); 
 
        int a = RandomNumber<unsigned>(10); 
        int b = RandomNumber<unsigned>(10); 
        EMessageStatus ok; 
        if (RandomNumber<bool>()) { 
            TAutoPtr<TRequestSum> request(new TRequestSum); 
            request->Record.SetA(a); 
            request->Record.SetB(b); 
            Cerr << "sending " << a << " + " << b << "\n"; 
            ok = client.ClientSession->SendMessageAutoPtr(request, &addr); 
        } else { 
            TAutoPtr<TRequestMul> request(new TRequestMul); 
            request->Record.SetA(a); 
            request->Record.SetB(b); 
            Cerr << "sending " << a << " * " << b << "\n"; 
            ok = client.ClientSession->SendMessageAutoPtr(request, &addr); 
        } 
 
        if (ok != MESSAGE_OK) { 
            Cerr << "failed to send message " << ok << "\n"; 
        } 
 
        Sleep(TDuration::Seconds(1)); 
    } 
 
    return 0; 
}