StatViewServlet refactor

2023-03-04 19:17:16 +00:00 · 2012-08-07 23:12:10 +08:00
parent 82370048b3
commit 6326a8f7c1
4 changed files with 364 additions and 291 deletions
@@ -0,0 +1,139 @@
+package com.alibaba.druid.support;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+
+import com.alibaba.druid.pool.DruidDataSource;
+import com.alibaba.druid.stat.DruidStatManagerFacade;
+import com.alibaba.druid.stat.JdbcSqlStat;
+import com.alibaba.druid.util.MapComparator;
+import com.alibaba.druid.util.StringUtils;
+import com.alibaba.fastjson.JSON;
+
+/**
+ * @author sandzhang<sandzhangtoo@gmail.com>
+ */
+public class JSONDruidStatService {
+
+    private final static JSONDruidStatService instance               = new JSONDruidStatService();
+
+    private static DruidStatManagerFacade     druidStatManager       = DruidStatManagerFacade.getInstance();
+
+    public final static int                   RESULT_CODE_SUCCESS    = 1;
+    public final static int                   RESULT_CODE_ERROR      = -1;
+
+    private final static int                  DEFAULT_PAGE           = 1;
+    private final static int                  DEFAULT_PER_PAGE_COUNT = Integer.MAX_VALUE;
+    private static final String               DEFAULT_ORDER_TYPE     = "asc";
+
+    private JSONDruidStatService() {
+    }
+
+    public static JSONDruidStatService getInstance() {
+        return instance;
+    }
+
+    public String service(String url) {
+
+        Map<String, String> parameters = StringUtils.getParameters(url);
+
+        if (url.equals("/basic.json")) {
+            return returnJSONResult(RESULT_CODE_SUCCESS, druidStatManager.returnJSONBasicStat());
+        }
+
+        if (url.equals("/reset-all.json")) {
+            druidStatManager.resetAll();
+            return returnJSONResult(RESULT_CODE_SUCCESS, null);
+        }
+
+        if (url.equals("/datasource.json")) {
+            return returnJSONResult(RESULT_CODE_SUCCESS, druidStatManager.getDataSourceStatList());
+        }
+
+        if (url.startsWith("/datasource-")) {
+            Integer id = StringUtils.subStringToInteger(url, "datasource-", ".");
+            Object result = druidStatManager.getDataSourceStatData(id);
+            return returnJSONResult(result == null ? RESULT_CODE_ERROR : RESULT_CODE_SUCCESS, result);
+        }
+
+        if (url.startsWith("/connectionInfo-") && url.endsWith(".json")) {
+            Integer id = StringUtils.subStringToInteger(url, "connectionInfo-", ".");
+            List<?> connectionInfoList = druidStatManager.getPoolingConnectionInfoByDataSourceId(id);
+            return returnJSONResult(connectionInfoList == null ? RESULT_CODE_ERROR : RESULT_CODE_SUCCESS,
+                                    connectionInfoList);
+        }
+
+        if (url.startsWith("/activeConnectionStackTrace-") && url.endsWith(".json")) {
+            Integer id = StringUtils.subStringToInteger(url, "activeConnectionStackTrace-", ".");
+            return returnJSONActiveConnectionStackTrace(id);
+        }
+
+        if (url.startsWith("/sql.json")) {
+            return returnJSONResult(RESULT_CODE_SUCCESS, getSqlStatDataList(parameters));
+        }
+
+        if (url.startsWith("/sql-") && url.indexOf(".json") > 0) {
+            Integer id = StringUtils.subStringToInteger(url, "sql-", ".json");
+
+            Object result = druidStatManager.getSqlStatData(id);
+            return returnJSONResult(result == null ? RESULT_CODE_ERROR : RESULT_CODE_SUCCESS, result);
+        }
+
+        return returnJSONResult(RESULT_CODE_ERROR, "Do not support this request, please contact with administrator.");
+    }
+
+    private List<Map<String, Object>> getSqlStatDataList(Map<String, String> parameters) {
+        String orderBy = parameters.get("orderBy");
+        String orderType = parameters.get("orderType");
+        Integer page = Integer.parseInt(parameters.get("page"));
+        Integer perPageCount = Integer.parseInt(parameters.get("perPageCount"));
+
+        if (!"desc".equals(orderType)) orderType = DEFAULT_ORDER_TYPE;
+
+        if (page == null) page = DEFAULT_PAGE;
+        if (perPageCount == null) perPageCount = DEFAULT_PER_PAGE_COUNT;
+
+        List<Map<String, Object>> array = new ArrayList<Map<String, Object>>();
+        for (DruidDataSource datasource : druidStatManager.getDruidDataSourceInstances()) {
+            for (JdbcSqlStat sqlStat : datasource.getDataSourceStat().getSqlStatMap().values()) {
+                if (sqlStat.getExecuteCount() == 0 && sqlStat.getRunningCount() == 0) {
+                    continue;
+                }
+
+                array.add(druidStatManager.getSqlStatData(sqlStat));
+            }
+        }
+
+        // orderby
+        if (orderBy != null && orderBy.trim().length() != 0) {
+            Collections.sort(array, new MapComparator<String, Object>(orderBy, DEFAULT_ORDER_TYPE.equals(orderType)));
+        }
+
+        // page
+        int fromIndex = (page - 1) * perPageCount;
+        int toIndex = page * perPageCount;
+        if (toIndex > array.size()) toIndex = array.size();
+
+        return array.subList(fromIndex, toIndex);
+    }
+
+    private String returnJSONActiveConnectionStackTrace(Integer id) {
+        List<String> result = druidStatManager.getActiveConnectionStackTraceByDataSourceId(id);
+
+        if (result == null) {
+            return returnJSONResult(RESULT_CODE_ERROR, "require set removeAbandoned=true");
+        }
+        return returnJSONResult(RESULT_CODE_SUCCESS, result);
+    }
+
+    private String returnJSONResult(int resultCode, Object content) {
+        Map<String, Object> dataMap = new LinkedHashMap<String, Object>();
+        dataMap.put("ResultCode", resultCode);
+        dataMap.put("Content", content);
+        return JSON.toJSONString(dataMap);
+    }
+
+}