NodeJs使用nodejs-websocket + protobuf

参考：

HTML5+NodeJs实现WebSocket即时通讯 (某人的blog)

nodejs-websocket使用示例 (www.npmjs.com网站，有示例)

Buffer API （nodejs api 中文版）

nodejs-websocket + protobufjs

一 安装nodejs-websocket

在服务端项目文件夹下，新建一个pageage.json

{

  "name": "realtime-server",

  "version": "0.0.1",

  "description": "my first realtime server",

  "dependencies": {

    "nodejs-websocket": "^1.7.2",

    "protobufjs": "^6.8.8"

  }

}

在服务端项目文件夹下shift+右键，在此处打开命令窗口，输入

npm install nodejs-websocket

安装完毕后，项目目录下会增加node_modules文件夹

　　

二 安装protobufjs

在服务端项目文件夹下shift+右键，在此处打开命令窗口，输入

npm install protobufjs

输入后回车，等待安装完毕，如下图：

三 准备测试用.proto文件

准备一个测试用的proto文件，如下

login.proto

//登录

package login;

//登录请求

message LoginReq{

	required int32 uid = 1;      //用户id

}

proto文件转成.js

将login.proto放在服务端项目文件夹下

shift+右键，打开命令窗口，生成js

pbjs -t static-module -w commonjs -o login.js login.proto

生成d.ts (服务端不需要，客户端需要)

pbts -o login.d.ts login.js

四 服务端代码

服务端连接成功后，等待接收。 如果接收到登录请求，则解析loginReq，然后返回一个loginReq。

直接使用login.proto文件示例：

var ws = require("nodejs-websocket");

var protobufjs = require("protobufjs");

var root = protobufjs.loadSync("./login.proto");   //直接使用login.proto文件

console.log("开始创建websocket");

var server = ws.createServer(function(conn){

	console.log("连接成功");

    conn.on("binary", function (inStream) {

		console.log("接收消息");

		var data;

		inStream.on("readable", function () {

            data = inStream.read();

        })

		inStream.on("end", function () {

            console.log("Received " + data.length + " bytes of binary data");

			//解析接收的数据,cmd

			var cmd = data.readUInt16BE(0);

			console.log("接收数据的cmd:",cmd);

			let bytes = Buffer.from(data,1);

			//解析接收的数据,loginReq

			var LoginReq = root.lookupType("login.LoginReq");

			var loginReq = LoginReq.decode(bytes);

			console.log("接收数据的uid:", loginReq.uid);

			//发送的数据,loginReq

			var sendLoginReq = LoginReq.create();

			sendLoginReq.uid = 123;

			var sendData = LoginReq.encode(sendLoginReq).finish();

			//发送的数据,cmd

			var sendBuffer = Buffer.alloc(2);

			sendBuffer.writeInt16BE(100);

			//拼接数据并发送

			var totalBuffer = Buffer.concat([sendBuffer,sendData],sendData.length + sendBuffer.length);

			conn.sendBinary(totalBuffer);

        })

    })

    conn.on("close", function (code, reason) {

        console.log("关闭连接")

    });

    conn.on("error", function (code, reason) {

        console.log("异常关闭")

    });

}).listen(8001)

console.log("开始创建websocket完毕");

　

使用转换后login.js文件示例：

var root = require("./login");    //使用login.js文件

//连接成功

io.on('connection', function(socket){

    console.log('a user connected');

    //监听登录请求

    socket.on('login', function(data,callback){　　　　　　　　　

         //解析登录数据

        var loginReqObj = root.login.LoginReq.decode(data);

        console.log("请求登录的用户:",loginReqObj.uid);

        //返回登录响应

        let loginResObj = root.login.LoginRes.create();

        loginResObj.code = 200;

        var buffer = root.login.LoginReq.encode(loginResObj).finish();

        socket.emit('login', buffer);

    });

});

　　

　

五 客户端代码

客户端请求连接服务端，连接成功后发送登录请求loginReq

	/**连接成功*/

	private onConnect(e:egret.Event){

		console.log("ClientSocket 连接成功");

		this.resetReconnect();

		App.EventMananger.sendEvent(ClientSocket.SOCKET_CONNECT);

		//测试

		let loginData:login.LoginReq = new login.LoginReq();

		loginData.uid = 123;

		let buffer = login.LoginReq.encode(loginData).finish();

		this.send(100, buffer);

	}

	/**

	 * 发送数据

	 * @param cmd 数据协议

	 * @param sendByte 发送的数据

	 */

	public send(cmd:number, sendByte:Uint8Array){

		console.log("ClientSocket 发送:",cmd);

		//发送的数据cmd+proto

		let sendByteArray = new egret.ByteArray(sendByte);

		let byteArray:egret.ByteArray = new egret.ByteArray();

		byteArray.writeUnsignedShort(cmd);

		byteArray.writeBytes(sendByteArray);

		//发送

        this.socket.writeBytes(byteArray);

		this.socket.flush();

	}

	/**接收数据*/

	private onReceive(e:egret.Event){

		//读取socket数据

		var byte:egret.ByteArray = new egret.ByteArray();

        this.socket.readBytes(byte);

		//读取cmd+proto

		let cmd:number = byte.readUnsignedShort();

		console.log("接收数据,cmd:",cmd);

		//读取loginReq

		let revByteArray:egret.ByteArray = new egret.ByteArray();

		byte.readBytes(revByteArray);

		let buffer:login.LoginReq = login.LoginReq.decode(revByteArray.bytes);

		console.log("接收数据,uid:",buffer.uid);

	}

　　

六 测试结果　　

运行服务端和客户端

1. 客户端请求连接服务端

2. 连接成功后，客户端发送登录请求loginReq

3. 服务端接收到登录请求，解析loginReq并打印。 然后再返回一个loginReq。

4. 客户端接收到服务端返回的loginReq，解析并打印。

服务端输出：

客户端输出：

到此，一个简单的nodejs-websocket + protobufjs服务单搭建和联调测试完成。

问题：

一 nodejs服务端关于Buffer的操作

看了一下Buffer的Api，没有找到writeBytes的方法，只有concat来拼接数据。