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
}

 

 四 服务端代码

 服务端链接成功后，等待接收。 若是接收到登陆请求，则解析loginReq，而后返回一个loginReq。

var ws = require("nodejs-websocket");
var protobufjs = require("protobufjs");
var root = protobufjs.loadSync("./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完毕");

　　

五 客户端代码

客户端请求链接服务端，链接成功后发送登陆请求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来拼接数据。