Skip to content

Commit

Permalink
ingest feature flags from .env file
Browse files Browse the repository at this point in the history 
Signed-off-by: kohinoor98 <[email protected]>
  • Loading branch information
@kohinoor98
kohinoor98 committed Oct 7, 2023
1 parent e617da3 commit 8f78f46
Showing 1 changed file with 58 additions and 17 deletions.
75 changes: 58 additions & 17 deletions buildSrc/src/main/java/org/opensearch/gradle/testclusters/RunTask.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,16 @@

import java.io.BufferedReader;
import java.io.Closeable;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.function.BooleanSupplier;
import java.util.function.Function;
import java.util.stream.Collectors;
Expand Down Expand Up @@ -77,15 +80,17 @@ public class RunTask extends DefaultTestClustersTask {
@Option(option = "debug-jvm", description = "Run OpenSearch as a debug client, where it will try to connect to a debugging server at startup.")
public void setDebug(boolean enabled) {
if (debugServer != null && debugServer == true) {
throw new IllegalStateException("Either --debug-jvm or --debug-server-jvm option should be specified (but not both)");
throw new IllegalStateException(
"Either --debug-jvm or --debug-server-jvm option should be specified (but not both)");
}
this.debug = enabled;
}

@Option(option = "debug-server-jvm", description = "Run OpenSearch as a debug server that will accept connections from a debugging client.")
public void setDebugServer(boolean enabled) {
if (debug != null && debug == true) {
throw new IllegalStateException("Either --debug-jvm or --debug-server-jvm option should be specified (but not both)");
throw new IllegalStateException(
"Either --debug-jvm or --debug-server-jvm option should be specified (but not both)");
}
this.debugServer = enabled;
}
Expand Down Expand Up @@ -135,21 +140,56 @@ public String getDataDir() {
return dataDir.toString();
}

private File locateEnvFile() {
return new File(getProject().getProjectDir(), ".env");
}

private Properties readPropertiesFromFile(File file) {
Properties properties = new Properties();
try (FileInputStream fis = new FileInputStream(file)) {
properties.load(fis);
} catch (IOException e) {
throw new GradleException("Failed to load properties from file: " + file.getAbsolutePath(), e);
}
return properties;
}

private String normalizeKeyForGradle(String key) {
String normalized = key.toLowerCase();
normalized = normalized.replace('_', '.');
return normalized;
}

private void setPropertiesAsSystemProperties(Properties properties) {
for (String key : properties.stringPropertyNames()) {
String normalizedKey = normalizeKeyForGradle(key);
String value = properties.getProperty(key);
System.setProperty(normalizedKey, value);
}
}

private void loadEnvProperties() {
File envFile = locateEnvFile();
if (envFile != null && envFile.exists()) {
Properties envProps = readPropertiesFromFile(envFile);
setPropertiesAsSystemProperties(envProps);
}
}

@Override
public void beforeStart() {
int debugPort = DEFAULT_DEBUG_PORT;
int httpPort = DEFAULT_HTTP_PORT;
int transportPort = DEFAULT_TRANSPORT_PORT;
loadEnvProperties();
Map<String, String> additionalSettings = System.getProperties()
.entrySet()
.stream()
.filter(entry -> entry.getKey().toString().startsWith(CUSTOM_SETTINGS_PREFIX))
.collect(
Collectors.toMap(
entry -> entry.getKey().toString().substring(CUSTOM_SETTINGS_PREFIX.length()),
entry -> entry.getValue().toString()
)
);
.entrySet()
.stream()
.filter(entry -> entry.getKey().toString().startsWith(CUSTOM_SETTINGS_PREFIX))
.collect(
Collectors.toMap(
entry -> entry.getKey().toString().substring(CUSTOM_SETTINGS_PREFIX.length()),
entry -> entry.getValue().toString()));
boolean singleNode = getClusters().stream().flatMap(c -> c.getNodes().stream()).count() == 1;
final Function<OpenSearchNode, Path> getDataPath;
if (singleNode) {
Expand Down Expand Up @@ -181,14 +221,14 @@ public void beforeStart() {
}
if (debug) {
logger.lifecycle(
"Running opensearch in debug mode (client), {} expecting running debug server on port {}",
node,
debugPort
);
"Running opensearch in debug mode (client), {} expecting running debug server on port {}",
node,
debugPort);
node.jvmArgs("-agentlib:jdwp=transport=dt_socket,server=n,suspend=y,address=" + debugPort);
debugPort += 1;
} else if (debugServer) {
logger.lifecycle("Running opensearch in debug mode (server), {} running server with debug port {}", node, debugPort);
logger.lifecycle("Running opensearch in debug mode (server), {} running server with debug port {}",
node, debugPort);
node.jvmArgs("-agentlib:jdwp=transport=dt_socket,server=y,suspend=n,address=" + debugPort);
debugPort += 1;
}
Expand Down Expand Up @@ -228,7 +268,8 @@ public void runAndWait() throws IOException {
if (readData == false) {
// no data was ready to be consumed and rather than continuously spinning, pause
// for some time to avoid excessive CPU usage. Ideally we would use the JDK
// WatchService to receive change notifications but the WatchService does not have
// WatchService to receive change notifications but the WatchService does not
// have
// a native MacOS implementation and instead relies upon polling with possible
// delays up to 10s before a notification is received. See JDK-7133447.
try {
Expand Down

0 comments on commit 8f78f46

Please sign in to comment.