|
@@ -1,15 +1,18 @@
|
|
|
package cn.reghao.tnb.content.app.disk.service;
|
|
package cn.reghao.tnb.content.app.disk.service;
|
|
|
|
|
|
|
|
|
|
+import cn.reghao.file.api.iface.OssService;
|
|
|
import cn.reghao.jutil.jdk.converter.ByteConverter;
|
|
import cn.reghao.jutil.jdk.converter.ByteConverter;
|
|
|
import cn.reghao.jutil.jdk.converter.ByteType;
|
|
import cn.reghao.jutil.jdk.converter.ByteType;
|
|
|
import cn.reghao.jutil.jdk.db.Page;
|
|
import cn.reghao.jutil.jdk.db.Page;
|
|
|
import cn.reghao.jutil.jdk.db.PageList;
|
|
import cn.reghao.jutil.jdk.db.PageList;
|
|
|
import cn.reghao.oss.sdk.model.constant.ObjectType;
|
|
import cn.reghao.oss.sdk.model.constant.ObjectType;
|
|
|
|
|
+import cn.reghao.oss.sdk.model.dto.ObjectInfo;
|
|
|
import cn.reghao.tnb.common.auth.UserContext;
|
|
import cn.reghao.tnb.common.auth.UserContext;
|
|
|
import cn.reghao.tnb.content.app.disk.db.mapper.NetDiskMapper;
|
|
import cn.reghao.tnb.content.app.disk.db.mapper.NetDiskMapper;
|
|
|
import cn.reghao.tnb.content.app.disk.model.po.NetDisk;
|
|
import cn.reghao.tnb.content.app.disk.model.po.NetDisk;
|
|
|
import cn.reghao.tnb.content.app.disk.model.query.DiskQuery;
|
|
import cn.reghao.tnb.content.app.disk.model.query.DiskQuery;
|
|
|
import cn.reghao.tnb.content.app.disk.model.vo.*;
|
|
import cn.reghao.tnb.content.app.disk.model.vo.*;
|
|
|
|
|
+import org.apache.dubbo.config.annotation.DubboReference;
|
|
|
import org.springframework.cache.annotation.Cacheable;
|
|
import org.springframework.cache.annotation.Cacheable;
|
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
|
@@ -25,6 +28,8 @@ import java.util.stream.Collectors;
|
|
|
*/
|
|
*/
|
|
|
@Service
|
|
@Service
|
|
|
public class NetDiskQuery {
|
|
public class NetDiskQuery {
|
|
|
|
|
+ @DubboReference(check = false, timeout = 60_000)
|
|
|
|
|
+ private OssService ossService;
|
|
|
private final NetDiskMapper netDiskMapper;
|
|
private final NetDiskMapper netDiskMapper;
|
|
|
private final ByteConverter byteConverter;
|
|
private final ByteConverter byteConverter;
|
|
|
|
|
|
|
@@ -78,7 +83,6 @@ public class NetDiskQuery {
|
|
|
|
|
|
|
|
public PageList<FileInfo> findByPid(int pageSize, int pageNumber, String pid) {
|
|
public PageList<FileInfo> findByPid(int pageSize, int pageNumber, String pid) {
|
|
|
String[] sortProps = {"fileType", "filename"};
|
|
String[] sortProps = {"fileType", "filename"};
|
|
|
-
|
|
|
|
|
long loginUser = UserContext.getUserId();
|
|
long loginUser = UserContext.getUserId();
|
|
|
DiskQuery diskQuery = new DiskQuery.Builder()
|
|
DiskQuery diskQuery = new DiskQuery.Builder()
|
|
|
.pid(pid)
|
|
.pid(pid)
|
|
@@ -86,6 +90,7 @@ public class NetDiskQuery {
|
|
|
.build();
|
|
.build();
|
|
|
|
|
|
|
|
Page page = new Page(pageNumber, pageSize);
|
|
Page page = new Page(pageNumber, pageSize);
|
|
|
|
|
+ int total = netDiskMapper.countByDiskQuery(diskQuery);
|
|
|
List<NetDisk> netDiskList = netDiskMapper.findDiskQueryByPage(page, diskQuery);
|
|
List<NetDisk> netDiskList = netDiskMapper.findDiskQueryByPage(page, diskQuery);
|
|
|
|
|
|
|
|
List<FileInfo> list = netDiskList.stream().map(netDisk -> {
|
|
List<FileInfo> list = netDiskList.stream().map(netDisk -> {
|
|
@@ -93,7 +98,6 @@ public class NetDiskQuery {
|
|
|
return new FileInfo(netDisk, sizeStr);
|
|
return new FileInfo(netDisk, sizeStr);
|
|
|
}).collect(Collectors.toList());
|
|
}).collect(Collectors.toList());
|
|
|
|
|
|
|
|
- int total = list.size();
|
|
|
|
|
return PageList.pageList(pageNumber, pageSize, total, list);
|
|
return PageList.pageList(pageNumber, pageSize, total, list);
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -161,7 +165,16 @@ public class NetDiskQuery {
|
|
|
return null;
|
|
return null;
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+ int channelCode = 111;
|
|
|
String objectId = netDisk.getFileId();
|
|
String objectId = netDisk.getFileId();
|
|
|
|
|
+ try {
|
|
|
|
|
+ ObjectInfo objectInfo = ossService.getObjectInfo(channelCode, objectId);
|
|
|
|
|
+ String signedUrl = ossService.getSignedUrl(channelCode, objectId);
|
|
|
|
|
+ return new FileUrl(objectInfo, signedUrl);
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
/*DiskFile diskFile = fileService.getDiskFile(objectId);
|
|
/*DiskFile diskFile = fileService.getDiskFile(objectId);
|
|
|
if (diskFile != null) {
|
|
if (diskFile != null) {
|
|
|
return new FileUrl(diskFile);
|
|
return new FileUrl(diskFile);
|