Druid——基于SpringMVC自定义Controller简单封装DEMO

悠悠 2022-11-13 00:54 170阅读 0赞

# DEMO #

import com.alibaba.druid.sql.SQLUtils;
    import com.alibaba.druid.sql.ast.SQLStatement;
    import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
    import com.alibaba.druid.stat.DruidStatManagerFacade;
    import com.alibaba.druid.support.http.stat.WebAppStatManager;
    import com.alibaba.druid.support.spring.stat.SpringStatManager;
    import com.alibaba.druid.util.MapComparator;
    import com.hailiu.web.bean.ResponseBean;
    import io.swagger.annotations.Api;
    import io.swagger.annotations.ApiOperation;
    import org.springframework.web.bind.annotation.*;
    
    import java.util.Collections;
    import java.util.List;
    import java.util.Map;
    
    /**
     * Druid监控
     *
     * @author ShenTuZhiGang
     * @version 1.0.0
     * @date 2021-03-24 16:38
     */
    @RestController
    @RequestMapping("/api/v1/admin/system/druid")
    @Api(tags = "Druid前端控制器")
    public class DruidAdminController {
    
        private final static String           DEFAULT_PAGE           = String.valueOf(1);
        private final static String           DEFAULT_PER_PAGE_COUNT = String.valueOf(Integer.MAX_VALUE);
        private static final String           ORDER_TYPE_DESC        = "desc";
        private static final String           ORDER_TYPE_ASC         = "asc";
        private static final String           DEFAULT_ORDER_TYPE     = ORDER_TYPE_ASC;
        private static final String           DEFAULT_ORDER_BY       = "SQL";
    
        /**
         * Druid druidStatManager# 该类可以获取所有数据源的监控数据。
         */
        private final DruidStatManagerFacade druidStatManager = DruidStatManagerFacade.getInstance();
        /**
         *
         */
        private final WebAppStatManager webAppStatManager = WebAppStatManager.getInstance();
        /**
         *
         */
        private final SpringStatManager springStatManager = SpringStatManager.getInstance();
        /**
         * 基本信息
         * basic info.
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "基本信息", notes = "基本信息")
        @RequestMapping(value = "/basic",method = RequestMethod.GET)
        public ResponseBean<?> basic(){
            return ResponseBean.success(druidStatManager.returnJSONBasicStat());
        }
    
        /**
         * 数据源
         * DatsSourceStat List
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "数据源", notes = "数据源")
        @RequestMapping(value = "/datasource",method = RequestMethod.GET)
        public ResponseBean<?> datasource(){
            return ResponseBean.success(druidStatManager.getDataSourceStatDataList());
        }
    
        /**
         * 数据源细节
         * DatasourceStat for id={id},id is DatsSourceStat.ID
         * Example: druid/datasource-1
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "数据源细节", notes = "数据源细节")
        @RequestMapping(value = "/datasource-{id}",method = RequestMethod.GET)
        public ResponseBean<?> datasource(@PathVariable(value = "id")Integer id){
            Object result = druidStatManager.getDataSourceStatData(id);
            return result == null ? ResponseBean.error() : ResponseBean.success(result);
        }
    
        /**
         * 活跃连接堆栈
         * Active Connection StackTrace List
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "活跃连接堆栈", notes = "活跃连接堆栈")
        @RequestMapping(value = "/activeConnectionStackTrace",method = RequestMethod.GET)
        public ResponseBean<?> activeConnectionStackTrace(){
            return ResponseBean.success(druidStatManager.getActiveConnStackTraceList());
        }
    
        /**
         * 活跃连接堆栈细节
         * StackTrace for activeConnection in datasource which datasource.id={datasourceId},id is DatsSourceStat.ID
         * Required: set removeAbandoned=true
         * Example: druid/activeConnectionStackTrace-1
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "活跃连接堆栈细节", notes = "活跃连接堆栈细节")
        @RequestMapping(value = "/activeConnectionStackTrace-{id}",method = RequestMethod.GET)
        public ResponseBean<?> activeConnectionStackTrace(@PathVariable(value = "id")Integer id){
            List<String> result = druidStatManager.getActiveConnectionStackTraceByDataSourceId(id);
            if (result == null) {
                return ResponseBean.error("require set removeAbandoned=true");
            }
            return ResponseBean.success(result);
        }
    
        /**
         * 连接池中连接信息
         * Info for polling connection
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "连接池中连接信息", notes = "连接池中连接信息")
        @RequestMapping(value = "/connectionInfo-{id}",method = RequestMethod.GET)
        public ResponseBean<?> connectionInfo(@PathVariable(value = "id")Integer id){
            List<?> connectionInfoList = druidStatManager.getPoolingConnectionInfoByDataSourceId(id);
            return connectionInfoList == null ? ResponseBean.error() : ResponseBean.success(connectionInfoList);
        }
    
        /**
         * SQL监控
         * SqlStat List
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "SQL监控", notes = "SQL监控")
        @RequestMapping(value = "/sql",method = RequestMethod.GET)
        public ResponseBean<?> sql(@RequestParam(value = "dataSourceId", required = false)Integer dataSourceId,
                                   @RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                   @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                   @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                   @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            List<Map<String, Object>> array = druidStatManager.getSqlStatDataList(dataSourceId);
            // order by the statData array
            if (orderBy.trim().length() != 0) {
                Collections.sort(array, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
            }
            array = array.subList((page - 1) * perPageCount, Math.min(page * perPageCount,array.size()));
            return ResponseBean.success(array);
        }
    
        /**
         * SQL监控细节
         * SqlStat for id={id} ,id is SqlStat.ID
         * Example: druid/sql-1
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "SQL监控细节", notes = "SQL监控细节")
        @RequestMapping(value = "/sql-{id}",method = RequestMethod.GET)
        public ResponseBean<?> sql(@PathVariable(value = "id")Integer id){
            Map<String, Object> map =  druidStatManager.getSqlStatData(id);
    
            if (map == null) {
                return ResponseBean.error();
            }
    
            String dbType = (String) map.get("DbType");
            String sql = (String) map.get("SQL");
    
            map.put("formattedSql", SQLUtils.format(sql, dbType));
            List<SQLStatement> statementList = SQLUtils.parseStatements(sql, dbType);
    
            if (!statementList.isEmpty()) {
                SQLStatement sqlStmt = statementList.get(0);
                SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(dbType);
                sqlStmt.accept(visitor);
                map.put("parsedTable", visitor.getTables().toString());
                map.put("parsedFields", visitor.getColumns().toString());
                map.put("parsedConditions", visitor.getConditions().toString());
                map.put("parsedRelationships", visitor.getRelationships().toString());
                map.put("parsedOrderbycolumns", visitor.getOrderByColumns().toString());
            }
    
            return map == null ? ResponseBean.error() : ResponseBean.success(map);
        }
    
        /**
         * SQL防火墙
         * WallStat List
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "SQL防火墙", notes = "SQL防火墙")
        @RequestMapping(value = "/wall",method = RequestMethod.GET)
        public ResponseBean<?> wall(@RequestParam(value = "dataSourceId",required = false)Integer dataSourceId,
                                    @RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                    @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                    @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                    @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            Map<String, Object> result =  druidStatManager.getWallStatMap(dataSourceId);
            if (result != null) {
                List<Map<String, Object>> tables = (List<Map<String, Object>>) result.get("tables");
                if (tables != null) {
                    // order by the statData array
                    if (orderBy.trim().length() != 0) {
                        Collections.sort(tables, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
                    }
                    tables = tables.subList((page - 1) * perPageCount, Math.min(page * perPageCount,tables.size()));
                    result.put("tables", tables);
                }
    
                List<Map<String, Object>> functions = (List<Map<String, Object>>) result.get("functions");
                if (functions != null) {
                    if (orderBy.trim().length() != 0) {
                        Collections.sort(functions, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
                    }
                    functions = functions.subList((page - 1) * perPageCount, Math.min(page * perPageCount,functions.size()));
                    result.put("functions", functions);
                }
            } else {
                result = Collections.emptyMap();
            }
            return ResponseBean.success(result);
        }
    
        /**
         * SQL防火墙细节
         * WallStat for id={id},id is DatsSourceStat.ID
         * Example: druid/wall-1
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "SQL防火墙细节", notes = "SQL防火墙细节")
        @RequestMapping(value = "/wall-{id}",method = RequestMethod.GET)
        public ResponseBean<?> wall(@PathVariable(value = "id")Integer id){
            Object result = druidStatManager.getWallStatMap(id);
            return result == null ? ResponseBean.error() : ResponseBean.success(result);
        }
    
        /**
         * Web URI监控
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Web URI监控", notes = "Web URI监控")
        @RequestMapping(value = "/weburi",method = RequestMethod.GET)
        public ResponseBean<?> weburi(@RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                      @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                      @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                      @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            List<Map<String, Object>> array = webAppStatManager.getURIStatData();
            // order by the statData array
            if (orderBy.trim().length() != 0) {
                Collections.sort(array, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
            }
            array = array.subList((page - 1) * perPageCount, Math.min(page * perPageCount,array.size()));
            return ResponseBean.success(array);
        }
    
        /**
         * Web URI监控细节
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Web URI监控细节", notes = "Web URI监控细节")
        @RequestMapping(value = "/weburi-{uri}",method = RequestMethod.GET)
        public ResponseBean<?> weburi(@PathVariable(value = "uri")String uri){
            return ResponseBean.success(webAppStatManager.getURIStatData(uri));
        }
    
        /**
         * Web应用监控
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Web应用监控", notes = "Web应用监控")
        @RequestMapping(value = "/webapp",method = RequestMethod.GET)
        public ResponseBean<?> webapp(@RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                      @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                      @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                      @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            List<Map<String, Object>> array = webAppStatManager.getWebAppStatData();
            // order by the statData array
            if (orderBy.trim().length() != 0) {
                Collections.sort(array, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
            }
            array = array.subList((page - 1) * perPageCount, Math.min(page * perPageCount,array.size()));
            return ResponseBean.success(array);
        }
    
        /**
         * Session监控
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Session监控", notes = "Session监控")
        @RequestMapping(value = "/websession",method = RequestMethod.GET)
        public ResponseBean<?> websession(@RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                          @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                          @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                          @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            List<Map<String, Object>> array = webAppStatManager.getSessionStatData();
            // order by the statData array
            if (orderBy.trim().length() != 0) {
                Collections.sort(array, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
            }
            array = array.subList((page - 1) * perPageCount, Math.min(page * perPageCount,array.size()));
            return ResponseBean.success(array);
        }
    
        /**
         * Session监控细节
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Session监控细节", notes = "Session监控细节")
        @RequestMapping(value = "/websession-{id}",method = RequestMethod.GET)
        public ResponseBean<?> websession(@PathVariable(value = "id")String id){
            return ResponseBean.success(webAppStatManager.getSessionStat(id));
        }
    
        /**
         * Spring 监控
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Spring 监控", notes = "Spring 监控")
        @RequestMapping(value = "/spring",method = RequestMethod.GET)
        public ResponseBean<?> spring(@RequestParam(defaultValue = DEFAULT_ORDER_BY) String orderBy,
                                      @RequestParam(defaultValue = DEFAULT_ORDER_TYPE) String  orderType,
                                      @RequestParam(defaultValue = DEFAULT_PAGE) Integer page,
                                      @RequestParam(defaultValue = DEFAULT_PER_PAGE_COUNT) Integer perPageCount){
            List<Map<String, Object>> array = springStatManager.getMethodStatData();
            // order by the statData array
            if (orderBy.trim().length() != 0) {
                Collections.sort(array, new MapComparator<String, Object>(orderBy, ORDER_TYPE_DESC.equals(orderType)));
            }
            array = array.subList((page - 1) * perPageCount, Math.min(page * perPageCount,array.size()));
            return ResponseBean.success(array);
        }
    
        /**
         * Spring 监控细节
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "Spring 监控细节", notes = "Spring 监控细节")
        @RequestMapping(value = "/spring-detail",method = RequestMethod.GET)
        public ResponseBean<?> springDetail(String clazz, String method){
            return ResponseBean.success(springStatManager.getMethodStatData(clazz,method));
        }
    
        /**
         * Reset all stat
         * 重置
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "重置", notes = "重置")
        @RequestMapping(value = "/reset-all",method = RequestMethod.POST)
        public ResponseBean<?> resetAll(){
            druidStatManager.resetAll();
            return ResponseBean.success("重置成功");
        }
    
        /**
         * 记录日志并重置
         *
         * @return ResponseBean
         */
        @ApiOperation(value = "记录日志并重置", notes = "记录日志并重置")
        @RequestMapping(value = "/log-and-reset",method = RequestMethod.POST)
        public ResponseBean<?> logAndResetDataSource(){
            druidStatManager.logAndResetDataSource();
            return ResponseBean.success("重置成功");
        }
    }