-
Notifications
You must be signed in to change notification settings - Fork 870
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add host.id resource provider (#10627)
Co-authored-by: Jason Plumb <jplumb@splunk.com>
- Loading branch information
1 parent
c8f2cc5
commit 0029332
Showing
3 changed files
with
317 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
184 changes: 184 additions & 0 deletions
184
...rary/src/main/java/io/opentelemetry/instrumentation/resources/HostIdResourceProvider.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,184 @@ | ||
/* | ||
* Copyright The OpenTelemetry Authors | ||
* SPDX-License-Identifier: Apache-2.0 | ||
*/ | ||
|
||
package io.opentelemetry.instrumentation.resources; | ||
|
||
import static java.util.logging.Level.FINE; | ||
|
||
import io.opentelemetry.api.common.Attributes; | ||
import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties; | ||
import io.opentelemetry.sdk.autoconfigure.spi.ResourceProvider; | ||
import io.opentelemetry.sdk.autoconfigure.spi.internal.ConditionalResourceProvider; | ||
import io.opentelemetry.sdk.resources.Resource; | ||
import io.opentelemetry.semconv.ResourceAttributes; | ||
import java.io.BufferedReader; | ||
import java.io.IOException; | ||
import java.io.InputStreamReader; | ||
import java.nio.charset.StandardCharsets; | ||
import java.nio.file.FileSystems; | ||
import java.nio.file.Files; | ||
import java.nio.file.Path; | ||
import java.util.ArrayList; | ||
import java.util.Collections; | ||
import java.util.List; | ||
import java.util.Locale; | ||
import java.util.function.Function; | ||
import java.util.function.Supplier; | ||
import java.util.logging.Logger; | ||
|
||
/** | ||
* {@link ResourceProvider} for automatically configuring <code>host.id</code> according to <a | ||
* href="https://github.com/open-telemetry/semantic-conventions/blob/main/docs/resource/host.md#non-privileged-machine-id-lookup">the | ||
* semantic conventions</a> | ||
*/ | ||
public final class HostIdResourceProvider implements ConditionalResourceProvider { | ||
|
||
private static final Logger logger = Logger.getLogger(HostIdResourceProvider.class.getName()); | ||
|
||
public static final String REGISTRY_QUERY = | ||
"reg query HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Cryptography /v MachineGuid"; | ||
|
||
private final Supplier<String> getOsType; | ||
|
||
private final Function<Path, List<String>> machineIdReader; | ||
|
||
private final Supplier<List<String>> queryWindowsRegistry; | ||
|
||
public HostIdResourceProvider() { | ||
this( | ||
HostIdResourceProvider::getOsTypeSystemProperty, | ||
HostIdResourceProvider::readMachineIdFile, | ||
HostIdResourceProvider::queryWindowsRegistry); | ||
} | ||
|
||
// Visible for testing | ||
|
||
HostIdResourceProvider( | ||
Supplier<String> getOsType, | ||
Function<Path, List<String>> machineIdReader, | ||
Supplier<List<String>> queryWindowsRegistry) { | ||
this.getOsType = getOsType; | ||
this.machineIdReader = machineIdReader; | ||
this.queryWindowsRegistry = queryWindowsRegistry; | ||
} | ||
|
||
@Override | ||
public Resource createResource(ConfigProperties config) { | ||
if (runningWindows()) { | ||
return readWindowsGuid(); | ||
} | ||
if (runningLinux()) { | ||
return readLinuxMachineId(); | ||
} | ||
logger.log(FINE, "Unsupported OS type: {0}", getOsType.get()); | ||
return Resource.empty(); | ||
} | ||
|
||
private boolean runningLinux() { | ||
return getOsType.get().toLowerCase(Locale.ROOT).equals("linux"); | ||
} | ||
|
||
private boolean runningWindows() { | ||
return getOsType.get().startsWith("Windows"); | ||
} | ||
|
||
// see | ||
// https://github.com/apache/commons-lang/blob/master/src/main/java/org/apache/commons/lang3/SystemUtils.java | ||
// for values | ||
private static String getOsTypeSystemProperty() { | ||
return System.getProperty("os.name", ""); | ||
} | ||
|
||
private Resource readLinuxMachineId() { | ||
Path path = FileSystems.getDefault().getPath("/etc/machine-id"); | ||
List<String> lines = machineIdReader.apply(path); | ||
if (lines.isEmpty()) { | ||
return Resource.empty(); | ||
} | ||
return Resource.create(Attributes.of(ResourceAttributes.HOST_ID, lines.get(0))); | ||
} | ||
|
||
private static List<String> readMachineIdFile(Path path) { | ||
try { | ||
List<String> lines = Files.readAllLines(path); | ||
if (lines.isEmpty()) { | ||
logger.fine("Failed to read /etc/machine-id: empty file"); | ||
} | ||
return lines; | ||
} catch (IOException e) { | ||
logger.log(FINE, "Failed to read /etc/machine-id", e); | ||
return Collections.emptyList(); | ||
} | ||
} | ||
|
||
private Resource readWindowsGuid() { | ||
List<String> lines = queryWindowsRegistry.get(); | ||
|
||
for (String line : lines) { | ||
if (line.contains("MachineGuid")) { | ||
String[] parts = line.trim().split("\\s+"); | ||
if (parts.length == 3) { | ||
return Resource.create(Attributes.of(ResourceAttributes.HOST_ID, parts[2])); | ||
} | ||
} | ||
} | ||
logger.fine("Failed to read Windows registry: No MachineGuid found in output: " + lines); | ||
return Resource.empty(); | ||
} | ||
|
||
private static List<String> queryWindowsRegistry() { | ||
try { | ||
ProcessBuilder processBuilder = new ProcessBuilder("cmd", "/c", REGISTRY_QUERY); | ||
processBuilder.redirectErrorStream(true); | ||
Process process = processBuilder.start(); | ||
|
||
List<String> output = getProcessOutput(process); | ||
int exitedValue = process.waitFor(); | ||
if (exitedValue != 0) { | ||
logger.fine( | ||
"Failed to read Windows registry. Exit code: " | ||
+ exitedValue | ||
+ " Output: " | ||
+ String.join("\n", output)); | ||
|
||
return Collections.emptyList(); | ||
} | ||
|
||
return output; | ||
} catch (IOException | InterruptedException e) { | ||
logger.log(FINE, "Failed to read Windows registry", e); | ||
return Collections.emptyList(); | ||
} | ||
} | ||
|
||
public static List<String> getProcessOutput(Process process) throws IOException { | ||
List<String> result = new ArrayList<>(); | ||
|
||
try (BufferedReader processOutputReader = | ||
new BufferedReader( | ||
new InputStreamReader(process.getInputStream(), StandardCharsets.UTF_8))) { | ||
String readLine; | ||
|
||
while ((readLine = processOutputReader.readLine()) != null) { | ||
result.add(readLine); | ||
} | ||
} | ||
return result; | ||
} | ||
|
||
@Override | ||
public boolean shouldApply(ConfigProperties config, Resource existing) { | ||
return !config | ||
.getMap("otel.resource.attributes") | ||
.containsKey(ResourceAttributes.HOST_ID.getKey()) | ||
&& existing.getAttribute(ResourceAttributes.HOST_ID) == null; | ||
} | ||
|
||
@Override | ||
public int order() { | ||
// Run after cloud provider resource providers | ||
return Integer.MAX_VALUE - 1; | ||
} | ||
} |
125 changes: 125 additions & 0 deletions
125
.../src/test/java/io/opentelemetry/instrumentation/resources/HostIdResourceProviderTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,125 @@ | ||
/* | ||
* Copyright The OpenTelemetry Authors | ||
* SPDX-License-Identifier: Apache-2.0 | ||
*/ | ||
|
||
package io.opentelemetry.instrumentation.resources; | ||
|
||
import static org.assertj.core.api.Assertions.assertThat; | ||
|
||
import io.opentelemetry.api.common.AttributeKey; | ||
import io.opentelemetry.sdk.autoconfigure.spi.internal.DefaultConfigProperties; | ||
import io.opentelemetry.sdk.resources.Resource; | ||
import io.opentelemetry.semconv.ResourceAttributes; | ||
import java.nio.file.Path; | ||
import java.util.Arrays; | ||
import java.util.Collection; | ||
import java.util.Collections; | ||
import java.util.List; | ||
import java.util.function.Function; | ||
import java.util.function.Supplier; | ||
import java.util.stream.Collectors; | ||
import java.util.stream.Stream; | ||
import org.assertj.core.api.MapAssert; | ||
import org.junit.jupiter.api.DynamicTest; | ||
import org.junit.jupiter.api.Test; | ||
import org.junit.jupiter.api.TestFactory; | ||
|
||
class HostIdResourceProviderTest { | ||
|
||
private static class LinuxTestCase { | ||
private final String name; | ||
private final String expectedValue; | ||
private final Function<Path, List<String>> pathReader; | ||
|
||
private LinuxTestCase( | ||
String name, String expectedValue, Function<Path, List<String>> pathReader) { | ||
this.name = name; | ||
this.expectedValue = expectedValue; | ||
this.pathReader = pathReader; | ||
} | ||
} | ||
|
||
private static class WindowsTestCase { | ||
private final String name; | ||
private final String expectedValue; | ||
private final Supplier<List<String>> queryWindowsRegistry; | ||
|
||
private WindowsTestCase( | ||
String name, String expectedValue, Supplier<List<String>> queryWindowsRegistry) { | ||
this.name = name; | ||
this.expectedValue = expectedValue; | ||
this.queryWindowsRegistry = queryWindowsRegistry; | ||
} | ||
} | ||
|
||
@TestFactory | ||
Collection<DynamicTest> createResourceLinux() { | ||
return Stream.of( | ||
new LinuxTestCase("default", "test", path -> Collections.singletonList("test")), | ||
new LinuxTestCase("empty file or error reading", null, path -> Collections.emptyList())) | ||
.map( | ||
testCase -> | ||
DynamicTest.dynamicTest( | ||
testCase.name, | ||
() -> { | ||
HostIdResourceProvider provider = | ||
new HostIdResourceProvider(() -> "linux", testCase.pathReader, null); | ||
|
||
assertHostId(testCase.expectedValue, provider); | ||
})) | ||
.collect(Collectors.toList()); | ||
} | ||
|
||
@TestFactory | ||
Collection<DynamicTest> createResourceWindows() { | ||
return Stream.of( | ||
new WindowsTestCase( | ||
"default", | ||
"test", | ||
() -> | ||
Arrays.asList( | ||
"HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Cryptography", | ||
" MachineGuid REG_SZ test")), | ||
new WindowsTestCase("short output", null, Collections::emptyList)) | ||
.map( | ||
testCase -> | ||
DynamicTest.dynamicTest( | ||
testCase.name, | ||
() -> { | ||
HostIdResourceProvider provider = | ||
new HostIdResourceProvider( | ||
() -> "Windows 95", null, testCase.queryWindowsRegistry); | ||
|
||
assertHostId(testCase.expectedValue, provider); | ||
})) | ||
.collect(Collectors.toList()); | ||
} | ||
|
||
private static void assertHostId(String expectedValue, HostIdResourceProvider provider) { | ||
MapAssert<AttributeKey<?>, Object> that = | ||
assertThat(provider.createResource(null).getAttributes().asMap()); | ||
|
||
if (expectedValue == null) { | ||
that.isEmpty(); | ||
} else { | ||
that.containsEntry(ResourceAttributes.HOST_ID, expectedValue); | ||
} | ||
} | ||
|
||
@Test | ||
void shouldApply() { | ||
HostIdResourceProvider provider = new HostIdResourceProvider(); | ||
assertThat( | ||
provider.shouldApply( | ||
DefaultConfigProperties.createFromMap(Collections.emptyMap()), | ||
Resource.getDefault())) | ||
.isTrue(); | ||
assertThat( | ||
provider.shouldApply( | ||
DefaultConfigProperties.createFromMap( | ||
Collections.singletonMap("otel.resource.attributes", "host.id=foo")), | ||
null)) | ||
.isFalse(); | ||
} | ||
} |