From 5461b8ebf2f8bf66a9d34351eb8783484304958a Mon Sep 17 00:00:00 2001 From: 648540858 <648540858@qq.com> Date: 星期一, 22 八月 2022 16:17:28 +0800 Subject: [PATCH] 支持全局异常和统一返回结果,未完待续 --- src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java | 187 +++++++++++++--------------------------------- 1 files changed, 53 insertions(+), 134 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java index 95995b8..08b0793 100644 --- a/src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java +++ b/src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java @@ -4,24 +4,27 @@ import com.alibaba.fastjson.JSONObject; import com.genersoft.iot.vmp.VManageBootstrap; import com.genersoft.iot.vmp.common.VersionPo; -import com.genersoft.iot.vmp.conf.DynamicTask; import com.genersoft.iot.vmp.conf.SipConfig; import com.genersoft.iot.vmp.conf.UserSetting; import com.genersoft.iot.vmp.conf.VersionInfo; +import com.genersoft.iot.vmp.conf.exception.ControllerException; import com.genersoft.iot.vmp.media.zlm.ZLMHttpHookSubscribe; import com.genersoft.iot.vmp.media.zlm.dto.IHookSubscribe; import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; import com.genersoft.iot.vmp.service.IMediaServerService; import com.genersoft.iot.vmp.utils.SpringBeanFactory; +import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; import com.genersoft.iot.vmp.vmanager.bean.WVPResult; import gov.nist.javax.sip.SipStackImpl; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; +import org.ehcache.xml.model.ThreadPoolsType; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; -import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; +import org.springframework.util.ObjectUtils; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.*; @@ -30,7 +33,6 @@ import javax.sip.SipProvider; import java.util.Iterator; import java.util.List; -import java.util.Set; @SuppressWarnings("rawtypes") @Tag(name = "鏈嶅姟鎺у埗") @@ -54,45 +56,34 @@ @Autowired private UserSetting userSetting; - @Autowired - private DynamicTask dynamicTask; - @Value("${server.port}") private int serverPort; + + + @Autowired + private ThreadPoolTaskExecutor taskExecutor; @GetMapping(value = "/media_server/list") @ResponseBody @Operation(summary = "娴佸獟浣撴湇鍔″垪琛�") - public WVPResult<List<MediaServerItem>> getMediaServerList() { - WVPResult<List<MediaServerItem>> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - result.setData(mediaServerService.getAll()); - return result; + public List<MediaServerItem> getMediaServerList() { + return mediaServerService.getAll(); } @GetMapping(value = "/media_server/online/list") @ResponseBody @Operation(summary = "鍦ㄧ嚎娴佸獟浣撴湇鍔″垪琛�") - public WVPResult<List<MediaServerItem>> getOnlineMediaServerList() { - WVPResult<List<MediaServerItem>> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - result.setData(mediaServerService.getAllOnline()); - return result; + public List<MediaServerItem> getOnlineMediaServerList() { + return mediaServerService.getAllOnline(); } @GetMapping(value = "/media_server/one/{id}") @ResponseBody @Operation(summary = "鍋滄瑙嗛鍥炴斁") @Parameter(name = "id", description = "娴佸獟浣撴湇鍔D", required = true) - public WVPResult<MediaServerItem> getMediaServer(@PathVariable String id) { - WVPResult<MediaServerItem> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - result.setData(mediaServerService.getOne(id)); - return result; + public MediaServerItem getMediaServer(@PathVariable String id) { + return mediaServerService.getOne(id); } @Operation(summary = "娴嬭瘯娴佸獟浣撴湇鍔�") @@ -101,7 +92,7 @@ @Parameter(name = "secret", description = "娴佸獟浣撴湇鍔ecret", required = true) @GetMapping(value = "/media_server/check") @ResponseBody - public WVPResult<MediaServerItem> checkMediaServer(@RequestParam String ip, @RequestParam int port, @RequestParam String secret) { + public MediaServerItem checkMediaServer(@RequestParam String ip, @RequestParam int port, @RequestParam String secret) { return mediaServerService.checkMediaServer(ip, port, secret); } @@ -110,122 +101,87 @@ @Parameter(name = "port", description = "娴佸獟浣撴湇鍔TT绔彛", required = true) @GetMapping(value = "/media_server/record/check") @ResponseBody - public WVPResult<String> checkMediaRecordServer(@RequestParam String ip, @RequestParam int port) { + public void checkMediaRecordServer(@RequestParam String ip, @RequestParam int port) { boolean checkResult = mediaServerService.checkMediaRecordServer(ip, port); - WVPResult<String> result = new WVPResult<>(); - if (checkResult) { - result.setCode(0); - result.setMsg("success"); - - } else { - result.setCode(-1); - result.setMsg("杩炴帴澶辫触"); + if (!checkResult) { + throw new ControllerException(ErrorCode.ERROR100.getCode(), "杩炴帴澶辫触"); } - return result; } @Operation(summary = "淇濆瓨娴佸獟浣撴湇鍔�") @Parameter(name = "mediaServerItem", description = "娴佸獟浣撲俊鎭�", required = true) @PostMapping(value = "/media_server/save") @ResponseBody - public WVPResult<String> saveMediaServer(@RequestBody MediaServerItem mediaServerItem) { + public void saveMediaServer(@RequestBody MediaServerItem mediaServerItem) { MediaServerItem mediaServerItemInDatabase = mediaServerService.getOne(mediaServerItem.getId()); if (mediaServerItemInDatabase != null) { - if (StringUtils.isEmpty(mediaServerItemInDatabase.getSendRtpPortRange()) && StringUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) { + if (ObjectUtils.isEmpty(mediaServerItemInDatabase.getSendRtpPortRange()) && ObjectUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) { mediaServerItem.setSendRtpPortRange("30000,30500"); } mediaServerService.update(mediaServerItem); } else { - if (StringUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) { + if (ObjectUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) { mediaServerItem.setSendRtpPortRange("30000,30500"); } - return mediaServerService.add(mediaServerItem); + mediaServerService.add(mediaServerItem); } - - WVPResult<String> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - return result; } @Operation(summary = "绉婚櫎娴佸獟浣撴湇鍔�") @Parameter(name = "id", description = "娴佸獟浣揑D", required = true) @DeleteMapping(value = "/media_server/delete") @ResponseBody - public WVPResult<String> deleteMediaServer(@RequestParam String id) { - if (mediaServerService.getOne(id) != null) { - mediaServerService.delete(id); - mediaServerService.deleteDb(id); - } else { - WVPResult<String> result = new WVPResult<>(); - result.setCode(-1); - result.setMsg("鏈壘鍒版鑺傜偣"); - return result; + public void deleteMediaServer(@RequestParam String id) { + if (mediaServerService.getOne(id) == null) { + throw new ControllerException(ErrorCode.ERROR100.getCode(), "鏈壘鍒版鑺傜偣"); } - WVPResult<String> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - return result; + mediaServerService.delete(id); + mediaServerService.deleteDb(id); } @Operation(summary = "閲嶅惎鏈嶅姟") @GetMapping(value = "/restart") @ResponseBody - public Object restart() { - Thread restartThread = new Thread(new Runnable() { - @Override - public void run() { - try { - Thread.sleep(3000); - SipProvider up = (SipProvider) SpringBeanFactory.getBean("udpSipProvider"); - SipStackImpl stack = (SipStackImpl) up.getSipStack(); - stack.stop(); - Iterator listener = stack.getListeningPoints(); - while (listener.hasNext()) { - stack.deleteListeningPoint((ListeningPoint) listener.next()); - } - Iterator providers = stack.getSipProviders(); - while (providers.hasNext()) { - stack.deleteSipProvider((SipProvider) providers.next()); - } - VManageBootstrap.restart(); - } catch (InterruptedException ignored) { - } catch (ObjectInUseException e) { - e.printStackTrace(); + public void restart() { + taskExecutor.execute(()-> { + try { + Thread.sleep(3000); + SipProvider up = (SipProvider) SpringBeanFactory.getBean("udpSipProvider"); + SipStackImpl stack = (SipStackImpl) up.getSipStack(); + stack.stop(); + Iterator listener = stack.getListeningPoints(); + while (listener.hasNext()) { + stack.deleteListeningPoint((ListeningPoint) listener.next()); } + Iterator providers = stack.getSipProviders(); + while (providers.hasNext()) { + stack.deleteSipProvider((SipProvider) providers.next()); + } + VManageBootstrap.restart(); + } catch (InterruptedException | ObjectInUseException e) { + throw new ControllerException(ErrorCode.ERROR100.getCode(), e.getMessage()); } }); - - restartThread.setDaemon(false); - restartThread.start(); - return "success"; - } + }; @Operation(summary = "鑾峰彇鐗堟湰淇℃伅") @GetMapping(value = "/version") @ResponseBody - public WVPResult<VersionPo> getVersion() { - WVPResult<VersionPo> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - result.setData(versionInfo.getVersion()); - return result; + public VersionPo VersionPogetVersion() { + return versionInfo.getVersion(); } @GetMapping(value = "/config") @Operation(summary = "鑾峰彇閰嶇疆淇℃伅") @Parameter(name = "type", description = "閰嶇疆绫诲瀷锛坰ip, base锛�", required = true) @ResponseBody - public WVPResult<JSONObject> getVersion(String type) { - WVPResult<JSONObject> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); + public JSONObject getVersion(String type) { JSONObject jsonObject = new JSONObject(); jsonObject.put("server.port", serverPort); - if (StringUtils.isEmpty(type)) { + if (ObjectUtils.isEmpty(type)) { jsonObject.put("sip", JSON.toJSON(sipConfig)); jsonObject.put("base", JSON.toJSON(userSetting)); } else { @@ -240,50 +196,13 @@ break; } } - result.setData(jsonObject); - return result; + return jsonObject; } @GetMapping(value = "/hooks") @ResponseBody @Operation(summary = "鑾峰彇褰撳墠鎵�鏈塰ook") - public WVPResult<List<IHookSubscribe>> getHooks() { - WVPResult<List<IHookSubscribe>> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - List<IHookSubscribe> all = zlmHttpHookSubscribe.getAll(); - result.setData(all); - return result; + public List<IHookSubscribe> getHooks() { + return zlmHttpHookSubscribe.getAll(); } - -// //@ApiOperation("褰撳墠杩涜涓殑鍔ㄦ�佷换鍔�") -// @GetMapping(value = "/dynamicTask") -// @ResponseBody -// public WVPResult<JSONObject> getDynamicTask(){ -// WVPResult<JSONObject> result = new WVPResult<>(); -// result.setCode(0); -// result.setMsg("success"); -// -// JSONObject jsonObject = new JSONObject(); -// -// Set<String> allKeys = dynamicTask.getAllKeys(); -// jsonObject.put("server.port", serverPort); -// if (StringUtils.isEmpty(type)) { -// jsonObject.put("sip", JSON.toJSON(sipConfig)); -// jsonObject.put("base", JSON.toJSON(userSetting)); -// }else { -// switch (type){ -// case "sip": -// jsonObject.put("sip", sipConfig); -// break; -// case "base": -// jsonObject.put("base", userSetting); -// break; -// default: -// break; -// } -// } -// result.setData(jsonObject); -// return result; -// } } -- Gitblit v1.8.0