sonarqube/server/sonar-webserver-common/src/main/java/org/sonar/server/app/ProcessCommandWrapperImpl.java

/*
 * SonarQube
 * Copyright (C) 2009-2019 SonarSource SA
 * mailto:info AT sonarsource DOT com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonar.server.app;

import java.io.File;
import org.sonar.api.config.Configuration;
import org.sonar.process.ProcessId;
import org.sonar.process.sharedmemoryfile.DefaultProcessCommands;
import org.sonar.process.sharedmemoryfile.ProcessCommands;

import static org.sonar.process.ProcessEntryPoint.PROPERTY_PROCESS_INDEX;
import static org.sonar.process.ProcessEntryPoint.PROPERTY_SHARED_PATH;

public class ProcessCommandWrapperImpl implements ProcessCommandWrapper {

  private static final ProcessMethod<Void> SET_OPERATIONAL = processCommands -> {
    processCommands.setOperational();
    return null;
  };
  private static final ProcessMethod<Void> ASK_FOR_RESTART = processCommands -> {
    processCommands.askForRestart();
    return null;
  };
  private static final ProcessMethod<Void> ASK_FOR_HARD_STOP = processCommands -> {
    processCommands.askForHardStop();
    return null;
  };
  private static final ProcessMethod<Boolean> IS_OPERATIONAL = ProcessCommands::isOperational;

  private final Configuration config;

  public ProcessCommandWrapperImpl(Configuration config) {
    this.config = config;
  }

  @Override
  public void requestSQRestart() {
    call(ASK_FOR_RESTART, selfProcessNumber());
  }

  @Override
  public void requestHardStop() {
    call(ASK_FOR_HARD_STOP, selfProcessNumber());
  }

  @Override
  public void notifyOperational() {
    call(SET_OPERATIONAL, selfProcessNumber());
  }

  @Override
  public boolean isCeOperational() {
    return call(IS_OPERATIONAL, ProcessId.COMPUTE_ENGINE.getIpcIndex());
  }

  private int selfProcessNumber() {
    return nonNullAsInt(PROPERTY_PROCESS_INDEX);
  }

  private <T> T call(ProcessMethod<T> command, int processNumber) {
    File shareDir = nonNullValueAsFile(PROPERTY_SHARED_PATH);
    try (DefaultProcessCommands commands = DefaultProcessCommands.secondary(shareDir, processNumber)) {
      return command.callOn(commands);
    }
  }

  private interface ProcessMethod<T> {
    T callOn(ProcessCommands processCommands);
  }

  private int nonNullAsInt(String key) {
    return config.getInt(key).orElseThrow(() -> new IllegalArgumentException(String.format("Property %s is not set", key)));
  }

  private File nonNullValueAsFile(String key) {
    return new File(config.get(key).orElseThrow(() -> new IllegalArgumentException(String.format("Property %s is not set", key))));
  }
}