通過NodeJS輕松使用GRPC和協(xié)議緩沖區(qū)的方法

更新時間：2024年10月16日 09:57:49 作者：學會種一棵樹

本文介紹了GRPC和協(xié)議緩沖區(qū)的基本概念,并展示了如何在NodeJS應用程序中使用它們,GRPC是一個高性能RPC框架,協(xié)議緩沖區(qū)則用于定義服務和序列化消息,本文給大家介紹如何在NodeJS應用程序中使用GRPC和協(xié)議緩沖區(qū),感興趣的朋友一起看看吧

GRPC是什么

GRPC是一個開源高性能RPC框架，那么RPC到底是做什么的呢？請看下面的例子

function getEmployee() {
  return "ABCD";
}
function greetEmployee()
{
  let employee = getEmployee();
  console.log("Greet",employee) 
}

在這里，我們有一個getEmployee函數(shù)，它返回一個Employee Name，另一個函數(shù)greetEmployee，它調(diào)用getEmployee并獲取該員工的Name，然后打印一個Greeting。

這里的greetEmployee調(diào)用getEmployee是一個常規(guī)函數(shù)調(diào)用。

現(xiàn)在，如果getEmployee和greetEmployee函數(shù)位于不同的地址空間中，或者它們位于由網(wǎng)絡分隔的2個不同的主機中，則該函數(shù)調(diào)用稱為“遠程過程調(diào)用”(Remote Procedure Call, RPC)。在這里，具有getEmployee函數(shù)的System充當服務器，而具有greetEmployee函數(shù)的System充當客戶端。

什么是Protocol Buffer (協(xié)議緩沖區(qū))

協(xié)議緩沖區(qū)是默認情況下在GRPC中使用的接口定義語言。

它有助于定義服務器提供的各種服務
它有助于定義系統(tǒng)中使用的有效負載的結(jié)構(gòu)
它有助于對消息進行序列化（轉(zhuǎn)換為特殊的二進制格式），并通過服務器和客戶端之間的線路進行發(fā)送。

在本文后面的部分中，我們將在使用NodeJS應用程序時看到如何使用protocol buffers(協(xié)議緩沖區(qū))。

支持哪些不同類型的RPC？

Unary RPC

這是可用的最簡單的RPC?？蛻舳嗽诖颂幭蚍掌靼l(fā)送請求消息。服務器處理該請求，然后將響應消息發(fā)送回客戶端

在本文中，這是我們將重點介紹的grpc。

Server Streaming RPC

在此RPC中，客戶端將請求消息發(fā)送到服務器，然后服務器以流方式將一系列消息發(fā)送回客戶端。

Client Streaming RPC

在此RPC中，客戶端以流方式向服務器發(fā)送一系列消息。然后，服務器處理所有這些請求，然后將響應消息發(fā)送回客戶端。

Bidirectional Streaming RPC

在此RPC中，客戶端以流方式向服務器發(fā)送一系列消息。然后，服務器處理請求，然后以流方式將一系列消息發(fā)送回客戶端。

如何在NodeJS中使用GRPC和協(xié)議緩沖區(qū)

使用以下命令創(chuàng)建一個名為grpc-nodejs-demo的文件夾并在其中初始化nodejs

mkdir grpc-nodejs-demo
cd grpc-nodejs-demo
npm init

這將創(chuàng)建一個package.json文件。

修改package.json文件

將package.json文件替換為以下內(nèi)容

{
  "name": "grpc-nodejs-demo",
  "version": "1.0.0",
  "description": "",
  "main": "server.js",
  "scripts": {
    "test": "echo \"Error: no test specified\" && exit 1"
  },
  "dependencies": {
    "@grpc/proto-loader": "^0.1.0",
    "grpc": "^1.11.0",
    "lodash": "^4.6.1"
  },
  "author": "Name",
  "license": "MIT"
}

在這里我們添加了3個依賴項

@ grpc / proto_loader和grpc依賴項將幫助我們在應用程序中使用GRPC和協(xié)議緩沖區(qū)
lodash是通用實用程序依賴項。這將有助于簡化一些代碼邏輯

package.json文件更新后。運行以下命令以安裝依賴項

npm install

定義協(xié)議緩沖區(qū)

在此示例中，我們將構(gòu)建一個將employee ID作為輸入，并提供employee詳細信息作為輸出的服務。

所需的服務接口和有效負載將在協(xié)議緩沖區(qū)文件中指定。協(xié)議緩沖區(qū)文件的擴展名為.proto

現(xiàn)在，我們來創(chuàng)建.proto文件。

