/** * Copyright © 2015-2020 JeePlus All rights reserved. */ package com.jeeplus.modules.tools.web; import java.io.BufferedReader; import java.io.ByteArrayOutputStream; import java.io.InputStream; import java.io.InputStreamReader; import java.net.HttpURLConnection; import java.net.URL; import java.util.HashMap; import java.util.List; import java.util.Map; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.shiro.authz.annotation.Logical; import org.apache.shiro.authz.annotation.RequiresPermissions; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.ModelAttribute; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.servlet.mvc.support.RedirectAttributes; import com.google.common.collect.Lists; import com.jeeplus.common.config.Global; import com.jeeplus.common.persistence.Page; import com.jeeplus.common.utils.DateUtils; import com.jeeplus.common.utils.StringUtils; import com.jeeplus.common.utils.excel.ExportExcel; import com.jeeplus.common.utils.excel.ImportExcel; import com.jeeplus.common.web.BaseController; import com.jeeplus.modules.tools.entity.TestInterface; import com.jeeplus.modules.tools.service.TestInterfaceService; import com.jeeplus.modules.tools.utils.HttpPostTest; /** * 接口Controller * @author lgf * @version 2016-01-07 */ @Controller @RequestMapping(value = "${adminPath}/tools/testInterface") public class TestInterfaceController extends BaseController { @Autowired private TestInterfaceService testInterfaceService; @ModelAttribute public TestInterface get(@RequestParam(required=false) String id) { TestInterface entity = null; if (StringUtils.isNotBlank(id)){ entity = testInterfaceService.get(id); } if (entity == null){ entity = new TestInterface(); } return entity; } /** * 接口列表页面 */ @RequiresPermissions("tools:testInterface:list") @RequestMapping(value = {"list", ""}) public String list(TestInterface testInterface, HttpServletRequest request, HttpServletResponse response, Model model) { Page page = testInterfaceService.findPage(new Page(request, response), testInterface); model.addAttribute("page", page); return "modules/tools/testInterfaceList"; } /** * 查看,增加,编辑接口表单页面 */ @RequiresPermissions(value={"tools:testInterface:view","tools:testInterface:add","tools:testInterface:edit"},logical=Logical.OR) @RequestMapping(value = "form") public String form(TestInterface testInterface, Model model) { model.addAttribute("testInterface", testInterface); return "modules/tools/testInterfaceForm"; } /** * 保存接口 */ @RequiresPermissions(value={"tools:testInterface:add","tools:testInterface:edit"},logical=Logical.OR) @RequestMapping(value = "save") public String save(TestInterface testInterface, Model model, RedirectAttributes redirectAttributes) { if (!beanValidator(model, testInterface)){ return form(testInterface, model); } testInterfaceService.save(testInterface); addMessage(redirectAttributes, "保存接口成功"); return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } /** * 删除接口 */ @RequiresPermissions("tools:testInterface:del") @RequestMapping(value = "delete") public String delete(TestInterface testInterface, RedirectAttributes redirectAttributes) { testInterfaceService.delete(testInterface); addMessage(redirectAttributes, "删除接口成功"); return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } /** * 批量删除接口 */ @RequiresPermissions("tools:testInterface:del") @RequestMapping(value = "deleteAll") public String deleteAll(String ids, RedirectAttributes redirectAttributes) { String idArray[] =ids.split(","); for(String id : idArray){ testInterfaceService.delete(testInterfaceService.get(id)); } addMessage(redirectAttributes, "删除接口成功"); return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } /** * 导出excel文件 */ @RequiresPermissions("tools:testInterface:export") @RequestMapping(value = "export", method=RequestMethod.POST) public String exportFile(TestInterface testInterface, HttpServletRequest request, HttpServletResponse response, RedirectAttributes redirectAttributes) { try { String fileName = "接口"+DateUtils.getDate("yyyyMMddHHmmss")+".xlsx"; Page page = testInterfaceService.findPage(new Page(request, response, -1), testInterface); new ExportExcel("接口", TestInterface.class).setDataList(page.getList()).write(response, fileName).dispose(); return null; } catch (Exception e) { addMessage(redirectAttributes, "导出接口记录失败!失败信息:"+e.getMessage()); } return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } /** * 导入Excel数据 */ @RequiresPermissions("tools:testInterface:import") @RequestMapping(value = "import", method=RequestMethod.POST) public String importFile(MultipartFile file, RedirectAttributes redirectAttributes) { try { int successNum = 0; ImportExcel ei = new ImportExcel(file, 1, 0); List list = ei.getDataList(TestInterface.class); for (TestInterface testInterface : list){ testInterfaceService.save(testInterface); } addMessage(redirectAttributes, "已成功导入 "+successNum+" 条接口记录"); } catch (Exception e) { addMessage(redirectAttributes, "导入接口失败!失败信息:"+e.getMessage()); } return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } /** * 下载导入接口数据模板 */ @RequiresPermissions("tools:testInterface:import") @RequestMapping(value = "import/template") public String importFileTemplate(HttpServletResponse response, RedirectAttributes redirectAttributes) { try { String fileName = "接口数据导入模板.xlsx"; List list = Lists.newArrayList(); new ExportExcel("接口数据", TestInterface.class, 1).setDataList(list).write(response, fileName).dispose(); return null; } catch (Exception e) { addMessage(redirectAttributes, "导入模板下载失败!失败信息:"+e.getMessage()); } return "redirect:"+Global.getAdminPath()+"/tools/testInterface/?repage"; } @RequiresPermissions("tools:testInterface:test") @RequestMapping(value="/test") public String form(String id, HttpServletRequest request, HttpServletResponse response, Model model) { TestInterface entity = testInterfaceService.get(id); model.addAttribute("testInterface", entity); return "modules/tools/interfaceTest"; } /** * 接口内部请求 * @param * @throws Exception */ @RequestMapping(value="/severTest") @ResponseBody public Object severTest(HttpServletRequest request, HttpServletResponse response, Model model){ Map map = new HashMap(); String errInfo = "success",str = "",rTime=""; try{ long startTime = System.currentTimeMillis(); String s_url = request.getParameter("serverUrl");//请求起始时间_毫秒 String type = request.getParameter("requestMethod"); String requestBody = request.getParameter("requestBody"); URL url; if(type.equals("POST")){//请求类型 POST or GET Map params = new HashMap(); if(requestBody!=null && !requestBody.equals("")){ String[] paramList = requestBody.split("&"); for(String param : paramList){ if(param.split("=").length == 2){ params.put(param.split("=")[0], param.split("=")[1]); }else{ params.put(param.split("=")[0], ""); } } } HttpPostTest test = new HttpPostTest(s_url, params); str= test.post(); }else{ url = new URL(s_url); HttpURLConnection connection = (HttpURLConnection) url.openConnection(); connection.setRequestMethod("GET"); BufferedReader in = new BufferedReader(new InputStreamReader(url.openStream(),"UTF-8")); //请求结束时间_毫秒 String temp = ""; while((temp = in.readLine()) != null){ str = str + temp; } } long endTime = System.currentTimeMillis(); rTime = String.valueOf(endTime - startTime); } catch(Exception e){ errInfo = "error"; str = e.getMessage(); } map.put("errInfo", errInfo); //状态信息 map.put("result", str); //返回结果 map.put("rTime", rTime); //服务器请求时间 毫秒 return map; } public static byte[] readInputStream(InputStream inStream) throws Exception{ ByteArrayOutputStream outStream = new ByteArrayOutputStream(); byte[] buffer = new byte[1024]; int len = 0; while( (len = inStream.read(buffer)) !=-1 ){ outStream.write(buffer, 0, len); } byte[] data = outStream.toByteArray();//网页的二进制数据 outStream.close(); inStream.close(); return data; } }