ambari/ambari-web/app/controllers/installer/step6_controller.js

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

var App = require('app');
var db = require('utils/db');

/**
 * By Step 6, we have the following information stored in App.db and set on this
 * controller by the router:
 *
 *   hosts: App.db.hosts (list of all hosts the user selected in Step 3)
 *   selectedServiceNames: App.db.selectedServiceNames (the services that the user selected in Step 4)
 *   masterComponentHosts: App.db.masterComponentHosts (master-components-to-hosts mapping the user selected in Step 5)
 *
 * Step 6 will set the following information in App.db:
 *   hostSlaveComponents: App.db.hostSlaveComponents (hosts-to-slave-components mapping the user selected in Steo 6)
 *   slaveComponentHosts: App.db.slaveComponentHosts (slave-components-to-hosts mapping the user selected in Step 6)
 *
 */
App.InstallerStep6Controller = Em.Controller.extend({

  hosts: [],
  // TODO: hook up with user host selection
  rawHosts: [],
  selectedServiceNames: null,
  masterComponentHosts: require('data/mock/master_component_hosts'),
  showHbase: false,

  hasMasterComponents: function (hostname) {
    var hasMaster = false;
    var masterComponentHosts = db.getMasterComponentHosts();
    return masterComponentHosts.someProperty('hostName', hostname);
  },


  isAllDataNodes: function () {
    return this.get('hosts').everyProperty('isDataNode', true);
  }.property('hosts.@each.isDataNode'),

  isAllTaskTrackers: function () {
    return this.get('hosts').everyProperty('isTaskTracker', true);
  }.property('hosts.@each.isTaskTracker'),

  isAllRegionServers: function () {
    return this.get('hosts').everyProperty('isRegionServer', true);
  }.property('hosts.@each.isRegionServer'),

  isNoDataNodes: function () {
    return this.get('hosts').everyProperty('isDataNode', false);
  }.property('hosts.@each.isDataNode'),

  isNoTaskTrackers: function () {
    return this.get('hosts').everyProperty('isTaskTracker', false);
  }.property('hosts.@each.isTaskTracker'),

  isNoRegionServers: function () {
    return this.get('hosts').everyProperty('isRegionServer', false);
  }.property('hosts.@each.isRegionServer'),

  isHbaseSelected: function () {
    var services = db.getSelectedServiceNames();
    console.log('isHbase selected is: ' + services.contains('HBASE'));
    return services.contains('HBASE');
  },


  selectAllDataNodes: function () {
    this.get('hosts').setEach('isDataNode', true);
  },

  selectAllTaskTrackers: function () {
    this.get('hosts').setEach('isTaskTracker', true);
  },

  selectAllRegionServers: function () {
    this.get('hosts').setEach('isRegionServer', true);
  },

  deselectAllDataNodes: function () {
    this.get('hosts').setEach('isDataNode', false);
  },

  deselectAllTaskTrackers: function () {
    this.get('hosts').setEach('isTaskTracker', false);
  },

  deselectAllRegionServers: function () {
    this.get('hosts').setEach('isRegionServer', false);
  },


  clearStep: function () {
    this.set('hosts', []);
  },

  loadStep: function () {
    console.log("TRACE: Loading step6: Assign Slaves");
    this.clearStep();
    this.set('showHbase', this.isHbaseSelected());
    this.setSlaveHost(this.getSlaveHosts());
  },

  navigateStep: function () {
    this.loadStep();
  },

  getHostNames: function () {
    var hostInfo = db.getHosts();
    var hostNames = [];
    for (var index in hostInfo) {
      if (hostInfo[index].bootStatus === 'success')
        hostNames.push(hostInfo[index].name);
    }
    return hostNames;
  },

  getSlaveHosts: function () {
    var hostObjs = new Ember.Set();
    var allHosts = this.getHostNames();
    var slaveHosts = App.db.getSlaveComponentHosts();
    if (slaveHosts === undefined || slaveHosts === null) {
      allHosts.forEach(function (_hostname) {
        var hostObj = {};
        hostObj.hostname = _hostname;
        hostObj.isDataNode = !this.hasMasterComponents(_hostname);
        hostObj.isTaskTracker = !this.hasMasterComponents(_hostname);
        hostObj.isRegionServer = !this.hasMasterComponents(_hostname);
        hostObjs.add(hostObj);
      }, this);
      return hostObjs;
    } else {
      allHosts.forEach(function (_hostName) {
        hostObjs.add({
          hostname: _hostName
        });
      });
      var datanodes = slaveHosts.findProperty('componentName', 'DataNode');
      datanodes.hosts.forEach(function (_datanode) {
        var datanode = hostObjs.findProperty('hostname', _datanode.hostname);
        if (datanode !== null) {
          datanode.isDataNode = true;
        }
      });
      var taskTrackers = slaveHosts.findProperty('componentName', 'TaskTracker');
      taskTrackers.hosts.forEach(function (_taskTracker) {
        var taskTracker = hostObjs.findProperty('hostname', _taskTracker.hostname);
        if (taskTracker !== null) {
          taskTracker.isTaskTracker = true;
        }
      });
      if (this.isHbaseSelected()) {
        var regionServers = slaveHosts.findProperty('componentName', 'RegionServer');
        regionServers.hosts.forEach(function (_regionServer) {
          var regionServer = hostObjs.findProperty('hostname', _regionServer.hostname);
          if (regionServer !== null) {
            regionServer.isRegionServer = true;
          }
        });
      }
      return hostObjs;
    }
  },

  setSlaveHost: function (hostObj) {
    hostObj.forEach(function (_hostObj) {
      this.get('hosts').pushObject(Ember.Object.create(_hostObj));
    }, this);
  },

  loadSlaveHost: function (hostNames) {

    hostNames.forEach(function (_hostName) {
      this.get('hosts').pushObject(Ember.Object.create({
        hostname: _hostName,
        isDataNode: !this.hasMasterComponents(_hostName),
        isTaskTracker: !this.hasMasterComponents(_hostName),
        isRegionServer: !this.hasMasterComponents(_hostName)
      }));
    }, this);
  },


  validate: function () {
    return !(this.get('isNoDataNodes') || this.get('isNoTaskTrackers') || this.get('isNoRegionServers'));
  },

  submit: function () {
    if (!this.validate()) {
      this.set('errorMessage', Ember.I18n.t('installer.step6.error.mustSelectOne'));
      return;
    }
    App.db.setHostSlaveComponents(this.get('host'));

    var dataNodeHosts = [];
    var taskTrackerHosts = [];
    var regionServerHosts = [];

    this.get('hosts').forEach(function (host) {
      if (host.get('isDataNode')) {
        dataNodeHosts.push({
          hostname: host.hostname,
          group: 'Default'
        });
      }
      if (host.get('isTaskTracker')) {
        taskTrackerHosts.push({
          hostname: host.hostname,
          group: 'Default'
        });
      }
      if (this.isHbaseSelected() && host.get('isRegionServer')) {
        regionServerHosts.push({
          hostname: host.hostname,
          group: 'Default'
        });
      }
    }, this);

    var slaveComponentHosts = [];
    slaveComponentHosts.push({
      componentName: 'DataNode',
      hosts: dataNodeHosts
    });
    slaveComponentHosts.push({
      componentName: 'TaskTracker',
      hosts: taskTrackerHosts
    });
    if (this.isHbaseSelected()) {
      slaveComponentHosts.push({
        componentName: 'RegionServer',
        hosts: regionServerHosts
      });
    }

    App.db.setSlaveComponentHosts(slaveComponentHosts);

    App.router.send('next');

  }
})
;