mirror of
https://gitee.com/huoyo/ko-time.git
synced 2025-12-08 01:38:26 +08:00
304 lines
11 KiB
Java
304 lines
11 KiB
Java
package cn.langpy.kotime.controller;
|
|
|
|
import cn.langpy.kotime.annotation.Auth;
|
|
import cn.langpy.kotime.config.DefaultConfig;
|
|
import cn.langpy.kotime.constant.KoConstant;
|
|
import cn.langpy.kotime.model.*;
|
|
import cn.langpy.kotime.service.GraphService;
|
|
import cn.langpy.kotime.util.ClassUtil;
|
|
import cn.langpy.kotime.util.Context;
|
|
import cn.langpy.kotime.util.InvalidAuthInfoException;
|
|
import cn.langpy.kotime.util.KoUtil;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.core.io.ClassPathResource;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.util.StringUtils;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import java.io.*;
|
|
import java.util.Collections;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.logging.Logger;
|
|
|
|
/**
|
|
* zhangchang
|
|
*/
|
|
@Controller
|
|
@RequestMapping("/koTime")
|
|
public class KoTimeController {
|
|
@Value("${ko-time.user-name:}")
|
|
private String userName;
|
|
@Value("${ko-time.password:}")
|
|
private String password;
|
|
|
|
@Value("${ko-time.agent-path:}")
|
|
private String agentPath;
|
|
|
|
private static Logger log = Logger.getLogger(KoTimeController.class.toString());
|
|
private final String uiKitCssText = getResourceText("kostatic/uikit.min.css");
|
|
private final String uiKitJsText = getResourceText("kostatic/uikit.min.js");
|
|
private final String metricFlowJsText = getResourceText("kostatic/Metricflow.js");
|
|
private final String jQueryJsText = getResourceText("kostatic/JQuery.min.js");
|
|
private final String uiKitIconsJs = getResourceText("kostatic/uikit-icons.js");
|
|
|
|
@PostMapping("/login")
|
|
@ResponseBody
|
|
public Map login(@RequestBody UserInfo userInfo) {
|
|
if (null == userInfo || !StringUtils.hasText(userInfo.getUserName()) || !StringUtils.hasText(userInfo.getPassword())) {
|
|
throw new InvalidAuthInfoException("failed to login for kotime,please fill userName and password!");
|
|
}
|
|
Map map = new HashMap();
|
|
if (userName.equals(userInfo.getUserName()) && password.equals(userInfo.getPassword())) {
|
|
String token = KoUtil.login(userInfo.getUserName());
|
|
map.put("state", 1);
|
|
map.put("token", token);
|
|
return map;
|
|
}
|
|
map.put("state", 0);
|
|
return map;
|
|
}
|
|
|
|
@GetMapping("/isLogin")
|
|
@ResponseBody
|
|
public Map isLogin(String token) {
|
|
Map map = new HashMap();
|
|
map.put("state", 1);
|
|
boolean checkLogin = KoUtil.isLogin(token);
|
|
map.put("isLogin", checkLogin ? 1 : 0);
|
|
return map;
|
|
}
|
|
|
|
|
|
@GetMapping
|
|
public void index(String test, HttpServletResponse response, HttpServletRequest request) {
|
|
if (!Context.getConfig().getEnable()) {
|
|
return;
|
|
}
|
|
if (null != test) {
|
|
return;
|
|
}
|
|
response.setContentType("text/html;charset=utf-8");
|
|
ClassPathResource classPathResource = new ClassPathResource(KoConstant.getViewName());
|
|
try (
|
|
InputStream inputStream = classPathResource.getInputStream();
|
|
InputStreamReader streamReader = new InputStreamReader(inputStream, "utf-8");
|
|
BufferedReader reader = new BufferedReader(streamReader);
|
|
PrintWriter out = response.getWriter()) {
|
|
|
|
String context = request.getContextPath();
|
|
if (StringUtils.hasText(Context.getConfig().getContextPath())) {
|
|
context = Context.getConfig().getContextPath();
|
|
}
|
|
StringBuilder stringBuilder = new StringBuilder();
|
|
String line = "";
|
|
while ((line = reader.readLine()) != null) {
|
|
if (line.indexOf(KoConstant.globalThreshold) > -1) {
|
|
line = line.replace(KoConstant.globalThreshold, Context.getConfig().getThreshold() + "");
|
|
} else if (line.indexOf(KoConstant.globalNeedLogin) > -1) {
|
|
line = line.replace(KoConstant.globalNeedLogin, Context.getConfig().getAuthEnable() + "");
|
|
} else if (line.indexOf(KoConstant.contextPath) > -1) {
|
|
line = line.replace(KoConstant.contextPath, context);
|
|
} else if (line.indexOf(KoConstant.exceptionTitleStyle) > -1) {
|
|
line = line.replace(KoConstant.exceptionTitleStyle, Context.getConfig().getExceptionEnable() == true ? "" : "display:none;");
|
|
} else if (line.indexOf("UIKitCss") > -1) {
|
|
line = line.replace("UIKitCss", uiKitCssText);
|
|
} else if (line.indexOf("UIKitJs") > -1) {
|
|
line = line.replace("UIKitJs", uiKitJsText);
|
|
} else if (line.indexOf("MetricFlowJs") > -1) {
|
|
line = line.replace("MetricFlowJs", metricFlowJsText);
|
|
}else if (line.indexOf("jQueryJs") > -1) {
|
|
line = line.replace("jQueryJs", jQueryJsText);
|
|
}else if (line.indexOf("uiKitIconsJs") > -1) {
|
|
line = line.replace("uiKitIconsJs", uiKitIconsJs);
|
|
}
|
|
stringBuilder.append(line + "\n");
|
|
}
|
|
line = stringBuilder.toString();
|
|
out.write(line);
|
|
out.flush();
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
private String getResourceText(String fileName) {
|
|
ClassPathResource classPathResource = new ClassPathResource(fileName);
|
|
try (InputStream inputStream = classPathResource.getInputStream();
|
|
InputStreamReader streamReader = new InputStreamReader(inputStream, "utf-8");
|
|
BufferedReader reader = new BufferedReader(streamReader)) {
|
|
String line = "";
|
|
StringBuilder stringBuilder = new StringBuilder();
|
|
while ((line = reader.readLine()) != null) {
|
|
stringBuilder.append(line + "\n");
|
|
}
|
|
return stringBuilder.toString();
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
return "";
|
|
}
|
|
|
|
|
|
@GetMapping("/getConfig")
|
|
@ResponseBody
|
|
@Auth
|
|
public DefaultConfig getConfig() {
|
|
DefaultConfig config = Context.getConfig();
|
|
return config;
|
|
}
|
|
|
|
@GetMapping("/getStatistic")
|
|
@ResponseBody
|
|
@Auth
|
|
public SystemStatistic getStatistic() {
|
|
GraphService graphService = GraphService.getInstance();
|
|
SystemStatistic system = graphService.getRunStatistic();
|
|
return system;
|
|
}
|
|
|
|
@GetMapping("/getApis")
|
|
@ResponseBody
|
|
@Auth
|
|
public List<MethodInfo> getApis(String question) {
|
|
GraphService graphService = GraphService.getInstance();
|
|
List<MethodInfo> list = null;
|
|
if (StringUtils.hasText(question)) {
|
|
list = graphService.searchMethods(question);
|
|
} else {
|
|
list = graphService.getControllers();
|
|
}
|
|
Collections.sort(list);
|
|
return list;
|
|
}
|
|
|
|
@GetMapping("/getParamGraph")
|
|
@ResponseBody
|
|
@Auth
|
|
public Map<String, ParamMetric> getParamGraph(String methodId) {
|
|
GraphService graphService = GraphService.getInstance();
|
|
Map<String, ParamMetric> list = graphService.getMethodParamGraph(methodId);
|
|
return list;
|
|
}
|
|
|
|
@GetMapping("/getApiTips")
|
|
@ResponseBody
|
|
@Auth
|
|
public List<String> getApiTips(String question) {
|
|
GraphService graphService = GraphService.getInstance();
|
|
List<String> list = graphService.getCondidates(question);
|
|
return list;
|
|
}
|
|
|
|
|
|
@GetMapping("/getExceptions")
|
|
@ResponseBody
|
|
@Auth
|
|
public List<ExceptionNode> getExceptions() {
|
|
GraphService graphService = GraphService.getInstance();
|
|
List<ExceptionNode> exceptionList = graphService.getExceptions();
|
|
return exceptionList;
|
|
}
|
|
|
|
@GetMapping("/getTree")
|
|
@ResponseBody
|
|
@Auth
|
|
public MethodInfo getTree(String methodName) {
|
|
GraphService graphService = GraphService.getInstance();
|
|
MethodInfo tree = graphService.getTree(methodName);
|
|
return tree;
|
|
}
|
|
|
|
@GetMapping("/getMethodsByExceptionId")
|
|
@ResponseBody
|
|
@Auth
|
|
public List<ExceptionInfo> getMethodsByExceptionId(String exceptionId,String message) {
|
|
GraphService graphService = GraphService.getInstance();
|
|
List<ExceptionInfo> exceptionInfos = graphService.getExceptionInfos(exceptionId,message);
|
|
return exceptionInfos;
|
|
}
|
|
|
|
@PostMapping("/updateConfig")
|
|
@ResponseBody
|
|
@Auth
|
|
public boolean updateConfig(@RequestBody DefaultConfig config) {
|
|
DefaultConfig koTimeConfig = Context.getConfig();
|
|
if (config.getEnable() != null) {
|
|
koTimeConfig.setEnable(config.getEnable());
|
|
}
|
|
if (config.getExceptionEnable() != null) {
|
|
koTimeConfig.setExceptionEnable(config.getExceptionEnable());
|
|
}
|
|
if (config.getLogEnable() != null) {
|
|
koTimeConfig.setLogEnable(config.getLogEnable());
|
|
}
|
|
if (config.getThreshold() != null) {
|
|
koTimeConfig.setThreshold(config.getThreshold());
|
|
}
|
|
if (config.getLanguage() != null) {
|
|
koTimeConfig.setLanguage(config.getLanguage());
|
|
}
|
|
return true;
|
|
}
|
|
@PostMapping("/updateClass")
|
|
@ResponseBody
|
|
@Auth
|
|
public Map updateClass(@RequestParam("classFile") MultipartFile classFile,String className) throws ClassNotFoundException {
|
|
Map map = new HashMap();
|
|
if (classFile==null || classFile.isEmpty()) {
|
|
map.put("state", 0);
|
|
map.put("message", "文件不能为空");
|
|
return map;
|
|
}
|
|
File file = null;
|
|
try {
|
|
String originalFilename = classFile.getOriginalFilename();
|
|
if (!originalFilename.endsWith(".class")) {
|
|
map.put("state", 0);
|
|
map.put("message", "仅支持.class文件");
|
|
return map;
|
|
}
|
|
String[] filename = originalFilename.split("\\.");
|
|
file = uploadFile(classFile.getBytes(),filename[0]);
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
File jar = null;
|
|
if (StringUtils.hasText(agentPath)) {
|
|
jar = ClassUtil.createJar();
|
|
agentPath = jar.getAbsolutePath();
|
|
}
|
|
ClassUtil.updateClass(agentPath,className.trim(),file.getAbsolutePath());
|
|
file.delete();
|
|
if (jar!=null) {
|
|
jar.delete();
|
|
}
|
|
map.put("state", 1);
|
|
map.put("message", "更新成功");
|
|
return map;
|
|
}
|
|
|
|
public static File uploadFile(byte[] file,String fileName) throws IOException {
|
|
FileOutputStream out = null;
|
|
try {
|
|
File targetFile = File.createTempFile(fileName, ".class", new File(System.getProperty("java.io.tmpdir")));
|
|
out = new FileOutputStream(targetFile.getAbsolutePath());
|
|
out.write(file);
|
|
out.flush();
|
|
out.close();
|
|
return targetFile;
|
|
} catch (Exception e) {
|
|
log.severe("" + e);
|
|
}finally {
|
|
if(out !=null){
|
|
out.flush();
|
|
out.close();
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|