com.marklogic.appdeployer.command.hosts.AssignHostsToGroupsCommand Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2023 MarkLogic Corporation
 *
 * Licensed 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.
 */
package com.marklogic.appdeployer.command.hosts;

import java.util.Map;

import com.marklogic.appdeployer.command.AbstractUndoableCommand;
import com.marklogic.appdeployer.command.CommandContext;
import com.marklogic.appdeployer.command.SortOrderConstants;
import com.marklogic.mgmt.resource.hosts.HostManager;

public class AssignHostsToGroupsCommand extends AbstractUndoableCommand {

	private static final String DEFAULT_GROUP_NAME = "Default";

	public AssignHostsToGroupsCommand() {
        setExecuteSortOrder(SortOrderConstants.ASSIGN_HOSTS_TO_GROUPS);
        setUndoSortOrder(SortOrderConstants.UNASSIGN_HOSTS_FROM_GROUPS);
    }

	@Override
	public void execute(CommandContext context) {
		context.getAdminManager().invokeActionRequiringRestart(() -> assignHostsToGroups(context));
	}

	protected boolean assignHostsToGroups(CommandContext context) {
		boolean requiresRestart = false;

		Map hostGroups = context.getAppConfig().getHostGroups();
		HostManager hostMgr = new HostManager(context.getManageClient());
		if (hostGroups != null) {
			for (Map.Entry entry : hostGroups.entrySet()) {
				String hostName = entry.getKey();
				String groupName = entry.getValue();

				if (!groupName.equals(hostMgr.getAssignedGroupName(hostName))) {
					if (logger.isInfoEnabled()) {
						logger.info(format("Assigning host %s to group %s", hostName, groupName));
					}
					hostMgr.setHostToGroup(hostName, groupName);
					requiresRestart = true;
				}
			}
		}

		return requiresRestart;
	}

	@Override
	public void undo(CommandContext context) {
		context.getAdminManager().invokeActionRequiringRestart(() -> assignHostsToDefault(context));
	}

	protected boolean assignHostsToDefault(CommandContext context) {
		boolean requiresRestart = false;

		Map hostGroups = context.getAppConfig().getHostGroups();
		if (hostGroups != null) {
			HostManager hostMgr = new HostManager(context.getManageClient());
			for (Map.Entry entry : hostGroups.entrySet()) {
				String hostName = entry.getKey();
				if (!DEFAULT_GROUP_NAME.equals(hostMgr.getAssignedGroupName(hostName))) {
					if (logger.isInfoEnabled()) {
						logger.info(format("Assigning host %s to group %s", hostName, DEFAULT_GROUP_NAME));
					}
					hostMgr.setHostToGroup(hostName, DEFAULT_GROUP_NAME);
					requiresRestart = true;
				}
			}
		}

		return requiresRestart;
	}
}