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: [],
	masterComponentHosts: require('data/mock/master_component_hosts'),
	showHbase: false,
	showTaskTracker: 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();
		return this.get('selectedServiceNames').contains('HBASE');
	},

	isMrSelected: function () {
		return this.get('selectedServiceNames').contains('MAPREDUCE');
	}.property('selectedServiceNames'),

	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', []);
		this.set('selectedServiceNames', []);
	},

	loadStep: function () {
		console.log("TRACE: Loading step6: Assign Slaves");
		this.clearStep();
		this.set('selectedServiceNames',db.getSelectedServiceNames());
		this.set('showHbase', this.isHbaseSelected());
		this.set('showTaskTracker', this.get('isMrSelected'));
		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;
				}
			});
			if (this.get('isMrSelected')) {
				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', 'HBASE_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 (this.get('isMrSelected') && host.get('isRegionServer')) {
				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',
			displayName: 'DataNode',
			hosts: dataNodeHosts
		});
		if (this.get('isMrSelected')) {
			slaveComponentHosts.push({
				componentName: 'TASKTRACKER',
				displayName: 'TaskTracker',
				hosts: taskTrackerHosts
			});
		}
		if (this.isHbaseSelected()) {
			slaveComponentHosts.push({
				componentName: 'HBASE_REGIONSERVER',
				displayName: 'RegionServer',
				hosts: regionServerHosts
			});
		}

		App.db.setSlaveComponentHosts(slaveComponentHosts);

		App.router.send('next');

	}
})
;