2121import javax .servlet .http .HttpServletRequest ;
2222import javax .servlet .http .HttpServletResponse ;
2323import java .io .*;
24- import java .util .Collections ;
25- import java .util .HashMap ;
26- import java .util .List ;
27- import java .util .Map ;
24+ import java .util .*;
2825import java .util .logging .Logger ;
26+ import java .util .stream .Collectors ;
2927
3028/**
3129 * zhangchang
@@ -71,7 +69,7 @@ public Map isLogin(String kotoken) {
7169 if (StringUtils .hasText (kotoken )) {
7270 if (kotoken .equals (Context .getConfig ().getStaticToken ())) {
7371 checkLogin = true ;
74- }else {
72+ } else {
7573 checkLogin = KoUtil .isLogin (kotoken );
7674 }
7775 }
@@ -81,7 +79,7 @@ public Map isLogin(String kotoken) {
8179
8280
8381 @ GetMapping
84- public void index (String kotoken ,String test , HttpServletResponse response , HttpServletRequest request ) {
82+ public void index (String kotoken , String test , HttpServletResponse response , HttpServletRequest request ) {
8583 if (!Context .getConfig ().getEnable ()) {
8684 return ;
8785 }
@@ -121,14 +119,14 @@ public void index(String kotoken,String test, HttpServletResponse response, Http
121119 line = line .replace ("UIKitJs" , uiKitJsText );
122120 } else if (line .indexOf ("MetricFlowJs" ) > -1 ) {
123121 line = line .replace ("MetricFlowJs" , metricFlowJsText );
124- }else if (line .indexOf ("jQueryJs" ) > -1 ) {
122+ } else if (line .indexOf ("jQueryJs" ) > -1 ) {
125123 line = line .replace ("jQueryJs" , jQueryJsText );
126- }else if (line .indexOf ("uiKitIconsJs" ) > -1 ) {
124+ } else if (line .indexOf ("uiKitIconsJs" ) > -1 ) {
127125 line = line .replace ("uiKitIconsJs" , uiKitIconsJs );
128- }else if (line .indexOf ("staticTokenVisitValue" ) > -1 ) {
129- line = line .replace ("staticTokenVisitValue" , staticTokenVisit + "" );
130- }else if (line .indexOf ("staticTokenValue" ) > -1 ) {
131- line = line .replace ("staticTokenValue" , "'" + kotoken + "'" );
126+ } else if (line .indexOf ("staticTokenVisitValue" ) > -1 ) {
127+ line = line .replace ("staticTokenVisitValue" , staticTokenVisit + "" );
128+ } else if (line .indexOf ("staticTokenValue" ) > -1 ) {
129+ line = line .replace ("staticTokenValue" , "'" + kotoken + "'" );
132130 }
133131 stringBuilder .append (line + "\n " );
134132 }
@@ -139,6 +137,7 @@ public void index(String kotoken,String test, HttpServletResponse response, Http
139137 e .printStackTrace ();
140138 }
141139 }
140+
142141 private String getResourceText (String fileName ) {
143142 ClassPathResource classPathResource = new ClassPathResource (fileName );
144143 try (InputStream inputStream = classPathResource .getInputStream ();
@@ -229,9 +228,9 @@ public MethodInfo getTree(String methodName) {
229228 @ GetMapping ("/getMethodsByExceptionId" )
230229 @ ResponseBody
231230 @ Auth
232- public List <ExceptionInfo > getMethodsByExceptionId (String exceptionId ,String message ) {
231+ public List <ExceptionInfo > getMethodsByExceptionId (String exceptionId , String message ) {
233232 GraphService graphService = GraphService .getInstance ();
234- List <ExceptionInfo > exceptionInfos = graphService .getExceptionInfos (exceptionId ,message );
233+ List <ExceptionInfo > exceptionInfos = graphService .getExceptionInfos (exceptionId , message );
235234 return exceptionInfos ;
236235 }
237236
@@ -257,12 +256,13 @@ public boolean updateConfig(@RequestBody DefaultConfig config) {
257256 }
258257 return true ;
259258 }
259+
260260 @ PostMapping ("/updateClass" )
261261 @ ResponseBody
262262 @ Auth
263- public Map updateClass (@ RequestParam ("classFile" ) MultipartFile classFile ,String className ) {
263+ public Map updateClass (@ RequestParam ("classFile" ) MultipartFile classFile , String className ) {
264264 Map map = new HashMap ();
265- if (classFile == null || classFile .isEmpty ()) {
265+ if (classFile == null || classFile .isEmpty ()) {
266266 map .put ("state" , 0 );
267267 map .put ("message" , "文件不能为空" );
268268 return map ;
@@ -288,15 +288,15 @@ public Map updateClass(@RequestParam("classFile") MultipartFile classFile,String
288288 map .put ("message" , "请确认类名是否正确" );
289289 return map ;
290290 }
291- file = uploadFile (classFile .getBytes (),filename [0 ]);
291+ file = uploadFile (classFile .getBytes (), filename [0 ]);
292292 } catch (IOException e ) {
293293 log .severe ("Error class file!" );
294294 map .put ("state" , 0 );
295295 map .put ("message" , "无法解析文件" );
296296 return map ;
297297 }
298298 final ClassService classService = ClassService .getInstance ();
299- classService .updateClass (className ,file .getAbsolutePath ());
299+ classService .updateClass (className , file .getAbsolutePath ());
300300 file .deleteOnExit ();
301301
302302 map .put ("state" , 1 );
@@ -305,19 +305,18 @@ public Map updateClass(@RequestParam("classFile") MultipartFile classFile,String
305305 }
306306
307307
308-
309- private static File uploadFile (byte [] file ,String fileName ) throws IOException {
308+ private static File uploadFile (byte [] file , String fileName ) throws IOException {
310309 FileOutputStream out = null ;
311310 try {
312- File targetFile = File .createTempFile (fileName , ".class" , new File (System .getProperty ("java.io.tmpdir" )));
311+ File targetFile = File .createTempFile (fileName , ".class" , new File (System .getProperty ("java.io.tmpdir" )));
313312 out = new FileOutputStream (targetFile .getAbsolutePath ());
314313 out .write (file );
315314 out .flush ();
316315 return targetFile ;
317316 } catch (Exception e ) {
318317 log .severe ("" + e );
319- }finally {
320- if (out !=null ){
318+ } finally {
319+ if (out != null ) {
321320 out .flush ();
322321 out .close ();
323322 }
@@ -342,6 +341,7 @@ public HeapMemoryInfo getHeapMemoryInfo() {
342341 HeapMemoryInfo heapMemoryInfo = usageService .getHeapMemoryInfo ();
343342 return heapMemoryInfo ;
344343 }
344+
345345 @ GetMapping ("/getPhysicalMemoryInfo" )
346346 @ ResponseBody
347347 @ Auth
@@ -363,9 +363,20 @@ public boolean clearData() {
363363 @ GetMapping ("/getThreadsInfo" )
364364 @ ResponseBody
365365 @ Auth
366- public List getThreadsInfo () {
366+ public Map getThreadsInfo (String state ) {
367367 ThreadUsageService usageService = ThreadUsageService .newInstance ();
368368 List <ThreadInfo > threads = usageService .getThreads ();
369- return threads ;
369+ threads = threads .stream ().sorted (Comparator .comparing (ThreadInfo ::getState )).collect (Collectors .toList ());
370+
371+ Map <String , Long > stateCounting = threads .stream ().collect (Collectors .groupingBy (ThreadInfo ::getState , Collectors .counting ()));
372+ stateCounting .put ("all" ,(long )threads .size ());
373+
374+ Map map = new HashMap ();
375+ map .put ("statistics" , stateCounting );
376+ if (StringUtils .hasText (state )) {
377+ threads = threads .stream ().filter (a -> a .getState ().equals (state )).collect (Collectors .toList ());
378+ }
379+ map .put ("threads" , threads );
380+ return map ;
370381 }
371382}
0 commit comments