com.linecorp.centraldogma.server.Main Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of centraldogma-server Show documentation
Show all versions of centraldogma-server Show documentation
Highly-available version-controlled service configuration repository based on Git, ZooKeeper and HTTP/2 (centraldogma-server)
/*
* Copyright 2017 LINE Corporation
*
* LINE Corporation 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:
*
* 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.linecorp.centraldogma.server;
import static com.google.common.base.MoreObjects.firstNonNull;
import java.io.File;
import java.io.IOException;
import java.nio.file.AtomicMoveNotSupportedException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import javax.annotation.Nullable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.converters.FileConverter;
import com.linecorp.armeria.common.util.SystemInfo;
/**
* Entry point of a standalone server. Use {@link CentralDogmaBuilder} to embed a server.
*/
public final class Main {
private static final Logger logger = LoggerFactory.getLogger(Main.class);
static {
try {
// Install the java.util.logging to SLF4J bridge.
final Class bridgeHandler =
Class.forName("org.slf4j.bridge.SLF4JBridgeHandler", true, Main.class.getClassLoader());
bridgeHandler.getMethod("removeHandlersForRootLogger").invoke(null);
bridgeHandler.getMethod("install").invoke(null);
logger.debug("Installed the java.util.logging-to-SLF4J bridge.");
} catch (Throwable cause) {
logger.debug("Failed to install the java.util.logging-to-SLF4J bridge:", cause);
}
}
enum State {
NONE,
INITIALIZED,
STARTED,
STOPPED,
DESTROYED
}
private static final File DEFAULT_DATA_DIR =
new File(System.getProperty("user.dir", ".") + File.separatorChar + "data");
private static final File DEFAULT_CONFIG_FILE =
new File(System.getProperty("user.dir", ".") +
File.separatorChar + "conf" +
File.separatorChar + "dogma.json");
private static final File DEFAULT_PID_FILE =
new File(System.getProperty("user.dir", ".") +
File.separatorChar + "dogma.pid");
@Nullable
@Parameter(names = "-config", description = "The path to the config file", converter = FileConverter.class)
private File configFile;
@Nullable
@Parameter(names = "-pidfile",
description = "The path to the file containing the pid of the server" +
" (defaults to ./dogma.pid)",
converter = FileConverter.class)
private File pidFile;
/**
* Note that {@link Boolean} was used in lieu of {@code boolean} so that JCommander does not print the
* default value of this option.
*/
@Nullable
@Parameter(names = { "-help", "-h" }, description = "Prints the usage", help = true)
private Boolean help;
private State state = State.NONE;
@Nullable
private CentralDogma dogma;
@Nullable
private PidFile procIdFile;
private boolean procIdFileCreated;
private Main(String[] args) {
final JCommander commander = new JCommander(this);
commander.setProgramName(getClass().getName());
commander.parse(args);
if (help != null && help) {
commander.usage();
} else {
procIdFile = new PidFile(firstNonNull(pidFile, DEFAULT_PID_FILE));
state = State.INITIALIZED;
}
}
synchronized void start() throws Exception {
switch (state) {
case NONE:
throw new IllegalStateException("not initialized");
case STARTED:
throw new IllegalStateException("started already");
case DESTROYED:
throw new IllegalStateException("can't start after destruction");
default:
break;
}
final File configFile = findConfigFile(this.configFile, DEFAULT_CONFIG_FILE);
final CentralDogma dogma;
if (configFile == null) {
dogma = new CentralDogmaBuilder(DEFAULT_DATA_DIR).build();
} else {
dogma = CentralDogma.forConfig(configFile);
}
dogma.start().get();
this.dogma = dogma;
state = State.STARTED;
// The server would be stopped even if we fail to create the PID file from here,
// because the state has been updated.
assert procIdFile != null;
procIdFile.create();
procIdFileCreated = true;
}
@Nullable
private static File findConfigFile(@Nullable File file, File defaultFile) {
if (file != null) {
if (file.isFile() && file.canRead()) {
return file;
} else {
throw new IllegalStateException("cannot access the specified config file: " + file);
}
}
// Try to use the default config file if not specified.
if (defaultFile.isFile() && defaultFile.canRead()) {
return defaultFile;
}
return null;
}
synchronized void stop() throws Exception {
switch (state) {
case NONE:
case INITIALIZED:
case STOPPED:
return;
case DESTROYED:
throw new IllegalStateException("can't stop after destruction");
}
final CentralDogma dogma = this.dogma;
assert dogma != null;
this.dogma = null;
dogma.stop().get();
state = State.STOPPED;
}
void destroy() {
switch (state) {
case NONE:
return;
case STARTED:
throw new IllegalStateException("can't destroy while running");
case DESTROYED:
return;
}
assert procIdFile != null;
if (procIdFileCreated) {
try {
procIdFile.destroy();
} catch (IOException e) {
logger.warn("Failed to destroy the PID file:", e);
}
}
state = State.DESTROYED;
}
/**
* Starts a new Central Dogma server.
*/
public static void main(String[] args) throws Exception {
final Main main = new Main(args);
// Register the shutdown hook.
Runtime.getRuntime().addShutdownHook(new Thread("Central Dogma shutdown hook") {
@Override
public void run() {
try {
main.stop();
} catch (Exception e) {
logger.warn("Failed to stop the Central Dogma:", e);
}
try {
main.destroy();
} catch (Exception e) {
logger.warn("Failed to destroy the Central Dogma:", e);
}
}
});
// Exit if initialization failed.
if (main.state != State.INITIALIZED) {
System.exit(1);
return;
}
try {
main.start();
} catch (Throwable cause) {
logger.error("Failed to start the Central Dogma:", cause);
// Trigger the shutdown hook.
System.exit(1);
}
}
/**
* Manages a process ID file for the Central Dogma server.
*/
static final class PidFile {
private final File file;
private PidFile(File file) {
this.file = file;
}
void create() throws IOException {
if (file.exists()) {
throw new IllegalStateException("Failed to create a PID file. A file already exists: " +
file.getPath());
}
final int pid = SystemInfo.pid();
final Path temp = Files.createTempFile("central-dogma", ".tmp");
Files.write(temp, Integer.toString(pid).getBytes());
try {
Files.move(temp, file.toPath(), StandardCopyOption.ATOMIC_MOVE);
} catch (AtomicMoveNotSupportedException e) {
Files.move(temp, file.toPath());
}
logger.debug("A PID file has been created: {}", file);
}
void destroy() throws IOException {
if (Files.deleteIfExists(file.toPath())) {
logger.debug("Successfully deleted the PID file: {}", file);
}
}
}
}