com.tencent.polaris.client.util.OkHttpUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of polaris-all Show documentation
All in one project for polaris-java
The newest version!
/*
 * Tencent is pleased to support the open source community by making polaris-java available.
 *
 * Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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 com.tencent.polaris.client.util;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import com.tencent.polaris.api.utils.CollectionUtils;
import com.tencent.polaris.api.utils.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * okhttp util.
 *
 * @author kan peng
 */
public final class OkHttpUtil {

	/**
	 * Logger.
	 */
	public final static Logger LOGGER = LoggerFactory.getLogger(OkHttpUtil.class);

	private OkHttpUtil() {

	}

	/**
	 * get request.
	 * @param path path
	 * @param headers headers
	 * @return response
	 */
	public static boolean get(String path, Map headers) {
		HttpURLConnection conn = null;
		try {
			URL url = new URL(path);
			conn = (HttpURLConnection) url.openConnection();

			conn.setRequestMethod("GET");
			conn.setConnectTimeout((int) TimeUnit.SECONDS.toMillis(2));
			conn.setReadTimeout((int) TimeUnit.SECONDS.toMillis(2));
			if (!CollectionUtils.isEmpty(headers)) {
				headers.forEach(conn::setRequestProperty);
			}
			BufferedReader reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));
			StringBuffer buffer = new StringBuffer();
			String str;
			while ((str = reader.readLine()) != null) {
				buffer.append(str);
			}
			String responseBody = buffer.toString();
			if (conn.getResponseCode() == 200 && StringUtils.isNotBlank(responseBody)) {
				LOGGER.debug("exec get request, url: {} success, response data: {}", url, responseBody);
				return true;
			}
		}
		catch (Exception e) {
			LOGGER.error("exec get request, url: {} failed!", path, e);
			return false;
		}
		finally {
			if (null != conn) {
				conn.disconnect();
			}
		}
		return false;
	}

	public static boolean checkUrl(String host, Integer port, String endpoint, Map headers) {
		if (!endpoint.startsWith("/")) {
			endpoint = "/" + endpoint;
		}
		String checkUrl = String.format("http://%s:%s%s", host, port, endpoint);
		return get(checkUrl, headers);
	}
}