sonarqube/sonar-ws/src/testFixtures/java/org/sonarqube/ws/tester/Elasticsearch.java

/*
 * SonarQube
 * Copyright (C) 2009-2023 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.sonarqube.ws.tester;

import java.io.IOException;
import java.net.InetAddress;
import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;

import static org.assertj.core.api.Assertions.assertThat;

/**
 * Helper to directly access Elasticsearch. It requires the HTTP port
 * to be open.
 */
public class Elasticsearch {

  private final int httpPort;

  public Elasticsearch(int httpPort) {
    this.httpPort = httpPort;
  }

  /**
   * Forbid indexing requests on the specified index. Index becomes read-only.
   */
  public void lockWrites(String index) throws IOException {
    putIndexSetting(httpPort, index, "blocks.write", "true");
  }

  /**
   * Enable indexing requests on the specified index.
   * @see #lockWrites(String)
   */
  public void unlockWrites(String index) throws IOException {
    putIndexSetting(httpPort, index, "blocks.write", "false");
  }

  public void makeYellow() throws IOException {
    putIndexSetting(httpPort, "issues", "number_of_replicas", "5");
  }

  public void makeGreen() throws IOException {
    putIndexSetting(httpPort, "issues", "number_of_replicas", "0");
  }

  private static void putIndexSetting(int searchHttpPort, String index, String key, String value) throws IOException {
    Request.Builder request = new Request.Builder()
      .url(baseUrl(searchHttpPort) + index + "/_settings")
      .put(RequestBody.create(MediaType.parse("application/json"), "{" +
        "    \"index\" : {" +
        "        \"" + key + "\" : \"" + value + "\"" +
        "    }" +
        "}"));
    OkHttpClient okClient = new OkHttpClient.Builder().build();
    try (Response response = okClient.newCall(request.build()).execute()) {
      assertThat(response.isSuccessful()).isTrue();
    }
  }

  private static String baseUrl(int searchHttpPort) {
    return "http://" + InetAddress.getLoopbackAddress().getHostAddress() + ":" + searchHttpPort + "/";
  }
}