648540858
3 years ago
7 changed files with 2 additions and 267 deletions
Binary file not shown.
@ -1,13 +0,0 @@ |
|||
package com.genersoft.iot.vmp.onvif; |
|||
|
|||
import be.teletask.onvif.models.OnvifDevice; |
|||
import com.genersoft.iot.vmp.onvif.dto.ONVIFCallBack; |
|||
|
|||
import java.util.List; |
|||
|
|||
public interface IONVIFServer { |
|||
|
|||
void search(int timeout, ONVIFCallBack<List<String>> callBack); |
|||
|
|||
void getRTSPUrl(int timeout, OnvifDevice device, ONVIFCallBack<String> callBack); |
|||
} |
|||
@ -1,5 +0,0 @@ |
|||
package com.genersoft.iot.vmp.onvif.dto; |
|||
|
|||
public interface ONVIFCallBack<T> { |
|||
void run(int errorCode, T t); |
|||
} |
|||
@ -1,115 +0,0 @@ |
|||
package com.genersoft.iot.vmp.onvif.impl; |
|||
|
|||
|
|||
import be.teletask.onvif.DiscoveryManager; |
|||
import be.teletask.onvif.OnvifManager; |
|||
import be.teletask.onvif.listeners.*; |
|||
import be.teletask.onvif.models.*; |
|||
import be.teletask.onvif.responses.OnvifResponse; |
|||
import com.genersoft.iot.vmp.onvif.IONVIFServer; |
|||
import com.genersoft.iot.vmp.onvif.dto.ONVIFCallBack; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
import org.springframework.stereotype.Service; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
@SuppressWarnings("rawtypes") |
|||
/** |
|||
* 处理onvif的各种操作 |
|||
*/ |
|||
@Service |
|||
public class ONVIFServerIMpl implements IONVIFServer { |
|||
|
|||
private final static Logger logger = LoggerFactory.getLogger(ONVIFServerIMpl.class); |
|||
|
|||
@Override |
|||
public void search(int timeout, ONVIFCallBack<List<String>> callBack) { |
|||
DiscoveryManager manager = new DiscoveryManager(); |
|||
manager.setDiscoveryTimeout(timeout); |
|||
Map<String, Device> deviceMap = new HashMap<>(); |
|||
// 搜索设备
|
|||
manager.discover(new DiscoveryListener() { |
|||
@Override |
|||
public void onDiscoveryStarted() { |
|||
logger.info("Discovery started"); |
|||
} |
|||
|
|||
@Override |
|||
public void onDevicesFound(List<Device> devices) { |
|||
if (devices == null || devices.size() == 0) return; |
|||
for (Device device : devices){ |
|||
logger.info(device.getHostName()); |
|||
deviceMap.put(device.getHostName(), device); |
|||
} |
|||
} |
|||
|
|||
// 搜索结束
|
|||
@Override |
|||
public void onDiscoveryFinished() { |
|||
ArrayList<String> result = new ArrayList<>(); |
|||
for (Device device : deviceMap.values()) { |
|||
logger.info(device.getHostName()); |
|||
result.add(device.getHostName()); |
|||
} |
|||
callBack.run(0, result); |
|||
} |
|||
}); |
|||
} |
|||
|
|||
@Override |
|||
public void getRTSPUrl(int timeout, OnvifDevice device, ONVIFCallBack<String> callBack) { |
|||
if (device.getHostName() == null ){ |
|||
callBack.run(400, null); |
|||
} |
|||
OnvifManager onvifManager = new OnvifManager(); |
|||
onvifManager.setOnvifResponseListener(new OnvifResponseListener(){ |
|||
|
|||
@Override |
|||
public void onResponse(OnvifDevice onvifDevice, OnvifResponse response) { |
|||
logger.info("[RESPONSE] " + onvifDevice.getHostName() |
|||
+ "======" + response.getErrorCode() |
|||
+ "======" + response.getErrorMessage()); |
|||
} |
|||
|
|||
@Override |
|||
public void onError(OnvifDevice onvifDevice, int errorCode, String errorMessage) { |
|||
logger.info("[ERROR] " + onvifDevice.getHostName() + "======" + errorCode + "=======" + errorMessage); |
|||
callBack.run(errorCode, errorMessage); |
|||
} |
|||
}); |
|||
|
|||
try { |
|||
onvifManager.getServices(device, (OnvifDevice onvifDevice, OnvifServices services) -> { |
|||
if (services.getProfilesPath().equals("/onvif/Media")) { |
|||
onvifDevice.setPath(services); |
|||
onvifManager.getMediaProfiles(onvifDevice, new OnvifMediaProfilesListener() { |
|||
@Override |
|||
public void onMediaProfilesReceived(OnvifDevice device, List<OnvifMediaProfile> mediaProfiles) { |
|||
for (OnvifMediaProfile mediaProfile : mediaProfiles) { |
|||
logger.info(mediaProfile.getName()); |
|||
logger.info(mediaProfile.getToken()); |
|||
if (mediaProfile.getName().equals("mainStream")) { |
|||
onvifManager.getMediaStreamURI(device, mediaProfile, (OnvifDevice onvifDevice, |
|||
OnvifMediaProfile profile, String uri) -> { |
|||
|
|||
uri = uri.replace("rtsp://", "rtsp://"+ device.getUsername() + ":"+ device.getPassword() + "@"); |
|||
logger.info(onvifDevice.getHostName() + "的地址" + uri); |
|||
callBack.run(0, uri); |
|||
}); |
|||
} |
|||
} |
|||
} |
|||
}); |
|||
} |
|||
}); |
|||
}catch (Exception e) { |
|||
callBack.run(400, e.getMessage()); |
|||
} |
|||
|
|||
|
|||
} |
|||
} |
|||
@ -1,123 +0,0 @@ |
|||
package com.genersoft.iot.vmp.vmanager.onvif; |
|||
|
|||
import be.teletask.onvif.models.OnvifDevice; |
|||
import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder; |
|||
import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage; |
|||
import com.genersoft.iot.vmp.onvif.IONVIFServer; |
|||
import com.genersoft.iot.vmp.vmanager.bean.WVPResult; |
|||
import io.swagger.annotations.Api; |
|||
import io.swagger.annotations.ApiImplicitParam; |
|||
import io.swagger.annotations.ApiImplicitParams; |
|||
import io.swagger.annotations.ApiOperation; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.http.ResponseEntity; |
|||
import org.springframework.web.bind.annotation.*; |
|||
import org.springframework.web.context.request.async.DeferredResult; |
|||
|
|||
import java.util.List; |
|||
import java.util.UUID; |
|||
|
|||
@SuppressWarnings(value = {"rawtypes", "unchecked"}) |
|||
@Api(tags = "onvif设备") |
|||
@CrossOrigin |
|||
@RestController |
|||
@RequestMapping("/api/onvif") |
|||
public class ONVIFController { |
|||
|
|||
|
|||
@Autowired |
|||
private DeferredResultHolder resultHolder; |
|||
|
|||
@Autowired |
|||
private IONVIFServer onvifServer; |
|||
|
|||
|
|||
@ApiOperation("搜索") |
|||
@ApiImplicitParams({ |
|||
@ApiImplicitParam(name="timeout", value = "超时时间", required = true, dataTypeClass = Integer.class), |
|||
}) |
|||
@GetMapping(value = "/search") |
|||
@ResponseBody |
|||
public DeferredResult<ResponseEntity<WVPResult>> search(@RequestParam(required = false)Integer timeout){ |
|||
DeferredResult<ResponseEntity<WVPResult>> result = new DeferredResult<>(timeout + 10L); |
|||
String uuid = UUID.randomUUID().toString(); |
|||
result.onTimeout(()->{ |
|||
RequestMessage msg = new RequestMessage(); |
|||
msg.setKey(DeferredResultHolder.CALLBACK_ONVIF ); |
|||
msg.setId(uuid); |
|||
WVPResult<String> wvpResult = new WVPResult(); |
|||
wvpResult.setCode(0); |
|||
wvpResult.setMsg("搜索超时"); |
|||
msg.setData(wvpResult); |
|||
resultHolder.invokeResult(msg); |
|||
}); |
|||
resultHolder.put(DeferredResultHolder.CALLBACK_ONVIF, uuid, result); |
|||
|
|||
onvifServer.search(timeout, (errorCode, onvifDevices) ->{ |
|||
RequestMessage msg = new RequestMessage(); |
|||
msg.setId(DeferredResultHolder.CALLBACK_ONVIF + uuid); |
|||
WVPResult<List<String>> resultData = new WVPResult(); |
|||
resultData.setCode(errorCode); |
|||
if (errorCode == 0) { |
|||
resultData.setMsg("success"); |
|||
resultData.setData(onvifDevices); |
|||
}else { |
|||
resultData.setMsg("fail"); |
|||
} |
|||
msg.setData(resultData); |
|||
msg.setData(resultData); |
|||
resultHolder.invokeResult(msg); |
|||
}); |
|||
|
|||
return result; |
|||
} |
|||
|
|||
@ApiOperation("获取onvif的rtsp地址") |
|||
@ApiImplicitParams({ |
|||
@ApiImplicitParam(name="timeout", value = "超时时间", required = true, dataTypeClass = Integer.class), |
|||
@ApiImplicitParam(name="hostname", value = "onvif地址", required = true, dataTypeClass = String.class), |
|||
@ApiImplicitParam(name="username", value = "用户名", required = true, dataTypeClass = String.class), |
|||
@ApiImplicitParam(name="password", value = "密码", required = true, dataTypeClass = String.class), |
|||
}) |
|||
@GetMapping(value = "/rtsp") |
|||
@ResponseBody |
|||
public DeferredResult<ResponseEntity<WVPResult>> getRTSPUrl(@RequestParam(value="timeout", required=false, defaultValue="3000") Integer timeout, |
|||
@RequestParam(required = true) String hostname, |
|||
@RequestParam(required = false) String username, |
|||
@RequestParam(required = false) String password |
|||
){ |
|||
|
|||
DeferredResult<ResponseEntity<WVPResult>> result = new DeferredResult<>(timeout + 10L); |
|||
String uuid = UUID.randomUUID().toString(); |
|||
result.onTimeout(()->{ |
|||
RequestMessage msg = new RequestMessage(); |
|||
msg.setId(uuid); |
|||
msg.setKey(DeferredResultHolder.CALLBACK_ONVIF); |
|||
WVPResult<String> wvpResult = new WVPResult(); |
|||
wvpResult.setCode(0); |
|||
wvpResult.setMsg("获取onvif的rtsp地址超时"); |
|||
msg.setData(wvpResult); |
|||
resultHolder.invokeResult(msg); |
|||
}); |
|||
resultHolder.put(DeferredResultHolder.CALLBACK_ONVIF, uuid, result); |
|||
OnvifDevice onvifDevice = new OnvifDevice(hostname, username, password); |
|||
onvifServer.getRTSPUrl(timeout, onvifDevice, (errorCode, url) ->{ |
|||
RequestMessage msg = new RequestMessage(); |
|||
msg.setId(DeferredResultHolder.CALLBACK_ONVIF + uuid); |
|||
WVPResult<String> resultData = new WVPResult(); |
|||
resultData.setCode(errorCode); |
|||
if (errorCode == 0) { |
|||
resultData.setMsg("success"); |
|||
resultData.setData(url); |
|||
}else { |
|||
resultData.setMsg(url); |
|||
} |
|||
msg.setData(resultData); |
|||
|
|||
resultHolder.invokeResult(msg); |
|||
}); |
|||
|
|||
return result; |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue