🔖 版本更新

2025-04-19 05:34:52 +00:00 · 2019-12-01 21:19:36 +08:00
parent 8f17423e1b
commit 9a7a6fb165
1375 changed files with 2429 additions and 367598 deletions
--- a/src/main/java/im/zhaojun/tencent/TencentServiceImpl.java
+++ b/src/main/java/im/zhaojun/tencent/TencentServiceImpl.java
@@ -0,0 +1,128 @@
+package im.zhaojun.tencent;
+
+import cn.hutool.core.util.URLUtil;
+import com.qcloud.cos.COSClient;
+import com.qcloud.cos.ClientConfig;
+import com.qcloud.cos.auth.BasicCOSCredentials;
+import com.qcloud.cos.auth.COSCredentials;
+import com.qcloud.cos.model.COSObjectSummary;
+import com.qcloud.cos.model.ListObjectsRequest;
+import com.qcloud.cos.model.ObjectListing;
+import com.qcloud.cos.region.Region;
+import im.zhaojun.common.model.enums.FileTypeEnum;
+import im.zhaojun.common.model.enums.StorageTypeEnum;
+import im.zhaojun.common.model.dto.FileItemDTO;
+import im.zhaojun.common.model.StorageConfig;
+import im.zhaojun.common.service.FileService;
+import im.zhaojun.common.service.StorageConfigService;
+import im.zhaojun.common.util.StringUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+import java.util.Map;
+
+@Service
+public class TencentServiceImpl implements FileService {
+
+    private static final Logger log = LoggerFactory.getLogger(TencentServiceImpl.class);
+
+    @Resource
+    private StorageConfigService storageConfigService;
+
+    private static final String BUCKET_NAME_KEY = "bucket-name";
+
+    private static final String SECRET_ID_KEY = "secretId";
+
+    private static final String SECRET_KEY = "secretKey";
+
+    private static final String DOMAIN_KEY = "domain";
+
+    private static final String ENDPOINT_KEY = "endPoint";
+
+    @Value("${zfile.cache.timeout}")
+    private Long timeout;
+
+    private String bucketName;
+
+    private String domain;
+
+    private COSClient cosClient;
+
+    private boolean isInitialized;
+
+    @Override
+    public void init() {
+        try {
+            Map<String, StorageConfig> stringStorageConfigMap =
+                    storageConfigService.selectStorageConfigMapByKey(StorageTypeEnum.TENCENT);
+            String secretId = stringStorageConfigMap.get(SECRET_ID_KEY).getValue();
+            String secretKey = stringStorageConfigMap.get(SECRET_KEY).getValue();
+            String regionName = stringStorageConfigMap.get(ENDPOINT_KEY).getValue();
+            bucketName = stringStorageConfigMap.get(BUCKET_NAME_KEY).getValue();
+            domain = stringStorageConfigMap.get(DOMAIN_KEY).getValue();
+
+            COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
+            Region region = new Region(regionName);
+            ClientConfig clientConfig = new ClientConfig(region);
+            cosClient = new COSClient(cred, clientConfig);
+            isInitialized = true;
+        } catch (Exception e) {
+            log.debug(StorageTypeEnum.TENCENT.getDescription() + "初始化异常, 已跳过");
+        }
+    }
+
+    @Override
+    public List<FileItemDTO> fileList(String path) {
+        path = StringUtils.removeFirstSeparator(path);
+
+        List<FileItemDTO> fileItemList = new ArrayList<>();
+        ObjectListing objectListing = cosClient.listObjects(new ListObjectsRequest().withBucketName(bucketName).withDelimiter("/").withPrefix(path));
+        for (COSObjectSummary s : objectListing.getObjectSummaries()) {
+            FileItemDTO fileItemDTO = new FileItemDTO();
+            fileItemDTO.setName(s.getKey().substring(path.length()));
+            fileItemDTO.setSize(s.getSize());
+            fileItemDTO.setTime(s.getLastModified());
+            fileItemDTO.setType(FileTypeEnum.FILE);
+            fileItemDTO.setPath(path);
+            fileItemDTO.setUrl(getDownloadUrl(StringUtils.concatUrl(path, fileItemDTO.getName())));
+            fileItemList.add(fileItemDTO);
+        }
+
+        for (String commonPrefix : objectListing.getCommonPrefixes()) {
+            FileItemDTO fileItemDTO = new FileItemDTO();
+            fileItemDTO.setName(commonPrefix.substring(path.length(), commonPrefix.length() - 1));
+            fileItemDTO.setType(FileTypeEnum.FOLDER);
+            fileItemDTO.setPath(path);
+            fileItemList.add(fileItemDTO);
+        }
+
+        return fileItemList;
+    }
+
+    @Override
+    public String getDownloadUrl(String path) {
+        Date expirationDate = new Date(System.currentTimeMillis() + timeout * 1000);
+        URL url = cosClient.generatePresignedUrl(bucketName, path, expirationDate);
+        return URLUtil.complateUrl(domain, url.getFile());
+    }
+
+    @Override
+    public StorageTypeEnum getStorageTypeEnum() {
+        return StorageTypeEnum.TENCENT;
+    }
+
+
+    @Override
+    public boolean getIsInitialized() {
+        return isInitialized;
+    }
+
+
+}