package com.xiaojukeji.kafka.manager.task.dispatch.metrics.collect;

import com.xiaojukeji.kafka.manager.common.constant.KafkaMetricsCollections;
import com.xiaojukeji.kafka.manager.common.entity.metrics.TopicMetrics;
import com.xiaojukeji.kafka.manager.common.entity.pojo.ClusterDO;
import com.xiaojukeji.kafka.manager.common.events.TopicMetricsCollectedEvent;
import com.xiaojukeji.kafka.manager.common.utils.SpringTool;
import com.xiaojukeji.kafka.manager.common.utils.ValidateUtils;
import com.xiaojukeji.kafka.manager.service.cache.KafkaMetricsCache;
import com.xiaojukeji.kafka.manager.service.service.ClusterService;
import com.xiaojukeji.kafka.manager.service.service.JmxService;
import com.xiaojukeji.kafka.manager.task.component.AbstractScheduledTask;
import com.xiaojukeji.kafka.manager.task.component.CustomScheduled;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.*;

/**
 * Topic社区指标收集
 * @author zengqiao
 * @date 20/7/21
 */
@CustomScheduled(name = "collectAndPublishCommunityTopicMetrics", cron = "31 0/1 * * * ?", threadNum = 5)
public class CollectAndPublishCommunityTopicMetrics extends AbstractScheduledTask<ClusterDO> {
    @Autowired
    private JmxService jmxService;

    @Autowired
    private ClusterService clusterService;

    @Override
    protected List<ClusterDO> listAllTasks() {
        // 获取需要进行指标采集的集群列表，这些集群将会被拆分到多台KM中进行执行。
        return clusterService.list();
    }

    @Override
    public void processTask(ClusterDO clusterDO) {
        // 这里需要实现对clusterDO这个集群进行Topic指标采集的代码逻辑

        // 进行Topic指标获取
        List<TopicMetrics> metricsList = getTopicMetrics(clusterDO.getId());

        // 获取到Topic流量指标之后，发布一个事件，
        SpringTool.publish(new TopicMetricsCollectedEvent(this, clusterDO.getId(), metricsList));
    }

    private List<TopicMetrics> getTopicMetrics(Long clusterId) {
        // 具体获取Topic流量指标的入口代码
        List<TopicMetrics> metricsList =
                jmxService.getTopicMetrics(clusterId, KafkaMetricsCollections.TOPIC_METRICS_TO_DB, true);
        if (ValidateUtils.isEmptyList(metricsList)) {
            KafkaMetricsCache.putTopicMetricsToCache(clusterId, new ArrayList<>());
            return new ArrayList<>();
        }
        KafkaMetricsCache.putTopicMetricsToCache(clusterId, metricsList);
        return metricsList;
    }
}