VILLASweb/app/mixins/live-data.js

/**
 * File: live-data.js
 * Author: Markus Grigull <mgrigull@eonerc.rwth-aachen.de>
 * Date: 06.10.2016
 * Copyright: 2016, Institute for Automation of Complex Power Systems, EONERC
 *   This file is part of VILLASweb. All Rights Reserved. Proprietary and confidential.
 *   Unauthorized copying of this file, via any medium is strictly prohibited.
 **********************************************************************************/

import Ember from 'ember';

const { service } = Ember.inject;

export default Ember.Mixin.create({
  store: service(),
  sessionUser: service('session-user'),

  INTERVAL: 5000,

  _sockets: {},

  init: function() {
    this._super();

    // fetch the simulations for the first time
    this._fetchRunningSimulations();

    // start the polling loop
    setInterval((function(self) {
      return function() {
        self._fetchRunningSimulations();
      };
    })(this), this.INTERVAL);
  },

  _fetchRunningSimulations: function() {
    // check if the user is logged in
    if (this.get('sessionUser.user') != null) {
      // get all simulations to find all running ones
      var self = this;

      this.get('store').findAll('simulation').then(function(simulations) {
        simulations.forEach(function(simulation) {
          // check if the simulation is running
          if (simulation.get('running')) {
            // get all models for this simulation
            simulation.get('models').then((models) => {
              models.forEach(function(simulationModel) {
                self._addSocket(simulationModel);
              });
            });
          }
        });
      });
    }
  },

  _addSocket(simulationModel) {
    // check if socket is already open
    simulationModel.get('simulator').then((simulator) => {
      let id = simulator.get('simulatorid');
      if (this.get('_sockets')[id] !== undefined) {
        //Ember.debug('skip ' + simulationModel.get('name'));
        return;
      }

      // get simulator endpoint
      simulationModel.get('simulator').then((simulator) => {
        // get simulator endpoint
        let endpoint = simulator.get('endpoint');
        if (endpoint) {
          // add new socket
          let socket = new WebSocket('ws://' + endpoint, 'live');
          socket.binaryType = 'arraybuffer';

          // register callbacks
          let self = this;

          socket.onopen = function(event) { self._onSocketOpen.apply(self, [event]); };
          socket.onclose = function(event) { self._onSocketClose.apply(self, [event]); };
          socket.onmessage = function(event) { self._onSocketMessage.apply(self, [event]); };
          socket.onerror = function(event) { self._onSocketError.apply(self, [event]); };

          // add socket to list of known sockets
          this.get('_sockets')[id] = socket;

          //Ember.debug('Socket created for ' + simulationModel.get('name') + ': ws://' + endpoint);
        } else {
          Ember.debug('Undefined endpoint for ' + simulationModel.get('name'));
        }
      });
    });
  },

  _removeSocket(socket) {
    // search through all sockets
    let sockets = this.get('_sockets');

    for (let id in sockets) {
      if (sockets[id] === socket) {
        // remove socket from list
        delete sockets[id];
      }
    }
  },

  _onSocketOpen(/* event */) {
    //Ember.debug('websocket opened');
  },

  _onSocketClose(event) {
    if (event.wasClean) {
      Ember.debug('websocket closed');
    } else {
      Ember.debug('websocket closed: ' + event.code);
    }

    // remove socket from array
    this._removeSocket(event.target);
  },

  _onSocketMessage(event) {
    // read the message into JSON
    var message = this._messageToJSON(event.data);

    // set simulator by socket
    if (message.simulator === 0) {
      // search for socket in list
      let sockets = this.get('_sockets');

      for (let id in sockets) {
        if (sockets[id] === event.target) {
          // set id as simulator
          message.simulator = id;
          break;
        }
      }
    }

    // create or update simulation data object
    var simulationData = this.store.peekRecord('simulation-data', message.simulator);
    if (simulationData != null) {
      simulationData.set('sequence', message.sequence);
      simulationData.set('values', message.values);
    } else {
      this.store.createRecord('simulation-data', {
        sequence: message.sequence,
        values: message.values,
        id: message.simulator
      });
    }
  },

  _onSocketError(/* event */) {
    Ember.debug('websocket error');
  },

  _messageToJSON(blob) {
    // parse incoming message into usable data
    var data = new DataView(blob);

    let OFFSET_ENDIAN = 1;
    let OFFSET_TYPE = 2;
    let OFFSET_VERSION = 4;

    var bits = data.getUint8(0);
    var simulator = data.getUint8(0x01);
    var endian = (bits >> OFFSET_ENDIAN) & 0x1 ? 0 : 1;
    var length = data.getUint16(0x02, endian);

    var values = new Float32Array(data.buffer, data.byteOffset + 0x10, length);

    return {
      endian: endian,
      version: (bits >> OFFSET_VERSION) & 0xF,
      type: (bits >> OFFSET_TYPE) & 0x3,
      length: length,
      sequence: data.getUint32(0x04, endian),
      timestamp: data.getUint32(0x08, endian) * 1e3 + data.getUint32(0x0C, endian) * 1e-6,
      values: values,
      simulator: simulator
    };
  }
});