"use strict";
const fs = require('fs');
const path = require('path');
const protobuf = require('pomelo-protobuf');
const Constants = require('../util/constants');
const crypto = require('crypto');
const logger = require('@sex-pomelo/sex-pomelo-logger').getLogger('pomelo', __filename);
module.exports = function(app, opts) {
return new ComponentProtobuf(app, opts);
};
/**
*
* @class
* @implements {Component}
*
* @param {Object} app current application context
* @param {Object} opts construct parameters
*/
let ComponentProtobuf = function(app, opts) {
this.app = app;
opts = opts || {};
this.watchers = {};
this.serverProtos = {};
this.clientProtos = {};
this.version = "";
let env = app.get(Constants.RESERVED.ENV);
let originServerPath = path.join(app.getBase(), Constants.FILEPATH.SERVER_PROTOS);
let presentServerPath = path.join(Constants.FILEPATH.CONFIG_DIR, env, path.basename(Constants.FILEPATH.SERVER_PROTOS));
let originClientPath = path.join(app.getBase(), Constants.FILEPATH.CLIENT_PROTOS);
let presentClientPath = path.join(Constants.FILEPATH.CONFIG_DIR, env, path.basename(Constants.FILEPATH.CLIENT_PROTOS));
this.serverProtosPath = opts.serverProtos || (fs.existsSync(originServerPath) ? Constants.FILEPATH.SERVER_PROTOS : presentServerPath);
this.clientProtosPath = opts.clientProtos || (fs.existsSync(originClientPath) ? Constants.FILEPATH.CLIENT_PROTOS : presentClientPath);
this.setProtos(Constants.RESERVED.SERVER, path.join(app.getBase(), this.serverProtosPath));
this.setProtos(Constants.RESERVED.CLIENT, path.join(app.getBase(), this.clientProtosPath));
protobuf.init({encoderProtos:this.serverProtos, decoderProtos:this.clientProtos});
};
ComponentProtobuf.prototype.name = '__protobuf__';
ComponentProtobuf.prototype.encode = function(key, msg) {
return protobuf.encode(key, msg);
};
ComponentProtobuf.prototype.encode2Bytes = function(key, msg) {
return protobuf.encode2Bytes(key, msg);
};
ComponentProtobuf.prototype.decode = function(key, msg) {
return protobuf.decode(key, msg);
};
ComponentProtobuf.prototype.getProtos = function() {
return {
server : this.serverProtos,
client : this.clientProtos,
version : this.version
};
};
ComponentProtobuf.prototype.getVersion = function() {
return this.version;
};
ComponentProtobuf.prototype.setProtos = function(type, path) {
if(!fs.existsSync(path)) {
return;
}
if(type === Constants.RESERVED.SERVER) {
this.serverProtos = protobuf.parse(require(path));
}
if(type === Constants.RESERVED.CLIENT) {
this.clientProtos = protobuf.parse(require(path));
}
let protoStr = JSON.stringify(this.clientProtos) + JSON.stringify(this.serverProtos);
this.version = crypto.createHash('md5').update(protoStr).digest('base64');
//Watch file
let watcher = fs.watch(path, this.onUpdate.bind(this, type, path));
if (this.watchers[type]) {
this.watchers[type].close();
}
this.watchers[type] = watcher;
};
ComponentProtobuf.prototype.onUpdate = function(type, path, event) {
if(event !== 'change') {
return;
}
let self = this;
fs.readFile(path, 'utf8' ,function(err, data) {
try {
let protos = protobuf.parse(JSON.parse(data));
if(type === Constants.RESERVED.SERVER) {
protobuf.setEncoderProtos(protos);
self.serverProtos = protos;
} else {
protobuf.setDecoderProtos(protos);
self.clientProtos = protos;
}
let protoStr = JSON.stringify(self.clientProtos) + JSON.stringify(self.serverProtos);
self.version = crypto.createHash('md5').update(protoStr).digest('base64');
logger.info('change proto file , type : %j, path : %j, version : %j', type, path, self.version);
} catch(e) {
logger.warn("change proto file error! path : %j", path);
logger.warn(e);
}
});
};
ComponentProtobuf.prototype.stop = function(force, cb) {
for (let type in this.watchers) {
this.watchers[type].close();
}
this.watchers = {};
process.nextTick(cb);
};