// // AgentPoolController.js // acScripts // // Created by Sam Gateau on 11/23/15. // Copyright 2015 High Fidelity, Inc. // // Distributed under the Apache License, Version 2.0. // See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html // function printDebug(message) { print(message); } (function() { var COMMAND_CHANNEL = "com.highfidelity.PlaybackChannel1"; var ANNOUNCE_CHANNEL = "com.highfidelity.playbackAgent.announceID"; // The time between alive messages on the command channel var ALIVE_PERIOD = 3; var NUM_CYCLES_BEFORE_RESET = 8; // Service Actions var MASTER_ID = -1; var AGENT_READY = "ready"; var AGENT_LOST = "agentLost"; var INVALID_ACTOR = -2; var AGENTS_BROADCAST = -1; var MASTER_ALIVE = -1; var MASTER_FIRE_AGENT = -2; var makeUniqueUUID = function(SEUUID) { //return SEUUID + Math.random(); // forget complexity, just give me a four digit pin return (Math.random() * 10000).toFixed(0); } var packAnnounceMessage = function(dest, command, src) { var message = { dest: dest, command: command, src: src }; return JSON.stringify(message); }; var unpackAnnounceMessage = function(message) { return JSON.parse(message); }; var packCommandMessage = function(id, action, argument) { var message = { id_key: id, action_key: action, argument_key: argument }; return JSON.stringify(message); }; var unpackCommandMessage = function(message) { return JSON.parse(message); }; // Actor //--------------------------------- var Actor = function() { this.agentID = INVALID_ACTOR; this.onHired = function(actor) {}; this.onLost = function(actor) {}; }; this.Actor = Actor; // master side //--------------------------------- var MasterController = function() { this.timeSinceLastAlive = 0; this.knownAgents = new Array; this.hiredActors = new Array; this.hiringAgentsQueue = new Array; this.subscribed = false; }; MasterController.prototype.destroy = function() { if (this.subscribed) { Messages.unsubscribe(ANNOUNCE_CHANNEL); Messages.unsubscribe(COMMAND_CHANNEL); this.subscribed = true; } }; MasterController.prototype.reset = function() { this.timeSinceLastAlive = 0; if (!this.subscribed) { Messages.subscribe(COMMAND_CHANNEL); Messages.subscribe(ANNOUNCE_CHANNEL); var localThis = this; Messages.messageReceived.connect(function (channel, message, senderID) { if (channel == ANNOUNCE_CHANNEL) { localThis._processAnnounceMessage(message, senderID); return; } }); } // ready to roll, enable this.subscribed = true; printDebug("Master Started"); }; MasterController.prototype._processAnnounceMessage = function(message, senderID) { var message = unpackAnnounceMessage(message); if (message.dest == MASTER_ID) { if (message.command == AGENT_READY) { // check to see if we know about this agent var agentIndex = this.knownAgents.indexOf(message.src); if (agentIndex < 0) { if (this.hiringAgentsQueue.length > 0) { var hiringHandler = this.hiringAgentsQueue.pop(); hiringHandler(message.src); } else { //No hiring in queue so bail return; } } else { // Master think the agent is hired but not the other way around, forget about it printDebug("New agent still sending ready ? " + message.src + " " + agentIndex + " Forgeting about it"); this.knownAgents[agentIndex] = INVALID_ACTOR; } } else if (message.command == AGENT_LOST) { // check to see if we know about this agent var agentIndex = this.knownAgents.indexOf(message.src); if (agentIndex < 0) { printDebug("Lost agent " + message.src + " " + agentIndex + " Forgeting about it"); this.knownAgents[agentIndex] = INVALID_ACTOR; var lostActor = this.hiredActors[agentIndex]; this.hiredActors[agentIndex] = null; lostActor.agentID = INVALID_ACTOR; lostActor.onLost(lostActor); } } } }; MasterController.prototype.sendCommand = function(target, action, argument) { if (this.subscribed) { var messageJSON = packCommandMessage(target, action, argument); Messages.sendMessage(COMMAND_CHANNEL, messageJSON); } }; MasterController.prototype.update = function(deltaTime) { this.timeSinceLastAlive += deltaTime; if (this.timeSinceLastAlive > ALIVE_PERIOD) { this.timeSinceLastAlive = 0; this.sendCommand(AGENTS_BROADCAST, MASTER_ALIVE); } }; MasterController.prototype.hireAgent = function(actor) { if (actor == null) { printDebug("trying to hire an agent with a null actor, abort"); return; } var localThis = this; this.hiringAgentsQueue.unshift(function(agentID) { printDebug("hiring callback with agent " + agentID+ " " + JSON.stringify(localThis) ); var indexOfNewAgent = localThis.knownAgents.push(agentID) actor.agentID = agentID; localThis.hiredActors.push(actor); printDebug("New agent available to be hired " + agentID + " " + indexOfNewAgent); var hireMessage = "HIRE." + indexOfNewAgent; var answerMessage = packAnnounceMessage(agentID, hireMessage, MASTER_ID); Messages.sendMessage(ANNOUNCE_CHANNEL, answerMessage); printDebug("message sent calling the actor" + JSON.stringify(actor) ); actor.onHired(actor); }) }; this.MasterController = MasterController; // agent side //--------------------------------- var AgentController = function() { this.subscribed = false; this._init(); this.onHired = function() {}; this.onCommand = function(command) {}; this.onFired = function() {}; }; AgentController.prototype._init = function() { this.actorIndex= INVALID_ACTOR; this.notifyAlive = false; this.timeSinceLastAlive = 0; this.numCyclesWithoutAlive = 0; this.actorUUID = makeUniqueUUID(Agent.sessionUUID); printDebug("this.actorUUID = " + this.actorUUID); } AgentController.prototype.destroy = function() { if (this.subscribed) { this.fire(); Messages.unsubscribe(ANNOUNCE_CHANNEL); Messages.unsubscribe(COMMAND_CHANNEL); this.subscribed = true; } }; AgentController.prototype.reset = function() { // If already hired, fire this.fired(); if (!this.subscribed) { Messages.subscribe(COMMAND_CHANNEL); Messages.subscribe(ANNOUNCE_CHANNEL); var localThis = this; Messages.messageReceived.connect(function (channel, message, senderID) { if (channel == ANNOUNCE_CHANNEL) { localThis._processAnnounceMessage(message, senderID); return; } if (channel == COMMAND_CHANNEL) { localThis._processCommandMessage(message, senderID); return; } }); } this.subscribed = true; printDebug("Client Started"); }; AgentController.prototype._processAnnounceMessage = function(message, senderID) { var announce = unpackCommandMessage(message); //printDebug("Client " + this.actorIndex + " Received Announcement = " + message); if (announce.dest == this.actorUUID) { if (announce.command != AGENT_READY) { // this may be a message to hire me if i m not already if (this.actorIndex == INVALID_ACTOR) { printDebug(announce.command); var parts = announce.command.split("."); var agentID = parts[0]; var actorIndex = parts[1]; //printDebug("Client " + Agent.sessionUUID + " - " + agentID + " Hired!"); // if (agentID == Agent.sessionUUID) { this.actorIndex = actorIndex; printDebug("Client " + this.actorIndex + " Hired!"); this.onHired(); // Messages.unsubscribe(ANNOUNCE_CHANNEL); // id announce channel // } } } } } AgentController.prototype._processCommandMessage = function(message, senderID) { var command = unpackCommandMessage(message); //printDebug("Received command = " + JSON.stringify(command)); if ((command.id_key == this.actorUUID) || (command.id_key == AGENTS_BROADCAST)) { if (command.action_key == MASTER_ALIVE) { this.notifyAlive = true; } else if (command.action_key == MASTER_FIRE_AGENT) { printDebug("Master firing Agent"); this.fire(); } else { printDebug("True action received = " + JSON.stringify(command) + senderID); this.onCommand(command); } } else { // ignored } }; AgentController.prototype.update = function(deltaTime) { this.timeSinceLastAlive += deltaTime; if (this.timeSinceLastAlive > ALIVE_PERIOD) { if (this.subscribed) { if (this.actorIndex == INVALID_ACTOR) { Messages.sendMessage(ANNOUNCE_CHANNEL, packAnnounceMessage(MASTER_ID, AGENT_READY, this.actorUUID)); //printDebug("Client Ready" + ANNOUNCE_CHANNEL + AGENT_READY); } else { this.numCyclesWithoutAlive++; if (this.notifyAlive) { this.notifyAlive = false; this.numCyclesWithoutAlive = 0; printDebug("Master Alive"); } else if (this.numCyclesWithoutAlive > NUM_CYCLES_BEFORE_RESET) { printDebug("Master Lost, firing Agent"); this.fired(); } } } this.timeSinceLastAlive = 0; } }; AgentController.prototype.fired = function() { // clear the state first var wasHired = (this.actorIndex != INVALID_ACTOR); // Post a last message to master in case it still listen to warn that this agent is losing it if (wasHired) { Messages.sendMessage(ANNOUNCE_CHANNEL, packAnnounceMessage(MASTER_ID, AGENT_LOST, this.actorUUID)); } // reset this._init(); // then custom fire if was hired if (wasHired) { this.onFired(); } } this.AgentController = AgentController; })();