com.mysql.jdbc.util.ServerController Maven / Gradle / Ivy
/*
Copyright (c) 2002, 2014, Oracle and/or its affiliates. All rights reserved.
The MySQL Connector/J is licensed under the terms of the GPLv2
, like most MySQL Connectors.
There are special exceptions to the terms and conditions of the GPLv2 as it is applied to
this software, see the FOSS License Exception
.
This program is free software; you can redistribute it and/or modify it under the terms
of the GNU General Public License as published by the Free Software Foundation; version 2
of the License.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with this
program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth
Floor, Boston, MA 02110-1301 USA
*/
package com.mysql.jdbc.util;
import java.io.File;
import java.io.IOException;
import java.util.Iterator;
import java.util.Properties;
import com.mysql.jdbc.StringUtils;
/**
* Controls a MySQL server using Java RunTime methods
*/
public class ServerController {
/**
* Where is the server installed?
*/
public static final String BASEDIR_KEY = "basedir";
/**
* Where are the databases installed?
*/
public static final String DATADIR_KEY = "datadir";
/**
* Where is the config file located?
*/
public static final String DEFAULTS_FILE_KEY = "defaults-file";
/**
* What is the name of the executable to run?
*/
public static final String EXECUTABLE_NAME_KEY = "executable";
/**
* What is the path to the mysql server executable (if not standard?)
*/
public static final String EXECUTABLE_PATH_KEY = "executablePath";
/**
* The default executable to run
*/
/**
* The process representing the MySQL server
*/
private Process serverProcess = null;
/**
* The list of properties for this server
*/
private Properties serverProps = null;
/**
* The system properties
*/
private Properties systemProps = null;
/**
* Creates a ServerController with the directory for the MySQL server.
*
* The 'datadir' is set to the same directory.
*
* @param baseDir
* the base directory for the MySQL server.
*/
public ServerController(String baseDir) {
setBaseDir(baseDir);
}
/**
* Creates a server controller for the MySQL server with the given basedir
* and datadir.
*
* @param basedir
* the basedir to use when starting MySQL.
* @param datadir
* the datadir to use when starting MySQL.
*/
public ServerController(String basedir, String datadir) {
}
/**
* Sets the basedir to use when starting MySQL.
*
* @param baseDir
* the basedir to use when starting MySQL.
*/
public void setBaseDir(String baseDir) {
getServerProps().setProperty(BASEDIR_KEY, baseDir);
}
/**
* Sets the data to use when starting MySQL.
*
* @param dataDir
* the basedir to use when starting MySQL.
*/
public void setDataDir(String dataDir) {
getServerProps().setProperty(DATADIR_KEY, dataDir);
}
/**
* Starts the server, returning a java.lang.Process instance that represents
* the mysql server.
*
* @return Process a java.lang.Process instance representing the mysql
* server process.
* @throws IOException
* if an error occurs while starting the mysql server.
*/
public Process start() throws IOException {
if (this.serverProcess != null) {
throw new IllegalArgumentException("Server already started");
}
this.serverProcess = Runtime.getRuntime().exec(getCommandLine());
return this.serverProcess;
}
/**
* Stops the server (if started)
*
* @param forceIfNecessary
* use forceStop if mysqladmin doesn't shut the server down
*
* @throws IOException
* if an error occurs while stopping the server
*/
public void stop(boolean forceIfNecessary) throws IOException {
if (this.serverProcess != null) {
String basedir = getServerProps().getProperty(BASEDIR_KEY);
StringBuilder pathBuf = new StringBuilder(basedir);
if (!basedir.endsWith(File.separator)) {
pathBuf.append(File.separator);
}
//String defaultsFilePath = getServerProps().getProperty(DEFAULTS_FILE_KEY);
pathBuf.append("bin");
pathBuf.append(File.separator);
pathBuf.append("mysqladmin shutdown");
System.out.println(pathBuf.toString());
Process mysqladmin = Runtime.getRuntime().exec(pathBuf.toString());
int exitStatus = -1;
try {
exitStatus = mysqladmin.waitFor();
} catch (InterruptedException ie) {
// ignore
}
//
// Terminate the process if mysqladmin couldn't do it, and the user requested a force stop.
//
if (exitStatus != 0 && forceIfNecessary) {
forceStop();
}
}
}
/**
* Forcefully terminates the server process (if started).
*/
public void forceStop() {
if (this.serverProcess != null) {
this.serverProcess.destroy();
this.serverProcess = null;
}
}
/**
* Returns the list of properties that will be used to start/control the
* server.
*
* @return Properties the list of properties.
*/
public synchronized Properties getServerProps() {
if (this.serverProps == null) {
this.serverProps = new Properties();
}
return this.serverProps;
}
/**
* Returns the full commandline used to start the mysql server, including
* and arguments to be passed to the server process.
*
* @return String the commandline used to start the mysql server.
*/
private String getCommandLine() {
StringBuilder commandLine = new StringBuilder(getFullExecutablePath());
commandLine.append(buildOptionalCommandLine());
return commandLine.toString();
}
/**
* Returns the fully-qualifed path to the 'mysqld' executable
*
* @return String the path to the server executable.
*/
private String getFullExecutablePath() {
StringBuilder pathBuf = new StringBuilder();
String optionalExecutablePath = getServerProps().getProperty(EXECUTABLE_PATH_KEY);
if (optionalExecutablePath == null) {
// build the path using the defaults
String basedir = getServerProps().getProperty(BASEDIR_KEY);
pathBuf.append(basedir);
if (!basedir.endsWith(File.separator)) {
pathBuf.append(File.separatorChar);
}
if (runningOnWindows()) {
pathBuf.append("bin");
} else {
pathBuf.append("libexec");
}
pathBuf.append(File.separatorChar);
} else {
pathBuf.append(optionalExecutablePath);
if (!optionalExecutablePath.endsWith(File.separator)) {
pathBuf.append(File.separatorChar);
}
}
String executableName = getServerProps().getProperty(EXECUTABLE_NAME_KEY, "mysqld");
pathBuf.append(executableName);
return pathBuf.toString();
}
/**
* Builds the list of command-line arguments that will be passed to the
* mysql server to be started.
*
* @return String the list of command-line arguments.
*/
private String buildOptionalCommandLine() {
StringBuilder commandLineBuf = new StringBuilder();
if (this.serverProps != null) {
for (Iterator