forked from tropicRPC/Mock-gRPC-API
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.js
90 lines (81 loc) · 2.03 KB
/
server.js
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
82
83
84
85
86
87
88
89
90
const grpc = require('grpc');
const protoLoader = require('@grpc/proto-loader');
const packageDef = protoLoader.loadSync('demo.proto', {});
const grpcObject = grpc.loadPackageDefinition(packageDef);
const tropicalAnimals = grpcObject.tropicalAnimals;
const server = new grpc.Server();
server.bind('0.0.0.0:40000', grpc.ServerCredentials.createInsecure());
server.addService(tropicalAnimals.ManageService.service, {
sendAnimal,
getAnimals,
getStreamAnimals,
sendStreamAnimals,
sendAndGetAnimalSoundsPingPong,
sendAndGetAnimalSoundsPost,
});
server.start();
const animals = [];
// Unary RPC
function sendAnimal(call, callback) {
const request = call.request;
const animal = {
animalId: animals.length + 1,
animal: request.animal,
sound: request.sound,
};
animals.push(animal);
console.log('ANIMAL******', animal);
callback(null, animal);
}
// Unary RPC
function getAnimals(call, callback) {
callback(null, { animals });
}
// Server Streaming RPC
function getStreamAnimals(call) {
animals.forEach((animal) => call.write(animal));
call.end();
}
// Client Streaming RPC
function sendStreamAnimals(call, callback) {
call.on('data', (data) => {
const animal = {
animalId: animals.length + 1,
animal: data.animal,
sound: data.sound,
};
animals.push(animal);
});
call.on('end', () => {
callback(null, { animals });
});
}
// Bi-Directional Streaming
function sendAndGetAnimalSoundsPingPong(call) {
call.on('data', (data) => {
const animal = {
animalId: animals.length + 1,
animal: data.animal,
sound: data.sound,
};
animals.push(animal);
call.write(animals[animals.length - 1]);
});
call.on('end', () => {
call.end();
});
}
function sendAndGetAnimalSoundsPost(call) {
call.on('data', (data) => {
const animal = {
animalId: animals.length + 1,
animal: data.animal,
sound: data.sound,
};
animals.push(animal);
});
call.on('end', () => {
animals.forEach((animal) => call.write(animal));
call.end();
});
}