nodejs websocket 发送protoBuf数据到服务端
当使用Node.js进行WebSocket通信时，发送经过Protobuf编码的数据到服务端是一个常见的需求。
在这篇博客中，我们将通过一个示例来演示如何实现这个功能。
假设我们有一个WebSocket服务端，地址为ws://example.com/ws
我要发送pb_http_frame中的Request对象，这个对象中有body字段是来自pb_login中Req对象
我们有以下两个Protobuf定义文件，登录时用到pb_login.Req，解析时用到pb_http_frame.Response（封装返回的结果，有code、body）：
pb_login.proto:
syntax = "proto3";
message Req {
int32 uid = 1;
string client = 2;
string platform = 3;
string token = 4;
}
message Resp {
// 定义Resp消息的字段
}
pb_http_frame.proto:
syntax = "proto3";
message Request {
int32 cmd = 1;
int32 sub = 2;
int32 stt = 3;
bytes body = 4;
}
message Response {
int32 code = 1;
bytes body = 2;
}首先，我们需要安装ws和protobufjs这两个库。你可以使用npm命令进行安装：
npm install ws protobufjs
接下来，我们开始编写代码。首先，引入所需的库和模块：
const WebSocket = require('ws');
const protobuf = require('protobufjs');然后，创建WebSocket连接并加载Protobuf定义文件：
const ws = new WebSocket('ws://example.com/ws');
protobuf.load('protos/pb_login.proto', function (err, root) {
if (err) throw err;
const frameRoot = protobuf.loadSync('protos/pb_http_frame.proto');
// 在这里编写后续的代码
});在加载Protobuf定义文件后，我们可以获取消息定义并创建消息对象：
const ProtoLoginReq = root.lookupType('Req');
const ProtoBuffFrameRequest = frameRoot.lookupType('Request');
const ProtoBuffFrameResponse = frameRoot.lookupType('Response');
const protoLoginReqPayload = {
uid: 112000087,
client: '4.3.02',
platform: 'ios',
token: '1F6BB53A6504904145CB38EEB57532DBB9FF5868F5A0FD61'
};
const protoLoginReqMessage = ProtoLoginReq.create(protoLoginReqPayload);
const protoBuffFrameRequestPayload = {
cmd: 1005,
sub: 0,
stt: 1,
body: ProtoLoginReq.encode(protoLoginReqMessage).finish()
};
const protoBuffFrameRequestMessage = ProtoBuffFrameRequest.create(
protoBuffFrameRequestPayload
);接下来，我们需要将消息对象编码为二进制数据，并发送到服务端：
const buffer = ProtoBuffFrameRequest.encode(protoBuffFrameRequestMessage).finish();
ws.on('open', () => {
ws.send(new Uint8Array(buffer));
});在WebSocket连接打开后，我们使用ws.send方法发送经过编码的数据。
最后，我们监听服务端返回的消息，并进行解码和处理：
ws.on('message', data => {
const decodeData = ProtoBuffFrameResponse.decode(data);
console.log('接收到数据------------', JSON.stringify(decodeData));
// 在这里进行解码后的数据处理
});在这个例子中，我们使用ProtoBuffFrameResponse.decode方法对服务端返回的数据进行解码，并根据解码后的结果进行相应的处理。
通过这个示例，你可以学习如何使用Node.js与WebSocket服务端进行通信，并发送经过Protobuf编码的数据。
完整代码如下：
const WebSocket = require('ws');
const protobuf = require('protobufjs');
const ws = new WebSocket('ws://example.com/ws');
protobuf.load('protos/pb_login.proto', function (err, root) {
if (err) throw err;
const frameRoot = protobuf.loadSync('protos/pb_http_frame.proto');
const ProtoLoginReq = root.lookupType('Req');
const ProtoBuffFrameRequest = frameRoot.lookupType('Request');
const ProtoBuffFrameResponse = frameRoot.lookupType('Response');
const protoLoginReqPayload = {
uid: 112000087,
client: '4.3.02',
platform: 'ios',
token: '1F6BB53A6504904145CB38EEB57532DBB9FF5868F5A0FD61'
};
const protoLoginReqMessage = ProtoLoginReq.create(protoLoginReqPayload);
const protoBuffFrameRequestPayload = {
cmd: 1005,
sub: 0,
stt: 1,
body: ProtoLoginReq.encode(protoLoginReqMessage).finish()
};
const protoBuffFrameRequestMessage = ProtoBuffFrameRequest.create(
protoBuffFrameRequestPayload
);
const buffer = ProtoBuffFrameRequest.encode(protoBuffFrameRequestMessage).finish();
ws.on('open', () => {
ws.send(new Uint8Array(buffer));
});
ws.on('message', data => {
const decodeData = ProtoBuffFrameResponse.decode(data);
console.log('接收到数据------------', JSON.stringify(decodeData));
// 在这里进行解码后的数据处理
});
});