From c000eb27004fa17b7b41efb8b7e1d0b9e8ec81c2 Mon Sep 17 00:00:00 2001 From: Jerry Yan <792602257@qq.com> Date: Thu, 2 Jan 2025 10:45:25 +0800 Subject: [PATCH] =?UTF-8?q?WVP=5FACTIVE=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/ycwl/basic/device/DeviceFactory.java | 3 + .../wvp_active/WvpActiveStorageConfig.java | 12 ++ .../device/enums/DeviceStoreTypeEnum.java | 1 + .../operator/WvpActiveStorageOperator.java | 106 ++++++++++++++++++ .../StorageNotSupportedException.java | 7 ++ .../tests/TestInitializationSpeed.java | 2 - 6 files changed, 129 insertions(+), 2 deletions(-) create mode 100644 src/main/java/com/ycwl/basic/device/entity/wvp_active/WvpActiveStorageConfig.java create mode 100644 src/main/java/com/ycwl/basic/device/operator/WvpActiveStorageOperator.java create mode 100644 src/main/java/com/ycwl/basic/storage/exceptions/StorageNotSupportedException.java diff --git a/src/main/java/com/ycwl/basic/device/DeviceFactory.java b/src/main/java/com/ycwl/basic/device/DeviceFactory.java index 3bd6213..f30aa1f 100644 --- a/src/main/java/com/ycwl/basic/device/DeviceFactory.java +++ b/src/main/java/com/ycwl/basic/device/DeviceFactory.java @@ -7,6 +7,7 @@ import com.ycwl.basic.device.enums.DeviceStoreTypeEnum; import com.ycwl.basic.device.operator.AliOssStorageOperator; import com.ycwl.basic.device.operator.IDeviceStorageOperator; import com.ycwl.basic.device.operator.LocalStorageOperator; +import com.ycwl.basic.device.operator.WvpActiveStorageOperator; import com.ycwl.basic.model.pc.device.entity.DeviceConfigEntity; import com.ycwl.basic.model.pc.device.entity.DeviceEntity; @@ -15,6 +16,8 @@ public class DeviceFactory { IDeviceStorageOperator operator = null; if (config.getStoreType() == DeviceStoreTypeEnum.ALI_OSS.getType()) { operator = new AliOssStorageOperator(config.getStoreConfigJson()); + } else if (config.getStoreType() == DeviceStoreTypeEnum.WVP_ACTIVE.getType()) { + operator = new WvpActiveStorageOperator(config.getStoreConfigJson()); } else if (config.getStoreType() == DeviceStoreTypeEnum.LOCAL.getType()) { operator = new LocalStorageOperator(config.getStoreConfigJson()); } diff --git a/src/main/java/com/ycwl/basic/device/entity/wvp_active/WvpActiveStorageConfig.java b/src/main/java/com/ycwl/basic/device/entity/wvp_active/WvpActiveStorageConfig.java new file mode 100644 index 0000000..6324f4d --- /dev/null +++ b/src/main/java/com/ycwl/basic/device/entity/wvp_active/WvpActiveStorageConfig.java @@ -0,0 +1,12 @@ +package com.ycwl.basic.device.entity.wvp_active; + +import lombok.Data; + +@Data +public class WvpActiveStorageConfig { + private String host; + private String loginName; + private String loginPassword; + private String app = "rtp"; + private String stream; +} diff --git a/src/main/java/com/ycwl/basic/device/enums/DeviceStoreTypeEnum.java b/src/main/java/com/ycwl/basic/device/enums/DeviceStoreTypeEnum.java index 5d24254..369d3ba 100644 --- a/src/main/java/com/ycwl/basic/device/enums/DeviceStoreTypeEnum.java +++ b/src/main/java/com/ycwl/basic/device/enums/DeviceStoreTypeEnum.java @@ -7,6 +7,7 @@ import lombok.Getter; @AllArgsConstructor public enum DeviceStoreTypeEnum { ALI_OSS(1, "阿里云OSS"), + WVP_ACTIVE(3, "WVP主动模式"), LOCAL(2, "本地文件"); private final int type; diff --git a/src/main/java/com/ycwl/basic/device/operator/WvpActiveStorageOperator.java b/src/main/java/com/ycwl/basic/device/operator/WvpActiveStorageOperator.java new file mode 100644 index 0000000..bc0b420 --- /dev/null +++ b/src/main/java/com/ycwl/basic/device/operator/WvpActiveStorageOperator.java @@ -0,0 +1,106 @@ +package com.ycwl.basic.device.operator; + +import cn.hutool.crypto.digest.MD5; +import cn.hutool.http.HttpRequest; +import cn.hutool.http.HttpUtil; +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.JSONObject; +import com.ycwl.basic.device.entity.common.FileObject; +import com.ycwl.basic.device.entity.wvp_active.WvpActiveStorageConfig; +import com.ycwl.basic.storage.exceptions.StorageConfigException; +import com.ycwl.basic.storage.exceptions.StorageUnsupportedException; + +import java.text.SimpleDateFormat; +import java.util.Collections; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + +public class WvpActiveStorageOperator extends ADeviceStorageOperator { + + public WvpActiveStorageOperator(String configJson) { + loadConfig(configJson); + } + private WvpActiveStorageConfig config; + + @Override + public void loadConfig(String configJson) { + this.config = JSON.parseObject(configJson, WvpActiveStorageConfig.class); + } + + @Override + public List getFileListByDtRange(Date startDate, Date endDate) { + return listDirByDtRange(startDate, endDate); + } + + @Override + public boolean removeFilesBeforeDate(Date date) { + return true; + } + + + private String token; + private String getToken() { + if (this.token != null) { + return this.token; + } + String url = this.config.getHost() + "/api/user/login"; + Map params = new HashMap<>(); + params.put("username", this.config.getLoginName()); + String password = MD5.create().digestHex(this.config.getLoginPassword()); + params.put("password", password); + String jsonResult = HttpUtil.get(url, params); + JSONObject result = JSON.parseObject(jsonResult); + if (result.getInteger("code") == 0) { + JSONObject data = result.getJSONObject("data"); + this.token = data.getString("accessToken"); + return this.token; + } else { + throw new StorageConfigException("获取token失败，原因为：" + result.getString("msg")); + } + } + public List listDirByDtRange(Date startDate, Date endDate) { + SimpleDateFormat normalDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); + String url = this.config.getHost() + "/api/cloud/record/list"; + Map params = new HashMap<>(); + params.put("app", this.config.getApp()); + params.put("stream", this.config.getStream()); + params.put("startTime", normalDateFormat.format(startDate)); + params.put("endTime", normalDateFormat.format(endDate)); + params.put("page", 1); + params.put("count", 100); + String jsonResult = HttpRequest.get(url).form(params).header("Access-Token", getToken()).execute().body(); + JSONObject result = JSON.parseObject(jsonResult); + if (result.getInteger("code") == 0) { + JSONObject data = result.getJSONObject("data"); + List recordList = data.getJSONArray("list").toJavaList(JSONObject.class); + return recordList.stream().map(record -> { + FileObject object = new FileObject(); + object.setName(record.getString("id")); + object.setPath(record.getString("folder")); + object.setUrl(getUrlForDownload(record.getInteger("id"))); + object.setNeedDownload(true); + object.setCreateTime(new Date(record.getLongValue("startTime"))); + object.setEndTime(new Date(record.getLongValue("endTime"))); + return object; + }).collect(Collectors.toList()); + } + return Collections.emptyList(); + } + + public String getUrlForDownload(int id) { + // assume path is recordId + String url = this.config.getHost() + "/api/cloud/record/play/path?recordId=" + id; + String jsonResult = HttpRequest.get(url).header("Access-Token", getToken()).execute().body(); + JSONObject result = JSON.parseObject(jsonResult); + if (result.getInteger("code") == 0) { + JSONObject data = result.getJSONObject("data"); + return data.getString("httpPath"); + } else { + throw new StorageUnsupportedException("获取播放地址失败，原因为：" + result.getString("msg")); + } + } + +} diff --git a/src/main/java/com/ycwl/basic/storage/exceptions/StorageNotSupportedException.java b/src/main/java/com/ycwl/basic/storage/exceptions/StorageNotSupportedException.java new file mode 100644 index 0000000..dc7aa11 --- /dev/null +++ b/src/main/java/com/ycwl/basic/storage/exceptions/StorageNotSupportedException.java @@ -0,0 +1,7 @@ +package com.ycwl.basic.storage.exceptions; + +public class StorageNotSupportedException extends StorageException{ + public StorageNotSupportedException(String message) { + super(message); + } +} diff --git a/src/main/java/com/ycwl/basic/storage/tests/TestInitializationSpeed.java b/src/main/java/com/ycwl/basic/storage/tests/TestInitializationSpeed.java index 35e7c50..b7721be 100644 --- a/src/main/java/com/ycwl/basic/storage/tests/TestInitializationSpeed.java +++ b/src/main/java/com/ycwl/basic/storage/tests/TestInitializationSpeed.java @@ -1,8 +1,6 @@ package com.ycwl.basic.storage.tests; import com.ycwl.basic.storage.StorageFactory; -import com.ycwl.basic.storage.adapters.AliOssAdapter; -import com.ycwl.basic.storage.adapters.IStorageAdapter; import com.ycwl.basic.storage.enums.StorageType; public class TestInitializationSpeed {