com.taotao.boot.monitor.collect.task.HttpPoolCollectTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of taotao-boot-starter-monitor Show documentation
taotao-boot-starter-monitor
The newest version!
/*
 * Copyright (c) 2020-2030, Shuigedeng ([email protected] & https://blog.taotaocloud.top/).
 *
 * 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
 *
 *      https://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 com.taotao.boot.monitor.collect.task;

import com.taotao.boot.common.utils.context.ContextUtils;
import com.taotao.boot.common.utils.log.LogUtils;
import com.taotao.boot.common.utils.reflect.ReflectionUtils;
import com.taotao.boot.core.http.HttpClient;
import com.taotao.boot.core.http.HttpClientManager;
import com.taotao.boot.monitor.annotation.FieldReport;
import com.taotao.boot.monitor.collect.AbstractCollectTask;
import com.taotao.boot.monitor.collect.CollectInfo;
import com.taotao.boot.monitor.properties.CollectTaskProperties;
import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManager;
import org.apache.hc.core5.pool.PoolStats;

import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;

/**
 * HTTP连接池性能采集
 *
 * @author shuigedeng
 * @version 2021.9
 * @since 2021-09-10 17:44:20
 */
public class HttpPoolCollectTask extends AbstractCollectTask {

    private static final String TASK_NAME = "ttc.monitor.collect.httpPool";

    private final CollectTaskProperties properties;

    public HttpPoolCollectTask(CollectTaskProperties properties) {
        this.properties = properties;
    }

    @Override
    public int getTimeSpan() {
        return properties.getHttpPoolTimeSpan();
    }

    @Override
    public boolean getEnabled() {
        return properties.isHttpPoolEnabled();
    }

    @Override
    public String getDesc() {
        return this.getClass().getName();
    }

    @Override
    public String getName() {
        return TASK_NAME;
    }

    @Override
    protected CollectInfo getData() {
        try {
            HttpClientManager httpClientManager = ContextUtils.getBean(HttpClientManager.class, true);
            if (Objects.isNull(httpClientManager)) {
                return null;
            }

            ConcurrentHashMap pool = httpClientManager.getPool();
            if (pool == null || pool.isEmpty()) {
                return null;
            }

            HttpPoolInfo info = new HttpPoolInfo();
            StringBuilder detail = new StringBuilder();
            pool.forEach((id, client) -> {
                PoolingHttpClientConnectionManager manager = ReflectionUtils.getFieldValue(client, "manager");
                PoolStats stats = manager.getTotalStats();
                info.availableCount += stats.getAvailable();
                info.pendingCount += stats.getPending();
                info.leasedCount += stats.getLeased();

                detail.append(String.format("[Client连接池:%s]\r\n", id));
                detail.append(String.format("路由数:%s\r\n", manager.getRoutes()));
                detail.append(String.format("路由连接数:%s\r\n", manager.getDefaultMaxPerRoute()));
                detail.append(String.format("最大的连接数:%s\r\n", manager.getMaxTotal()));
                detail.append(String.format("可用的连接数:%s\r\n", stats.getAvailable()));
                detail.append(String.format("等待的连接数:%s\r\n", stats.getPending()));
                detail.append(String.format("使用中的连接数:%s\r\n", stats.getLeased()));
            });
            info.poolDetail = detail.toString();
            return info;
        } catch (Exception e) {
            if (LogUtils.isErrorEnabled()) {
                LogUtils.error(e);
            }
        }
        return null;
    }

    private static class HttpPoolInfo implements CollectInfo {

        @FieldReport(name = TASK_NAME + ".available", desc = "HttpPool可用的连接数")
        private Integer availableCount = 0;

        @FieldReport(name = TASK_NAME + ".pending", desc = "HttpPool等待的连接数")
        private Integer pendingCount = 0;

        @FieldReport(name = TASK_NAME + ".leased", desc = "HttpPool使用中的连接数")
        private Integer leasedCount = 0;

        @FieldReport(name = TASK_NAME + ".detail", desc = "HttpPool详情")
        private String poolDetail = "";
    }
}