在項目中創(chuàng)建一個名為proto的文件夾。在proto文件夾中，創(chuàng)建一個名為employee.proto的文件，并向其中添加以下代碼

syntax = "proto3";
package employee;
service Employee {
  rpc getDetails (EmployeeRequest) returns (EmployeeResponse) {}
}
message EmployeeRequest {
  int32 id = 1;
}
message EmployeeResponse{
  EmployeeDetails message = 1;
}
message EmployeeDetails {
  int32 id = 1;
  string email = 2;
  string firstName = 3; 
  string lastName = 4;
}

那么，我們在這里到底做了什么？

syntax="proto3";表示我們要使用Protocol Buffer version 3。

package employee; 表示我們正在創(chuàng)建一個名為employee的程序包，我們將在其中定義我們的services

service Employee {
  rpc getDetails (EmployeeRequest) returns (EmployeeResponse) {}
}

上述腳本表明我們正在創(chuàng)建一個稱為Employee的服務。在此服務中，我們正在創(chuàng)建一個名為getDetails的函數(shù)（rpc），該函數(shù)接受EmployeeRequest類型的輸入并以EmployeeResponse格式提供響應

接下來，我們需要定義EmployeeRequest和EmployeeResponse。這是在以下腳本中完成的

message EmployeeRequest {
  int32 id = 1;
}
message EmployeeResponse{
  EmployeeDetails message = 1;
}
message EmployeeDetails {
  int32 id = 1;
  string email = 2;
  string firstName = 3; 
  string lastName = 4;
}

在這里，我們看到消息EmployeeRequest具有單個類型為int32和名稱id的字段。此處分配的數(shù)字1是字段號，它在消息的編碼和解碼過程中提供幫助。定義的每個字段應具有唯一的字段號

我們還看到EmployeeResponse有一個類型為EmployeeDetails的自定義字段，并且名稱消息的字段編號為1。這意味著甚至也必須定義EmployeeDetails，如上所示。

EmployeeDetails具有4個字段，包括類型int32和string。它們都有唯一的字段編號(unique field numbers)

Field numbers between 1 -15 use 1 byte of space during encoding. and field numbers from 2 - 2047 uses 2 bytes for encoding and hence will take up more space. So try to design in such a way that the field numbers are between 1 - 15 as much as possible

1 -15之間的編號在編碼過程中使用1個字節(jié)的空間。并且2-2047的字段編號使用2個字節(jié)進行編碼，因此會占用更多空間。因此，請嘗試以盡可能多的字段編號介于1到15之間的方式進行設(shè)計

Creating the GRPC Server

創(chuàng)建一個名為server.js的文件

首先，讓我們包括我們需要的所有庫，并定義.proto文件所在的位置

const PROTO_PATH = __dirname + '/proto/employee.proto';
const grpc = require('grpc');
const protoLoader = require('@grpc/proto-loader');
const _ = require('lodash');

接下來，我們需要加載.proto文件。這是使用protoLoader庫的loadSync方法完成的。

let packageDefinition = protoLoader.loadSync(
    PROTO_PATH,
    {keepCase: true,
     longs: String,
     enums: String,
     defaults: true,
     oneofs: true
    }
);

接下來，從已加載的原始文件包定義中，我們需要獲取所需的包。這是使用以下腳本完成的

let employee_proto = grpc.loadPackageDefinition(packageDefinition).employee;

在這里，我們將employee包放入employee_proto變量中。

employee_proto現(xiàn)在將具有所有原型定義。

接下來，我們需要創(chuàng)建一些虛擬employees數(shù)據(jù)供服務器使用。創(chuàng)建一個名為data.js的文件，并將以下腳本添加到其中

let employees = [{
    id: 1,
    email: "abcd@abcd.com",
    firstName: "First1",
    lastName: "Last1"   
},
{
    id: 2,
    email: "xyz@xyz.com",
    firstName: "First2",
    lastName: "Last2"   
},
{
    id: 3,
    email: "temp@temp.com",
    firstName: "First3",
    lastName: "Last3"   
},
];
exports.employees = employees;

接下來，我們需要將data.js導入server.js。為此，在server.js中添加以下腳本

function main() {
  let server = new grpc.Server();
  server.addService(employee_proto.Employee.service, {getDetails: getDetails});
  server.bind('0.0.0.0:4500', grpc.ServerCredentials.createInsecure());
  server.start();
}

let server = new grpc.Server();

是創(chuàng)建新GRPC Server的腳本

在.proto文件中，我們注意到在Employee Service內(nèi)部有一個名為getDetails的函數(shù)。

