com.taboola.async_profiler.utils.NetUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of async-profiler-actuator-endpoint Show documentation

Async Profiler Actuator Endpoint

There is a newer version: 3.0.0

Show newest version

package com.taboola.async_profiler.utils;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.Map;

public class NetUtils {

    public HttpURLConnection getHTTPConnection(String string,
                                               String path,
                                               Map queryParams,
                                               String requestMethod,
                                               int connectTimeout,
                                               int readTimeout) throws IOException {
        URL requestUrl = buildUrl(string, path, queryParams);

        HttpURLConnection connection = (HttpURLConnection) requestUrl.openConnection();//cached connection
        connection.setDoOutput(true);
        connection.setRequestMethod(requestMethod);
        connection.setRequestProperty("Connection", "keep-alive");
        connection.setConnectTimeout(connectTimeout);
        connection.setReadTimeout(readTimeout);

        return connection;
    }

    public URL buildUrl(String string, String path, Map queryParams) {
        try {
            URI baseUri = URI.create(string);
            StringBuilder queryStringBuilder = new StringBuilder();
            boolean first = true;
            for (Map.Entry entry : queryParams.entrySet()) {
                if (!first) {
                    queryStringBuilder.append("&");
                }

                queryStringBuilder.append(urlEncodedKeyValue(entry.getKey(), entry.getValue()));

                first = false;
            }

            return new URI(baseUri.getScheme(),
                    baseUri.getUserInfo(),
                    baseUri.getHost(),
                    baseUri.getPort(),
                    baseUri.getPath() + path,
                    queryStringBuilder.toString(),
                    null).toURL();

        } catch (MalformedURLException | URISyntaxException e) {
            throw new RuntimeException(e);
        }
    }

    public String urlEncoded(String s) {
        try {
            return URLEncoder.encode(s, StandardCharsets.UTF_8.name());
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException(e);
        }
    }

    private String urlEncodedKeyValue(String key, String value) {
        return urlEncoded(key) + "=" + urlEncoded(value);
    }
}