From 221f99c764b51c3cd284c6e5a41492c26c5c7ffc Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期五, 09 九月 2022 17:06:19 +0800
Subject: [PATCH] 优化sip消息

---
 src/main/java/com/genersoft/iot/vmp/vmanager/server/ServerController.java |  213 +++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 151 insertions(+), 62 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 4b7adc6..2884734 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
@@ -1,17 +1,28 @@
 package com.genersoft.iot.vmp.vmanager.server;
 
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
 import com.genersoft.iot.vmp.VManageBootstrap;
 import com.genersoft.iot.vmp.common.VersionPo;
+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.WVPResult;
+import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
 import gov.nist.javax.sip.SipStackImpl;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiOperation;
+
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+import org.springframework.util.ObjectUtils;
 import org.springframework.web.bind.annotation.*;
 
 import javax.sip.ListeningPoint;
@@ -21,96 +32,174 @@
 import java.util.List;
 
 @SuppressWarnings("rawtypes")
-@Api(tags = "鏈嶅姟鎺у埗")
+@Tag(name = "鏈嶅姟鎺у埗")
 @CrossOrigin
 @RestController
 @RequestMapping("/api/server")
 public class ServerController {
 
     @Autowired
-    private ConfigurableApplicationContext context;
+    private ZlmHttpHookSubscribe zlmHttpHookSubscribe;
 
     @Autowired
     private IMediaServerService mediaServerService;
 
     @Autowired
-    VersionInfo versionInfo;
+    private VersionInfo versionInfo;
+
+    @Autowired
+    private SipConfig sipConfig;
+
+    @Autowired
+    private UserSetting userSetting;
+
+    @Value("${server.port}")
+    private int serverPort;
 
 
-    @ApiOperation("娴佸獟浣撴湇鍔″垪琛�")
+    @Autowired
+    private ThreadPoolTaskExecutor taskExecutor;
+
+
     @GetMapping(value = "/media_server/list")
     @ResponseBody
-    public WVPResult<List<MediaServerItem>> getMediaServerList(){
-        WVPResult<List<MediaServerItem>> result = new WVPResult<>();
-        result.setCode(0);
-        result.setMsg("success");
-        result.setData(mediaServerService.getAll());
-        return result;
+    @Operation(summary = "娴佸獟浣撴湇鍔″垪琛�")
+    public List<MediaServerItem> getMediaServerList() {
+        return mediaServerService.getAll();
     }
 
-    @ApiOperation("鍦ㄧ嚎娴佸獟浣撴湇鍔″垪琛�")
     @GetMapping(value = "/media_server/online/list")
     @ResponseBody
-    public WVPResult<List<MediaServerItem>> getOnlineMediaServerList(){
-        WVPResult<List<MediaServerItem>> result = new WVPResult<>();
-        result.setCode(0);
-        result.setMsg("success");
-        result.setData(mediaServerService.getAllOnline());
-        return result;
+    @Operation(summary = "鍦ㄧ嚎娴佸獟浣撴湇鍔″垪琛�")
+    public List<MediaServerItem> getOnlineMediaServerList() {
+        return mediaServerService.getAllOnline();
     }
 
-    @ApiOperation("鑾峰彇娴佸獟浣撴湇鍔�")
     @GetMapping(value = "/media_server/one/{id}")
     @ResponseBody
-    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;
+    @Operation(summary = "鍋滄瑙嗛鍥炴斁")
+    @Parameter(name = "id", description = "娴佸獟浣撴湇鍔D", required = true)
+    public MediaServerItem getMediaServer(@PathVariable String id) {
+        return mediaServerService.getOne(id);
     }
 
-    @ApiOperation("閲嶅惎鏈嶅姟")
+    @Operation(summary = "娴嬭瘯娴佸獟浣撴湇鍔�")
+    @Parameter(name = "ip", description = "娴佸獟浣撴湇鍔P", required = true)
+    @Parameter(name = "port", description = "娴佸獟浣撴湇鍔TT绔彛", required = true)
+    @Parameter(name = "secret", description = "娴佸獟浣撴湇鍔ecret", required = true)
+    @GetMapping(value = "/media_server/check")
+    @ResponseBody
+    public MediaServerItem checkMediaServer(@RequestParam String ip, @RequestParam int port, @RequestParam String secret) {
+        return mediaServerService.checkMediaServer(ip, port, secret);
+    }
+
+    @Operation(summary = "娴嬭瘯娴佸獟浣撳綍鍍忕鐞嗘湇鍔�")
+    @Parameter(name = "ip", description = "娴佸獟浣撴湇鍔P", required = true)
+    @Parameter(name = "port", description = "娴佸獟浣撴湇鍔TT绔彛", required = true)
+    @GetMapping(value = "/media_server/record/check")
+    @ResponseBody
+    public void checkMediaRecordServer(@RequestParam String ip, @RequestParam int port) {
+        boolean checkResult = mediaServerService.checkMediaRecordServer(ip, port);
+        if (!checkResult) {
+            throw new ControllerException(ErrorCode.ERROR100.getCode(), "杩炴帴澶辫触");
+        }
+    }
+
+    @Operation(summary = "淇濆瓨娴佸獟浣撴湇鍔�")
+    @Parameter(name = "mediaServerItem", description = "娴佸獟浣撲俊鎭�", required = true)
+    @PostMapping(value = "/media_server/save")
+    @ResponseBody
+    public void saveMediaServer(@RequestBody MediaServerItem mediaServerItem) {
+        MediaServerItem mediaServerItemInDatabase = mediaServerService.getOne(mediaServerItem.getId());
+
+        if (mediaServerItemInDatabase != null) {
+            if (ObjectUtils.isEmpty(mediaServerItemInDatabase.getSendRtpPortRange()) && ObjectUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) {
+                mediaServerItem.setSendRtpPortRange("30000,30500");
+            }
+            mediaServerService.update(mediaServerItem);
+        } else {
+            if (ObjectUtils.isEmpty(mediaServerItem.getSendRtpPortRange())) {
+                mediaServerItem.setSendRtpPortRange("30000,30500");
+            }
+            mediaServerService.add(mediaServerItem);
+        }
+    }
+
+    @Operation(summary = "绉婚櫎娴佸獟浣撴湇鍔�")
+    @Parameter(name = "id", description = "娴佸獟浣揑D", required = true)
+    @DeleteMapping(value = "/media_server/delete")
+    @ResponseBody
+    public void deleteMediaServer(@RequestParam String id) {
+        if (mediaServerService.getOne(id) == null) {
+            throw new ControllerException(ErrorCode.ERROR100.getCode(), "鏈壘鍒版鑺傜偣");
+        }
+        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";
-    }
-
-    @ApiOperation("鐗堟湰淇℃伅")
+    @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 JSONObject getVersion(String type) {
+
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("server.port", serverPort);
+        if (ObjectUtils.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;
+            }
+        }
+        return jsonObject;
+    }
+
+    @GetMapping(value = "/hooks")
+    @ResponseBody
+    @Operation(summary = "鑾峰彇褰撳墠鎵�鏈塰ook")
+    public List<IHookSubscribe> getHooks() {
+        return zlmHttpHookSubscribe.getAll();
     }
 }

--
Gitblit v1.8.0