server.addService（employee_proto.Employee.service，{getDetails：getDetails}）;

是我們在其中添加Service實現(xiàn)的腳本。該腳本表明，我們在employee_proto.Employee Service中添加了getDetails函數(shù)。然后，我們將此服務添加到服務器。

server.bind('0.0.0.0:4500', grpc.ServerCredentials.createInsecure()); 是指示服務器將在端口4500上啟動且沒有身份驗證的腳本

server.start(); 是實際啟動服務器的腳本

現(xiàn)在待處理的主要事情是實現(xiàn)getDetails函數(shù)。下面的腳本顯示了實現(xiàn)

function getDetails(call, callback) {
  callback(null, 
    {
       message: _.find(employees, { id: call.request.id })
    });
}

這里的call具有請求參數(shù)，而回callback 是我們需要定義實現(xiàn)的地方。

在callback 內(nèi)部，我們收到一條消息：_.find（employees，{id：call.request.id}），其中顯示以下內(nèi)容

從Input獲取員工ID-call.request.id
搜索employees列表以查找具有該ID的員工
Return employee詳細信息

這樣就完成了服務器的實現(xiàn)。這是server.js的完整腳本

const PROTO_PATH = __dirname + '/proto/employee.proto';
const grpc = require('grpc');
const protoLoader = require('@grpc/proto-loader');
const _ = require('lodash');
let packageDefinition = protoLoader.loadSync(
    PROTO_PATH,
    {keepCase: true,
     longs: String,
     enums: String,
     defaults: true,
     oneofs: true
    });
let employee_proto = grpc.loadPackageDefinition(packageDefinition).employee;
let {employees} = require('./data.js');
function getDetails(call, callback) {
  console.log('client call :%j', call)
  callback(null, 
    {
       message: _.find(employees, { id: call.request.id })
    });
}
function main() {
  let server = new grpc.Server();
  server.addService(employee_proto.Employee.service, {getDetails: getDetails});
  server.bind('0.0.0.0:4500', grpc.ServerCredentials.createInsecure());
  server.start();
}
main();

Creating the GRPC Client

創(chuàng)建一個名為client.js的文件將以下腳本復制到client.js

const PROTO_PATH = __dirname + '/proto/employee.proto';
const grpc = require('grpc');
const protoLoader = require('@grpc/proto-loader');
let packageDefinition = protoLoader.loadSync(
    PROTO_PATH,
    {keepCase: true,
     longs: String,
     enums: String,
     defaults: true,
     oneofs: true
    }
);
let employee_proto = grpc.loadPackageDefinition(packageDefinition).employee;

上面的腳本以與server.js中相同的方式將employee包加載到employee_proto變量中

接下來，我們需要一種可以調(diào)用RPC的方式。在這種情況下，我們需要能夠調(diào)用服務器中實現(xiàn)的getDetails函數(shù)。

為此，我們需要在客戶端中創(chuàng)建一個存根(stub)。這是使用以下腳本完成的。

  let client = new employee_proto.Employee('localhost:4500', grpc.credentials.createInsecure());

該client Stub將幫助我們調(diào)用在服務器上運行的Employee Service中定義的getDetails函數(shù)。服務器依次在端口4500上運行。代碼行還指示未使用身份驗證

最后，我們可以使用以下腳本調(diào)用getDetails函數(shù)

let employeeId = 1;
 client.getDetails({id: employeeId}, function(err, response) {
    console.log('Employee Details for Employee Id:',employeeId,'\n' ,response.message);
  });

如前所述，客戶端存根可以像正常函數(shù)調(diào)用一樣幫助我們在服務器中調(diào)用getDetails函數(shù)。為此，我們將employeeId作為輸入。

最后，Response 進入response 變量。然后，我們將打印響應消息。

完整的client.js代碼如下

const PROTO_PATH = __dirname + '/proto/employee.proto';
const grpc = require('grpc');
const protoLoader = require('@grpc/proto-loader');
let packageDefinition = protoLoader.loadSync(
    PROTO_PATH,
    {keepCase: true,
     longs: String,
     enums: String,
     defaults: true,
     oneofs: true
    });
let employee_proto = grpc.loadPackageDefinition(packageDefinition).employee;
function main() {
  let client = new employee_proto.Employee('localhost:4500',
                                       grpc.credentials.createInsecure());
  let employeeId;
  if (process.argv.length >= 3) {
    employeeId = process.argv[2];
  } else {
    employeeId = 1;
  }
  client.getDetails({id: employeeId}, function(err, response) {
    console.log('Employee Details for Employee Id:',employeeId,'\n' ,response.message);
  });
}
main();