io.micronaut.discovery.cloud.ComputeInstanceMetadataResolverUtils Maven / Gradle / Ivy
/*
* Copyright 2017-2020 original authors
*
* Licensed 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
*
* http://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 io.micronaut.discovery.cloud;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import io.micronaut.core.annotation.Internal;
import io.micronaut.http.HttpMethod;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
/**
* Utility class for {@link ComputeInstanceMetadataResolver}'s.
*
* @author Alvaro Sanchez-Mariscal
* @since 1.1
*/
@Internal
public class ComputeInstanceMetadataResolverUtils {
/**
* Reads the result of a URL and parses it using the given {@link ObjectMapper}.
*
* @param url the URL to read
* @param connectionTimeoutMs connection timeout, in milliseconds
* @param readTimeoutMs read timeout, in milliseconds
* @param objectMapper Jackson's {@link ObjectMapper}
* @param requestProperties any request properties to pass
* @return a {@link JsonNode} instance
* @throws IOException if any I/O error occurs
*/
public static JsonNode readMetadataUrl(URL url, int connectionTimeoutMs, int readTimeoutMs, ObjectMapper objectMapper, Map requestProperties) throws IOException {
URLConnection urlConnection = url.openConnection();
if (url.getProtocol().equalsIgnoreCase("file")) {
urlConnection.connect();
try (InputStream in = urlConnection.getInputStream()) {
return objectMapper.readTree(in);
}
} else {
HttpURLConnection uc = (HttpURLConnection) urlConnection;
uc.setConnectTimeout(connectionTimeoutMs);
requestProperties.forEach(uc::setRequestProperty);
uc.setReadTimeout(readTimeoutMs);
uc.setRequestMethod(HttpMethod.GET.name());
uc.setDoOutput(true);
int responseCode = uc.getResponseCode();
try (InputStream in = uc.getInputStream()) {
return objectMapper.readTree(in);
}
}
}
/**
* Resolve a value as a string from the metadata json.
* @param json The json
* @param key The key
* @return An optional value
*/
public static Optional stringValue(JsonNode json, String key) {
return Optional.ofNullable(json.findValue(key)).map(JsonNode::asText);
}
/**
* Populates the instance instance metadata's {@link AbstractComputeInstanceMetadata#setMetadata(Map)} property.
* @param instanceMetadata The instance metadata
* @param metadata A map of metadata
*/
public static void populateMetadata(AbstractComputeInstanceMetadata instanceMetadata, Map metadata) {
if (metadata != null) {
Map finalMetadata = new HashMap<>(metadata.size());
for (Map.Entry entry : metadata.entrySet()) {
Object key = entry.getKey();
Object value = entry.getValue();
if (value instanceof String) {
finalMetadata.put(key.toString(), value.toString());
}
}
instanceMetadata.setMetadata(finalMetadata);
}
}
}