com.github.nosan.embedded.cassandra.AbstractCassandraDatabase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of embedded-cassandra Show documentation
Embedded Cassandra
There is a newer version: 5.0.0
/*
 * Copyright 2020-2021 the original author or authors.
 *
 * 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
 *
 *      https://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.github.nosan.embedded.cassandra;

import java.io.IOException;
import java.nio.file.Path;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;

abstract class AbstractCassandraDatabase implements CassandraDatabase {

	private final String name;

	private final Version version;

	private final Path configurationFile;

	private final Path workingDirectory;

	private final Map environmentVariables;

	private final Map configProperties;

	private final Map systemProperties;

	private final Set jvmOptions;

	private volatile Process process;

	AbstractCassandraDatabase(String name, Version version, Path configurationFile, Path workingDirectory,
			Map environmentVariables, Map configProperties,
			Map systemProperties, Set jvmOptions) {
		this.name = name;
		this.version = version;
		this.configurationFile = configurationFile;
		this.workingDirectory = workingDirectory;
		this.environmentVariables = Collections.unmodifiableMap(new LinkedHashMap<>(environmentVariables));
		this.configProperties = Collections.unmodifiableMap(new LinkedHashMap<>(configProperties));
		this.systemProperties = Collections.unmodifiableMap(new LinkedHashMap<>(systemProperties));
		this.jvmOptions = Collections.unmodifiableSet(new LinkedHashSet<>(jvmOptions));
	}

	@Override
	public final synchronized void start() throws IOException {
		this.process = doStart();
	}

	@Override
	public final synchronized void stop() throws IOException {
		Process process = this.process;
		if (process != null && process.isAlive()) {
			doStop(process);
			if (!process.destroy().waitFor(5, TimeUnit.SECONDS)) {
				if (!process.destroyForcibly().waitFor(3, TimeUnit.SECONDS)) {
					throw new IOException("Unable to stop " + this);
				}
			}
		}
	}

	@Override
	public final String getName() {
		return this.name;
	}

	@Override
	public final Map getEnvironmentVariables() {
		return this.environmentVariables;
	}

	@Override
	public final Map getConfigProperties() {
		return this.configProperties;
	}

	@Override
	public final Map getSystemProperties() {
		return this.systemProperties;
	}

	@Override
	public final Set getJvmOptions() {
		return this.jvmOptions;
	}

	@Override
	public final synchronized CompletableFuture onExit() {
		return this.process.onExit()
				.thenApply((Function) processHandler -> this);
	}

	@Override
	public final synchronized boolean isAlive() {
		Process process = this.process;
		return (process != null) && process.isAlive();
	}

	@Override
	public final Version getVersion() {
		return this.version;
	}

	@Override
	public final Path getWorkingDirectory() {
		return this.workingDirectory;
	}

	@Override
	public final Path getConfigurationFile() {
		return this.configurationFile;
	}

	@Override
	public final synchronized Process.Output getStdOut() {
		return this.process.getStdOut();
	}

	@Override
	public final synchronized Process.Output getStdErr() {
		return this.process.getStdErr();
	}

	@Override
	public final String toString() {
		return getClass().getSimpleName() + "{" + "process=" + this.process + '}';
	}

	protected abstract Process doStart() throws IOException;

	protected abstract void doStop(Process process) throws IOException;

}