Skip to content

Commit

Permalink
Merge pull request #65 from samhaar/ui
Browse files Browse the repository at this point in the history
fixes gRPC controller server streams display issue
  • Loading branch information
garyslootskiy authored Nov 11, 2020
2 parents a5f840f + f6de13e commit b8fcc82
Showing 1 changed file with 9 additions and 27 deletions.
36 changes: 9 additions & 27 deletions main_process/main_grpcController.js
Original file line number Diff line number Diff line change
Expand Up @@ -6,24 +6,18 @@ const protoLoader = require("@grpc/proto-loader");
const grpcController = {};

grpcController.openGrpcConnection = (event, reqResObj) => {
//check for connection, if not open one


const { service, rpc, packageName, url, queryArr } = reqResObj;

reqResObj.connectionType = "GRPC";
reqResObj.response.times = [];

// build out variables from reqresObj properties
const { service, rpc, packageName, url, queryArr } = reqResObj;
const services = reqResObj.servicesObj;
if (reqResObj.response.events === null) {
reqResObj.response.events = [];
}
if (reqResObj.response.headers === null) {
reqResObj.response.headers = {};
}

reqResObj.response.headers = {};
reqResObj.response.events = [];

// go through services object, find service where name matches our passed
// in service, then grab the rpc list of that service, also save that service
// let rpcList;
const services = reqResObj.servicesObj;
let foundService;
let rpcType;
let foundRpc;
Expand Down Expand Up @@ -66,7 +60,6 @@ grpcController.openGrpcConnection = (event, reqResObj) => {
}

if (rpcType === "UNARY") {
console.log("\n \n inside UNARY if statement");
const query = reqResObj.queryArr[0];
const time = {};

Expand Down Expand Up @@ -105,7 +98,6 @@ grpcController.openGrpcConnection = (event, reqResObj) => {
});

} else if (rpcType === "SERVER STREAM") {
console.log("SERVER STREAM inside gRPC");
const timesArr = [];
// Open Connection for SERVER Stream
reqResObj.connection = "open";
Expand All @@ -115,18 +107,10 @@ grpcController.openGrpcConnection = (event, reqResObj) => {
const time = {};
time.timeReceived = Date.now();
time.timeSent = reqResObj.timeSent;
// add server response to reqResObj and dispatch to state/store
if (!reqResObj.response.events[0]) {
reqResObj.response.events[0] = {}
reqResObj.response.events[0].response = [];
reqResObj.response.events[0].response.push(resp)
}
else {
reqResObj.response.events[0].response.push(resp)
}
reqResObj.response.times.push(time);
reqResObj.timeReceived = time.timeReceived; // overwritten on each call to get the final value


reqResObj.response.events.push(resp);
event.sender.send("reqResUpdate", reqResObj);
});
call.on("error", () => {
Expand All @@ -150,7 +134,6 @@ grpcController.openGrpcConnection = (event, reqResObj) => {
event.sender.send("reqResUpdate", reqResObj);
});
} else if (rpcType === "CLIENT STREAM") {
console.log("CLIENT STREAM inside gRPC");
// create call and open client stream connection
reqResObj.connection = "open";
const timeSent = Date.now();
Expand Down Expand Up @@ -200,7 +183,6 @@ grpcController.openGrpcConnection = (event, reqResObj) => {

//else BIDIRECTIONAL
else {
console.log("BIDIRECTIONAL gRPC");
let counter = 0;
const call = client[rpc](meta);
call.on("data", (response) => {
Expand Down

0 comments on commit b8fcc82

Please sign in to comment.