Examples with SubscriptionGroupWrapper org.apache.rocketmq.common.protocol.body.SubscriptionGroupWrapper used on opensource projects

Example 26 with SubscriptionGroupWrapper

use of org.apache.rocketmq.common.protocol.body.SubscriptionGroupWrapper in project rocketmq-comments by HScarb.

the class ExportMetricsCommand method execute.

@Override
public void execute(CommandLine commandLine, Options options, RPCHook rpcHook) throws SubCommandException {
    DefaultMQAdminExt defaultMQAdminExt = new DefaultMQAdminExt(rpcHook);
    defaultMQAdminExt.setInstanceName(Long.toString(System.currentTimeMillis()));
    try {
        String clusterName = commandLine.getOptionValue('c').trim();
        String filePath = !commandLine.hasOption('f') ? "/tmp/rocketmq/export" : commandLine.getOptionValue('f').trim();
        defaultMQAdminExt.start();
        Map<String, Map<String, Map<String, Object>>> evaluateReportMap = new HashMap<>();
        Map<String, Double> totalTpsMap = new HashMap<>();
        Map<String, Long> totalOneDayNumMap = new HashMap<>();
        initTotalMap(totalTpsMap, totalOneDayNumMap);
        ClusterInfo clusterInfoSerializeWrapper = defaultMQAdminExt.examineBrokerClusterInfo();
        Set<String> brokerNameSet = clusterInfoSerializeWrapper.getClusterAddrTable().get(clusterName);
        for (String brokerName : brokerNameSet) {
            BrokerData brokerData = clusterInfoSerializeWrapper.getBrokerAddrTable().get(brokerName);
            if (brokerData != null) {
                String addr = brokerData.getBrokerAddrs().get(0L);
                KVTable kvTable = defaultMQAdminExt.fetchBrokerRuntimeStats(addr);
                Properties properties = defaultMQAdminExt.getBrokerConfig(addr);
                SubscriptionGroupWrapper subscriptionGroupWrapper = defaultMQAdminExt.getUserSubscriptionGroup(addr, 10000);
                Map<String, Map<String, Object>> brokerInfo = new HashMap<>();
                // broker environment,machine configuration
                brokerInfo.put("runtimeEnv", getRuntimeEnv(kvTable, properties));
                brokerInfo.put("runtimeQuota", getRuntimeQuota(kvTable, defaultMQAdminExt, addr, totalTpsMap, totalOneDayNumMap, subscriptionGroupWrapper));
                // runtime version
                brokerInfo.put("runtimeVersion", getRuntimeVersion(defaultMQAdminExt, subscriptionGroupWrapper));
                evaluateReportMap.put(brokerName, brokerInfo);
            }
        }
        String path = filePath + "/metrics.json";
        Map<String, Object> totalData = new HashMap<>();
        totalData.put("totalTps", totalTpsMap);
        totalData.put("totalOneDayNum", totalOneDayNumMap);
        Map<String, Object> result = new HashMap<>();
        result.put("evaluateReport", evaluateReportMap);
        result.put("totalData", totalData);
        MixAll.string2FileNotSafe(JSON.toJSONString(result, true), path);
        System.out.printf("export %s success", path);
    } catch (Exception e) {
        throw new SubCommandException(this.getClass().getSimpleName() + " command failed", e);
    } finally {
        defaultMQAdminExt.shutdown();
    }
}