com.gemstone.gemfire.internal.tools.gfsh.app.commands.connect Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gemfire-core Show documentation
SnappyData store based off Pivotal GemFireXD
The newest version!
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * 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. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.internal.tools.gfsh.app.commands;

import java.util.ArrayList;

import com.gemstone.gemfire.internal.tools.gfsh.app.CommandExecutable;
import com.gemstone.gemfire.internal.tools.gfsh.app.Gfsh;

public class connect implements CommandExecutable
{
	private Gfsh gfsh;
	
	public connect(Gfsh gfsh)
	{
		this.gfsh = gfsh;
	}
	
	public void help()
	{
//		gfsh.println("connect [-s  | -l  []] [-r ] | [-?]");
		gfsh.println("connect [-s  | -l  []] | [-?]");
		gfsh.println("     -s   Connect to the specified cache servers.");
		gfsh.println("     -l  [] Connect to the specified locator");
		gfsh.println("           and the server group if specified.");
		//TODO: with integrated gfsh, we should not allow command regions other than __command
//		gfsh.println("     -r   Use the specified region in making connection.");
//		gfsh.println("            The default region path is '/__command'.");
		gfsh.println("     -t   readTimeout in msec.");
		gfsh.println("            The default value is 300000 ms (5 min).");
		gfsh.println();
	}
	
	public void execute(String command) throws Exception
	{
		if (command.startsWith("connect -?")) {
			help();
		} else {
			connect(command);
      if (gfsh.isConnected()) {
        gfsh.execute("refresh");
      }
		}
	}
	
	private void connect(String command)
	{
		ArrayList list = new ArrayList();
		gfsh.parseCommand(command, list);
		if (list.size() < 3) {
			gfsh.println("Error: incomplete command.");
			return;
		} 
		String endpoints = null;
		String locators = null;
//		String regionPath = null;
		String serverGroup = null;
		int readTimeout = 300000;
		try {
			for (int i = 1; i < list.size(); i++) {
				String token = list.get(i);
				if (token.equals("-s")) {
					i++;
					endpoints = list.get(i);
				} else if (token.equals("-l")) {
					i++;
					locators = list.get(i);
					if (i < list.size() - 1) {
						if (list.get(i+1).startsWith("-") == false) {
							serverGroup = list.get(++i);
						}
					}
				} /*else if (token.equals("-r")) {
					i++;
					regionPath = list.get(i);
				}*/ else if (token.equals("-t")) {
					i++;
					readTimeout = Integer.parseInt(list.get(i));
				}
			}
		} catch (Exception ex) {
			gfsh.println("Error: invalid command - " + command);
			return;
		}
		
		if (endpoints != null && locators != null) {
			gfsh.println("Error: invalid command. -s and -l are not allowed together.");
		}
		
//		if (regionPath != null) {
//			gfsh.setCommandRegionPath(regionPath);
//		}
		if (endpoints != null) {
			gfsh.setEndpoints(endpoints, false, null, readTimeout);
			connect();
		}
		if (locators != null) {
			gfsh.setEndpoints(locators, true, serverGroup, readTimeout);
			connect();
		}
	}
	
	private void connect()
	{
		try {
			gfsh.reconnect();
			gfsh.setCurrentPath("/");
			gfsh.setCurrentRegion(null);
			if (gfsh.isConnected()) {
				gfsh.println("connected: " + gfsh.getEndpoints());
			} else {
				gfsh.println("Error: Endpoints set but unable to connect: " + gfsh.getEndpoints());
			}
			
		} catch (Exception ex) {
			gfsh.println(gfsh.getCauseMessage(ex));
      if (gfsh.isDebug()) {
        ex.printStackTrace();
      }
		}

	}
}