|
1 | 1 | import * as grpc from '@grpc/grpc-js';
|
2 |
| -import readline from 'readline'; |
3 |
| -import { ChatClient } from './proto/chat_grpc_pb'; |
4 |
| -import { ClientMessage, ServerMessage } from './proto/chat_pb'; |
| 2 | +import { ExampleClient } from './proto/example_grpc_pb'; |
| 3 | +import { ClientMessage, ServerMessage } from './proto/example_pb'; |
5 | 4 |
|
6 |
| -const rl = readline.createInterface({ |
7 |
| - input: process.stdin, |
8 |
| - output: process.stdout, |
9 |
| -}); |
| 5 | +const host = '0.0.0.0:9090'; |
10 | 6 |
|
11 |
| -const client = new ChatClient( |
12 |
| - '0.0.0.0:9090', |
13 |
| - grpc.credentials.createInsecure() |
14 |
| -); |
| 7 | +const client = new ExampleClient(host, grpc.credentials.createInsecure()); |
15 | 8 |
|
16 | 9 | const deadline = new Date();
|
17 | 10 | deadline.setSeconds(deadline.getSeconds() + 5);
|
18 | 11 | client.waitForReady(deadline, (error?: Error) => {
|
19 | 12 | if (error) {
|
20 |
| - console.log(`Client error: ${error}`); |
| 13 | + console.log(`Client connect error: ${error.message}`); |
21 | 14 | } else {
|
22 |
| - rl.question("What's your name? ", startChat); |
| 15 | + onClientReady(); |
23 | 16 | }
|
24 | 17 | });
|
25 | 18 |
|
26 |
| -function startChat(user: string) { |
| 19 | +function onClientReady() { |
| 20 | + switch (process.argv[process.argv.length - 1]) { |
| 21 | + case '--unary': |
| 22 | + doUnaryCall(); |
| 23 | + break; |
| 24 | + case '--server-streaming': |
| 25 | + doServerStreamingCall(); |
| 26 | + break; |
| 27 | + case '--client-streaming': |
| 28 | + doClientStreamingCall(); |
| 29 | + break; |
| 30 | + case '--bidi-streaming': |
| 31 | + doBidirectionalStreamingCallStreamingCall(); |
| 32 | + break; |
| 33 | + default: |
| 34 | + throw new Error('Example not specified'); |
| 35 | + } |
| 36 | +} |
| 37 | + |
| 38 | +function doUnaryCall() { |
27 | 39 | const clientMessage = new ClientMessage();
|
28 |
| - clientMessage.setUser(user); |
29 |
| - const channel = client.join(clientMessage); |
30 |
| - channel.on('data', (message: ServerMessage) => { |
31 |
| - if (message.getUser() == user) { |
32 |
| - return; |
| 40 | + clientMessage.setClientMessage('Message from client'); |
| 41 | + client.unaryCall( |
| 42 | + clientMessage, |
| 43 | + (error: grpc.ServiceError | null, serverMessage?: ServerMessage) => { |
| 44 | + if (error) { |
| 45 | + console.error(error.message); |
| 46 | + } else if (serverMessage) { |
| 47 | + console.log( |
| 48 | + `(client) Got server message: ${serverMessage.getServerMessage()}` |
| 49 | + ); |
| 50 | + } |
33 | 51 | }
|
34 |
| - console.log(`${message.getUser()}: ${message.getText()}`); |
| 52 | + ); |
| 53 | +} |
| 54 | + |
| 55 | +function doServerStreamingCall() { |
| 56 | + const clientMessage = new ClientMessage(); |
| 57 | + clientMessage.setClientMessage('Message from client'); |
| 58 | + const stream = client.serverStreamingCall(clientMessage); |
| 59 | + stream.on('data', (serverMessage: ServerMessage) => { |
| 60 | + console.log( |
| 61 | + `(client) Got server message: ${serverMessage.getServerMessage()}` |
| 62 | + ); |
35 | 63 | });
|
36 |
| - rl.on('line', function (text) { |
37 |
| - const clientMessage = new ClientMessage(); |
38 |
| - clientMessage.setUser(user); |
39 |
| - clientMessage.setText(text); |
40 |
| - client.send(clientMessage, () => { |
41 |
| - // |
42 |
| - }); |
| 64 | +} |
| 65 | + |
| 66 | +function doClientStreamingCall() { |
| 67 | + const stream = client.clientStreamingCall( |
| 68 | + (error: grpc.ServiceError | null, serverMessage?: ServerMessage) => { |
| 69 | + if (error) { |
| 70 | + console.error(error.message); |
| 71 | + } else if (serverMessage) { |
| 72 | + console.log( |
| 73 | + `(client) Got server message: ${serverMessage.getServerMessage()}` |
| 74 | + ); |
| 75 | + } |
| 76 | + } |
| 77 | + ); |
| 78 | + const clientMessage = new ClientMessage(); |
| 79 | + clientMessage.setClientMessage('Message from client'); |
| 80 | + stream.write(clientMessage); |
| 81 | + |
| 82 | + // eslint-disable-next-line @typescript-eslint/no-empty-function |
| 83 | + setTimeout(() => {}, 1000); |
| 84 | +} |
| 85 | + |
| 86 | +function doBidirectionalStreamingCallStreamingCall() { |
| 87 | + const stream = client.bidirectionalStreamingCall(); |
| 88 | + |
| 89 | + // Server stream |
| 90 | + stream.on('data', (serverMessage: ServerMessage) => { |
| 91 | + console.log( |
| 92 | + `(client) Got server message: ${serverMessage.getServerMessage()}` |
| 93 | + ); |
43 | 94 | });
|
| 95 | + |
| 96 | + // Client stream |
| 97 | + const clientMessage = new ClientMessage(); |
| 98 | + clientMessage.setClientMessage('Message from client'); |
| 99 | + stream.write(clientMessage); |
44 | 100 | }
|
0 commit comments