package com.fdkankan.download.service.impl;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.ConcurrentHashSet;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.date.TimeInterval;
import cn.hutool.core.exceptions.ExceptionUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.fdkankan.common.constant.*;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.download.bean.*;
import com.fdkankan.download.entity.*;
import com.fdkankan.download.service.*;
import com.fdkankan.fyun.constant.FYunTypeEnum;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.model.constants.UploadFilePath;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.redis.util.RedisUtil;
import com.google.common.collect.Lists;
import java.io.File;
import java.io.FileInputStream;
import java.math.BigDecimal;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import lombok.var;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
/**
*
* TODO
*
*
* @author dengsixing
* @since 2022/2/22
**/
@RefreshScope
@Slf4j
@Service
public class SceneDownloadHandlerServiceImpl {
private static final String[] prefixArr = new String[]{
UploadFilePath.DATA_VIEW_PATH,
UploadFilePath.VOICE_VIEW_PATH,
UploadFilePath.VIDEOS_VIEW_PATH,
UploadFilePath.IMG_VIEW_PATH,
UploadFilePath.USER_VIEW_PATH,
};
private static final String[] prefixArr4v3 = new String[]{
"data/data%s/", "images/images%s/", "voice/voice%s/", "video/video%s/"
};
private static final List imageTypes = Lists.newArrayList();
static{
imageTypes.add(ImageType.builder().name("4k_face").size("4096").ranges(new String[]{"0", "511", "1023", "1535", "2047","2559","3071","3583"}).build());
imageTypes.add(ImageType.builder().name("2k_face").size("2048").ranges(new String[]{"0", "511", "1023", "1535"}).build());
imageTypes.add(ImageType.builder().name("1k_face").size("1024").ranges(new String[]{"0", "511"}).build());
imageTypes.add(ImageType.builder().name("512_face").size("512").ranges(new String[]{"0"}).build());
}
@Value("${url.v3.getInfo}")
private String v3GetInfoUrl;
@Value("${path.v4school}")
private String v4localPath;
@Value("${path.v3school}")
private String v3localPath;
@Value("${path.zip-local}")
private String zipLocalFormat;
@Value("${path.source-local}")
private String sourceLocal;
@Value("${path.zip-oss}")
private String zipOssFormat;
@Value("${path.zip-root}")
private String wwwroot;
@Value("${zip.nThreads}")
private int zipNthreads;
@Value("${fyun.bucket:4dkankan}")
private String bucket;
@Value("${fyun.type}")
private String uploadType;
@Value("${download.config.resource-url}")
private String resourceUrl;
@Value("${download.config.public-url}")
private String publicUrl;
@Value("${download.config.exe-name}")
private String exeName;
@Value("${download.config.exe-content}")
private String exeContent;
@Value("${download.config.exe-content-v3}")
private String exeContentV3;
@Autowired
private RedisUtil redisUtil;
@Resource
private FYunFileServiceInterface fYunFileService;
@Autowired
private IScenePlusService scenePlusService;
@Autowired
private IScenePlusExtService scenePlusExtService;
@Autowired
private ISceneProService sceneProService;
@Async("sceneDownLoadExecutror")
public void download(DownLoadTaskBean downLoadTaskBean){
//场景码
String num = null;
try {
num = downLoadTaskBean.getNum();
log.info("场景下载开始 - num[{}] - threadName[{}]", num, Thread.currentThread().getName());
long startTime = Calendar.getInstance().getTimeInMillis();
//执行场景下载逻辑
this.downloadHandler(downLoadTaskBean);
//耗时
long consumeTime = Calendar.getInstance().getTimeInMillis() - startTime;
log.info("场景下载结束 - num[{}] - threadName[{}] - consumeTime[{}]", num, Thread.currentThread().getName(), consumeTime);
}catch (Exception e){
log.error(ExceptionUtil.stacktraceToString(e));
}finally {
if(StrUtil.isNotEmpty(num)){
//本地正在下载任务出队
CurrentDownloadNumUtil.removeSceneNum(num, "v4");
//删除正在下载任务
redisUtil.lRemove(RedisKey.SCENE_DOWNLOAD_ING, 1, num);
}
}
}
@Async("sceneDownLoadExecutror")
public void downloadV3(DownLoadTaskBean downLoadTaskBean){
//场景码
String num = null;
try {
num = downLoadTaskBean.getSceneNum();
log.info("v3场景下载开始 - num[{}] - threadName[{}]", num, Thread.currentThread().getName());
long startTime = Calendar.getInstance().getTimeInMillis();
//执行场景下载逻辑
this.downloadHandlerV3(downLoadTaskBean);
//耗时
long consumeTime = Calendar.getInstance().getTimeInMillis() - startTime;
log.info("v3场景下载结束 - num[{}] - threadName[{}] - consumeTime[{}]", num, Thread.currentThread().getName(), consumeTime);
}catch (Exception e){
log.error(ExceptionUtil.stacktraceToString(e));
}finally {
if(StrUtil.isNotEmpty(num)){
//本地正在下载任务出队
CurrentDownloadNumUtil.removeSceneNum(num, "v3");
//删除正在下载任务
redisUtil.lRemove(RedisKey.SCENE_V3_DOWNLOAD_ING, 1, num);
}
}
}
public void downloadHandler(DownLoadTaskBean downLoadTaskBean) throws Exception{
String num = downLoadTaskBean.getNum();
//zip包路径
String zipPath = null;
try {
TimeInterval timer = DateUtil.timer();
//删除资源目录
FileUtil.del(String.format(this.sourceLocal, num, ""));
ScenePlus scenePlus = scenePlusService.getByNum(num);
if(Objects.isNull(scenePlus))
throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
ScenePlusExt scenePlusExt = scenePlusExtService.getByScenePlusId(scenePlus.getId());
String bucket = scenePlusExt.getYunFileBucket();
Set cacheKeys = new ConcurrentHashSet<>();
Map> allFiles = this.getAllFiles(num, v4localPath, bucket);
List ossFilePaths = allFiles.get("ossFilePaths");
List v4localFilePaths = allFiles.get("localFilePaths");
//key总个数
int total = ossFilePaths.size() + v4localFilePaths.size();
AtomicInteger count = new AtomicInteger(0);
//定义压缩包
zipPath = String.format(this.zipLocalFormat, num);
File zipFile = new File(zipPath);
if(!zipFile.getParentFile().exists()){
zipFile.getParentFile().mkdirs();
}
SceneViewInfo sceneViewInfo = this.getSceneJson(num);
String resolution = sceneViewInfo.getSceneResolution();
//国际版存在已经切好图的情况,下载时不需要再切图,只需要把文件直接下载下来打包就可以了
if(SceneKind.FACE.code().equals(sceneViewInfo.getSceneKind())){
resolution = "notNeadCut";
}
int imagesVersion = -1;
Integer version = sceneViewInfo.getVersion();
if(Objects.nonNull(version)){
imagesVersion = version;
}
//固定文件写入
this.zipLocalFiles(v4localFilePaths, v4localPath, num, count, total, "v4");
log.info("打包固定文件耗时, num:{}, time:{}", num, timer.intervalRestart());
//oss文件写入
this.zipOssFiles(ossFilePaths, num, count, total, resolution, imagesVersion, cacheKeys, "v4");
log.info("打包oss文件耗时, num:{}, time:{}", num, timer.intervalRestart());
this.downloadDbAndRedisData(num);
//重新写入scene.json(去掉密码访问设置)
this.zipSceneJson(num, sceneViewInfo);
//写入启动命令
this.zipBat(num, "v4");
//打压缩包
ZipUtil.zip(String.format(this.sourceLocal, num, ""), zipPath);
//上传压缩包
String uploadPath = String.format(this.zipOssFormat, num);
fYunFileService.uploadFileByCommand(bucket, zipPath, uploadPath);
//更新进度100
String url = this.publicUrl + uploadPath + "?t=" + Calendar.getInstance().getTimeInMillis();
this.updateProgress(null, num, SceneDownloadProgressStatus.DOWNLOAD_SUCCESS.code(), url, "v4");
}catch (Exception e){
//更新进度为下载失败
this.updateProgress( null, num, SceneDownloadProgressStatus.DOWNLOAD_FAILED.code(), null, "v4");
throw e;
}finally {
FileUtil.del(zipPath);
FileUtil.del(String.format(this.sourceLocal, num, ""));
}
}
private SceneViewInfo getSceneJson(String num){
String sceneJsonData = redisUtil.get(String.format(RedisKey.SCENE_JSON, num));
if(StrUtil.isEmpty(sceneJsonData)){
sceneJsonData = fYunFileService.getFileContent(bucket, String.format(UploadFilePath.DATA_VIEW_PATH, num) + "scene.json");
}
sceneJsonData = sceneJsonData.replace(this.publicUrl, "");
SceneViewInfo sceneInfoVO = JSON.parseObject(sceneJsonData, SceneViewInfo.class);
sceneInfoVO.setScenePassword(null);
if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){
sceneInfoVO.setFloorPlanAngle(0f);
}
if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){
sceneInfoVO.setFloorPlanCompass(0f);
}
SceneEditControlsVO controls = sceneInfoVO.getControls();
if(Objects.isNull(controls.getShowShare())){
controls.setShowShare(CommonStatus.YES.code().intValue());
}
if(Objects.isNull(controls.getShowCapture())){
controls.setShowCapture(CommonStatus.YES.code().intValue());
}
if(Objects.isNull(controls.getShowBillboardTitle())){
controls.setShowBillboardTitle(CommonStatus.YES.code().intValue());
}
if(Objects.isNull(controls.getShowDrawTitle())){
controls.setShowDrawTitle(CommonStatus.YES.code().intValue());
}
return sceneInfoVO;
}
public void downloadHandlerV3(DownLoadTaskBean downLoadTaskBean) throws Exception{
String num = downLoadTaskBean.getSceneNum();
//zip包路径
String zipPath = null;
try {
TimeInterval timer = DateUtil.timer();
//删除资源目录
FileUtil.del(String.format(this.sourceLocal, num, ""));
ScenePro scenePro = sceneProService.getByNum(num);
if(Objects.isNull(scenePro))
throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
Set cacheKeys = new ConcurrentHashSet<>();
Map> allFiles = this.getAllFilesV3(num, v3localPath, bucket);
List ossFilePaths = allFiles.get("ossFilePaths");
List v3localFilePaths = allFiles.get("localFilePaths");
//key总个数
int total = ossFilePaths.size() + v3localFilePaths.size();
AtomicInteger count = new AtomicInteger(0);
//定义压缩包
zipPath = String.format(this.zipLocalFormat, num);
File zipFile = new File(zipPath);
if(!zipFile.getParentFile().exists()){
zipFile.getParentFile().mkdirs();
}
int imagesVersion =0;
String resolution = "2k";
JSONObject getInfoJson = this.getInfo(num);
imagesVersion = getInfoJson.getInt("imagesVersion");
// 转台、激光显示4k图片
if(getInfoJson.getInt("sceneSource") == 3 || getInfoJson.getInt("sceneSource") == 4){
resolution = "4k";
}
//固定文件写入
timer.intervalRestart();
this.zipLocalFiles(v3localFilePaths, v3localPath, num, count, total, "v3");
log.info("打包固定文件耗时, num:{}, time:{}", num, timer.intervalRestart());
//oss文件写入
this.zipOssFiles(ossFilePaths, num, count, total, resolution, imagesVersion, cacheKeys, "v3");
log.info("打包oss文件耗时, num:{}, time:{}", num, timer.intervalRestart());
//重新写入scene.json(去掉密码访问设置)
this.zipGetInfoJson(num, getInfoJson);
//写入启动命令
this.zipBat(num, "v3");
//打压缩包
ZipUtil.zip(String.format(this.sourceLocal, num, ""), zipPath);
//上传压缩包
String uploadPath = String.format(this.zipOssFormat, num);
fYunFileService.uploadFileByCommand(bucket, zipPath, uploadPath);
//更新进度100
String url = this.publicUrl + uploadPath + "?t=" + Calendar.getInstance().getTimeInMillis();
this.updateProgress(null, num, SceneDownloadProgressStatus.DOWNLOAD_SUCCESS.code(), url, "v3");
}catch (Exception e){
//更新进度为下载失败
this.updateProgress( null, num, SceneDownloadProgressStatus.DOWNLOAD_FAILED.code(), null, "v3");
throw e;
}finally {
FileUtil.del(zipPath);
FileUtil.del(String.format(this.sourceLocal, num, ""));
}
}
private JSONObject getInfo(String num){
String url = String.format(v3GetInfoUrl, num);
String getInfoResult = HttpUtil.get(url);
JSONObject jsonObject = JSONUtil.parseObj(getInfoResult);
if(Objects.isNull(jsonObject)
|| !ServerCode.SUCCESS.code().equals(jsonObject.getInt("code"))
|| Objects.isNull(jsonObject.getJSONObject("data"))){
throw new RuntimeException("获取getInfo信息失败,url=" + url);
}
JSONObject data = jsonObject.getJSONObject("data");
if (data.getInt("sceneSource") != 2)
{
data.set("sceneScheme", 3);
}
data.set("needKey", 0);
data.set("sceneKey", "");
return data;
}
private void zipOssFiles(List ossFilePaths, String num, AtomicInteger count,
int total, String resolution, int imagesVersion, Set cacheKeys, String version) throws Exception{
if(CollUtil.isEmpty(ossFilePaths)){
return;
}
String imageNumPath = String.format(UploadFilePath.IMG_VIEW_PATH, num);
if("v3".equals(version)){
imageNumPath = String.format("images/images%s/", num);
}
ExecutorService executorService = Executors.newFixedThreadPool(this.zipNthreads);
List futureList = new ArrayList<>();
for (String filePath : ossFilePaths) {
String finalImageNumPath = imageNumPath;
Callable call = new Callable() {
@Override
public Boolean call() throws Exception {
zipOssFilesHandler(num, count, total, resolution,
imagesVersion, cacheKeys,filePath, finalImageNumPath, version);
return true;
}
};
futureList.add(executorService.submit(call));
}
//这里一定要加阻塞,不然会导致oss文件还没打包好,主程序已经结束返回了
Boolean zipSuccess = true;
for (Future future : futureList) {
try {
future.get();
}catch (Exception e){
log.error("打包oss文件失败", e);
zipSuccess = false;
}
}
executorService.shutdown();
if(!zipSuccess){
throw new Exception("打包oss文件失败");
}
}
private void zipOssFilesHandler(String num,
AtomicInteger count, int total, String resolution,
int imagesVersion, Set cacheKeys,
String filePath, String imageNumPath, String version) throws Exception{
if(filePath.endsWith("/")){
//更新进度
this.updateProgress(new BigDecimal(count.incrementAndGet()).divide(new BigDecimal(total), 6, BigDecimal.ROUND_HALF_UP),
num, SceneDownloadProgressStatus.DOWNLOADING.code(), null, version);
return;
}
//某个目录不需要打包
if(filePath.contains(imageNumPath + "panorama/panorama_edit/"))
return;
//切图
if(!"notNeadCut".equals(resolution)){
if((filePath.contains(imageNumPath + "panorama/") && filePath.contains("tiles/" + resolution))
|| filePath.contains(imageNumPath + "tiles/" + resolution + "/")) {
this.processImage(num, filePath, resolution, imagesVersion, cacheKeys);
//更新进度
this.updateProgress(new BigDecimal(count.incrementAndGet()).divide(new BigDecimal(total), 6, BigDecimal.ROUND_HALF_UP),
num, SceneDownloadProgressStatus.DOWNLOADING.code(), null, version);
return;
}
}
//其他文件打包
this.ProcessFiles(num, filePath, this.wwwroot, cacheKeys);
//更新进度
this.updateProgress(new BigDecimal(count.incrementAndGet()).divide(new BigDecimal(total), 6, BigDecimal.ROUND_HALF_UP),
num, SceneDownloadProgressStatus.DOWNLOADING.code(), null, version);
}
private void zipLocalFiles(List localFilePaths, String v3localPath, String num, AtomicInteger count, int total, String version) throws Exception{
String sourcePath = String.format(this.sourceLocal, num, "");
String localPath = "v4".equals(version) ? this.v4localPath : this.v3localPath;
for (String localFilePath : localFilePaths) {
try (FileInputStream in = new FileInputStream(new File(localFilePath));){
// this.zipInputStream(out, localFilePath.replace(v3localPath, ""), in);
FileUtil.copy(localFilePath, localFilePath.replace(localPath, sourcePath), true);
}catch (Exception e){
throw e;
}
//更新进度
this.updateProgress(
new BigDecimal(count.incrementAndGet()).divide(new BigDecimal(total), 6, BigDecimal.ROUND_HALF_UP),
num, SceneDownloadProgressStatus.DOWNLOAD_COMPRESSING.code(), null, version);
}
//写入code.txt
// this.zipBytes(out, "code.txt", num.getBytes());
FileUtil.writeUtf8String(num, String.format(sourceLocal, num, "code.txt"));
}
@Autowired
private ISceneEditInfoService sceneEditInfoService;
@Autowired
private ISceneEditInfoExtService sceneEditInfoExtService;
@Autowired
private ISceneEditControlsService sceneEditControlsService;
private void downloadDbAndRedisData(String num) throws Exception{
String cachePath = String.format(this.sourceLocal, num, this.wwwroot) + String.format(UploadFilePath.VIEW_PATH, num) + "cache/";
try {
//db t_scene_plus、t_scene_plus_ext、t_scene_edit_info、t_scene_edit_info_ext、t_scene_edit_controls
ScenePlus scenePlus = scenePlusService.getByNum(num);
ScenePlusExt scenePlusExt = scenePlusExtService.getByScenePlusId(scenePlus.getId());
SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());
SceneEditControls sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());
FileUtil.writeUtf8String(JSON.toJSONString(scenePlus), cachePath + "t_scene_plus.txt");
FileUtil.writeUtf8String(JSON.toJSONString(scenePlusExt), cachePath + "t_scene_plus_ext.txt");
FileUtil.writeUtf8String(JSON.toJSONString(sceneEditInfo), cachePath + "t_scene_edit_info.txt");
FileUtil.writeUtf8String(JSON.toJSONString(sceneEditInfoExt), cachePath + "t_scene_edit_info_ext.txt");
FileUtil.writeUtf8String(JSON.toJSONString(sceneEditControls), cachePath + "t_scene_edit_controls.txt");
//redis
Map hotdataStr = redisUtil.hmget(String.format(RedisKey.SCENE_HOT_DATA, num));
if(CollUtil.isNotEmpty(hotdataStr)){
Map hotdata = hotdataStr.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> JSON.parseObject(entry.getValue())));
FileUtil.writeUtf8String(JSON.toJSONString(hotdata), cachePath + "hotdata.json");
}
Set hoticons = redisUtil.sGet(String.format(RedisKey.SCENE_HOT_ICONS, num));
if(CollUtil.isNotEmpty(hoticons)){
FileUtil.writeUtf8String(JSON.toJSONString(hoticons), cachePath + "hoticons.json");
}
// Set hoticons = redisUtil.sGet(String.format(RedisKey.SCENE_HOT_ICONS, num));
// FileUtil.writeUtf8String(JSON.toJSONString(hoticons), cachePath + "hoticons.json");
// FileUtil.writeUtf8String(batContent, String.format(this.sourceLocal, num, exeName));
}catch (Exception e){
log.error("下载缓存数据出错,", e);
}
}
private void zipBat(String num, String version) throws Exception{
String batContent = String.format(this.exeContent, num);
if("v3".equals(version)){
batContent = String.format(this.exeContentV3, num);
}
// this.zipBytes(out, exeName, batContent.getBytes());
FileUtil.writeUtf8String(batContent, String.format(this.sourceLocal, num, exeName));
//更新进度为90%
this.updateProgress(new BigDecimal("0.9").divide(new BigDecimal("0.8"), 6, BigDecimal.ROUND_HALF_UP), num,
SceneDownloadProgressStatus.DOWNLOAD_COMPRESSING.code(), null, version);
}
private Map> getAllFiles(String num, String v4localPath, String bucket) throws Exception{
//列出oss所有文件路径
List ossFilePaths = new ArrayList<>();
for (String prefix : prefixArr) {
prefix = String.format(prefix, num);
List keys = fYunFileService.listRemoteFiles(bucket, prefix);
if(CollUtil.isEmpty(keys)){
continue;
}
if(FYunTypeEnum.AWS.code().equals(this.uploadType)){
keys = keys.stream().filter(key->{
if(key.contains("x-oss-process")){
return false;
}
return true;
}).collect(Collectors.toList());
}
ossFilePaths.addAll(keys);
}
//列出v3local所有文件路径
File file = new File(v4localPath);
List localFilePaths = FileUtils.list(file);
HashMap> map = new HashMap<>();
map.put("ossFilePaths", ossFilePaths);
map.put("localFilePaths", localFilePaths);
return map;
}
private Map> getAllFilesV3(String num, String v3localPath, String bucket) throws Exception{
//列出oss所有文件路径
List ossFilePaths = new ArrayList<>();
for (String prefix : prefixArr4v3) {
prefix = String.format(prefix, num);
List keys = fYunFileService.listRemoteFiles(bucket, prefix);
if(CollUtil.isEmpty(keys)){
continue;
}
if(FYunTypeEnum.AWS.code().equals(this.uploadType)){
keys = keys.stream().filter(key->{
if(key.contains("x-oss-process")){
return false;
}
return true;
}).collect(Collectors.toList());
}
ossFilePaths.addAll(keys);
}
//列出v3local所有文件路径
File file = new File(v3localPath);
List localFilePaths = FileUtils.list(file);
HashMap> map = new HashMap<>();
map.put("ossFilePaths", ossFilePaths);
map.put("localFilePaths", localFilePaths);
return map;
}
private void zipSceneJson(String num, SceneViewInfo sceneViewInfo) throws Exception{
//访问密码置0
SceneEditControlsVO controls = sceneViewInfo.getControls();
controls.setShowLock(CommonStatus.NO.code().intValue());
String sceneJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH, num) + "scene.json";
FileUtil.writeUtf8String(JSON.toJSONString(sceneViewInfo, SerializerFeature.WriteNullStringAsEmpty, SerializerFeature.WriteNullNumberAsZero), String.format(this.sourceLocal, num, this.wwwroot + sceneJsonPath));
}
private void zipGetInfoJson(String num, JSONObject getInfo) throws Exception{
//访问密码置0
String getInfoKey = String.format("data/data%s/", num) + "getInfo.json";
String getInfoStr = getInfo.toString().replace(this.publicUrl, "");
FileUtil.writeUtf8String(getInfoStr, String.format(this.sourceLocal, num, this.wwwroot + getInfoKey));
}
private void processImage(String sceneNum, String key, String resolution, int imagesVersion, Set imgKeys) throws Exception{
if(key.contains("x-oss-process") || key.endsWith("/")){
return;
}
String fileName = key.substring(key.lastIndexOf("/")+1, key.indexOf("."));
String ext = key.substring(key.lastIndexOf("."));
String[] arr = fileName.split("_skybox");
String dir = arr[0];
String num = arr[1];
if(StrUtil.isEmpty(fileName)
|| StrUtil.isEmpty(ext)
|| (".jpg".equals(ext) && ".png".equals(ext))
|| StrUtil.isEmpty(dir)
|| StrUtil.isEmpty(num)){
throw new Exception("本地下载图片资源不符合规则,key:" + key);
}
for (ImageType imageType : imageTypes) {
if(imageType.getName().equals("4k_face") && !"4k".equals(resolution)){
continue;
}
List items = Lists.newArrayList();
String[] ranges = imageType.getRanges();
for(int i = 0; i < ranges.length; i++){
String x = ranges[i];
for(int j = 0; j < ranges.length; j++){
String y = ranges[j];
items.add(
ImageTypeDetail.builder()
.i(String.valueOf(i))
.j(String.valueOf(j))
.x(x)
.y(y)
.build()
);
}
}
for (ImageTypeDetail item : items) {
String par = "?x-oss-process=image/resize,m_lfit,w_" + imageType.getSize() + "/crop,w_512,h_512,x_" + item.getX() + ",y_" + item.getY();
if(FYunTypeEnum.AWS.code().equals(uploadType)){
par += "&imagesVersion="+ imagesVersion;
}
var url = this.resourceUrl + key;
FYunTypeEnum storageType = FYunTypeEnum.get(uploadType);
switch (storageType){
case OSS:
url += par;
break;
case AWS:
url += URLEncoder.encode(par.replace("/", "@"), "UTF-8");
break;
}
var fky = key.split("/" + resolution + "/")[0] + "/" + dir + "/" + imageType.getName() + num + "_" + item.getI() + "_" + item.getJ() + ext;
if(imgKeys.contains(fky)){
continue;
}
imgKeys.add(fky);
// HttpUtil.downloadFile(url, String.format(sourceLocal, sceneNum, this.wwwroot + fky));
this.downloadFile(url, String.format(sourceLocal, sceneNum, this.wwwroot + fky));
}
}
}
public void downloadFile(String url, String path){
File file = new File(path);
if(!file.getParentFile().exists()){
file.getParentFile().mkdirs();
}
HttpUtil.downloadFile(url, path);
}
public void ProcessFiles(String num, String key, String prefix, Set cacheKeys) throws Exception{
if(cacheKeys.contains(key)){
return;
}
if(key.equals(String.format(UploadFilePath.DATA_VIEW_PATH, num) + "scene.json")){
return;
}
cacheKeys.add(key);
String fileName = key.substring(key.lastIndexOf("/") + 1);
String url = this.resourceUrl + key.replace(fileName, URLEncoder.encode(fileName, "UTF-8")) + "?t=" + Calendar.getInstance().getTimeInMillis();
if(key.contains("hot.json") || key.contains("link-scene.json")){
String content = fYunFileService.getFileContent(key);
if(StrUtil.isEmpty(content)){
return;
}
content = content.replace(publicUrl, "")
// .replace(publicUrl+"v3/", "")
.replace("https://spc.html","spc.html")
.replace("https://smobile.html", "smobile.html");
FileUtil.writeUtf8String(content, String.format(sourceLocal, num, prefix + key));
}else{
// HttpUtil.downloadFile(url, String.format(sourceLocal, num, prefix + key));
try {
this.downloadFile(url, String.format(sourceLocal, num, prefix + key));
}catch (Exception e){
log.info("下载文件报错,path:{}", String.format(sourceLocal, num, prefix + key));
}
}
}
public void updateProgress(BigDecimal precent, String num, Integer status, String url, String version){
SceneDownloadProgressStatus progressStatus = SceneDownloadProgressStatus.get(status);
switch (progressStatus){
case DOWNLOAD_SUCCESS:
precent = new BigDecimal("100");
break;
case DOWNLOAD_FAILED:
precent = new BigDecimal("0");
break;
default:
precent = precent.multiply(new BigDecimal("0.8")).multiply(new BigDecimal("100"));
}
DownLoadProgressBean progress = null;
String key = String.format(RedisKey.PREFIX_DOWNLOAD_PROGRESS_V4, num);
if("v3".equals(version)){
key = String.format(RedisKey.PREFIX_DOWNLOAD_PROGRESS, num);
}
String progressStr = redisUtil.get(key);
if(StrUtil.isEmpty(progressStr)){
progress = DownLoadProgressBean.builder().percent(precent.intValue()).status(status).url(url).build();
}else{
progress = JSONUtil.toBean(progressStr, DownLoadProgressBean.class);
//如果下载失败,进度不变
if(status == SceneDownloadProgressStatus.DOWNLOAD_FAILED.code() && progress.getPercent() != null){
precent = new BigDecimal(progress.getPercent());
}
progress.setPercent(precent.intValue());
progress.setStatus(status);
progress.setUrl(url);
}
redisUtil.set(key, JSONUtil.toJsonStr(progress));
}
}