知象光电

src/main/java/com/fdkankan/contro/controller/SceneFileController.java

@@ -1,9 +1,11 @@
 package com.fdkankan.contro.controller;
 
+import cn.hutool.core.collection.CollUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.fdkankan.common.constant.CommonSuccessStatus;
 import com.fdkankan.common.constant.ErrorCode;
 import com.fdkankan.common.constant.SceneSource;
 import com.fdkankan.common.exception.BusinessException;
@@ -15,6 +17,7 @@ import com.fdkankan.contro.vo.ReportFailLogVO;
 import com.fdkankan.contro.vo.ResponseSceneFile;
 import com.fdkankan.contro.vo.SceneParam;
 import com.fdkankan.contro.vo.SceneUploadCountParamVO;
+import com.fdkankan.fyun.config.FYunFileConfig;
 import com.fdkankan.fyun.face.FYunFileServiceInterface;
 import com.fdkankan.web.response.ResultData;
 import lombok.extern.log4j.Log4j2;
@@ -26,6 +29,7 @@ import javax.annotation.Resource;
 import javax.validation.Valid;
 import java.io.IOException;
 import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * 场景文件上传模块
@@ -201,13 +205,26 @@ public class SceneFileController{
 
     @Autowired
     private ISceneZxgdService sceneZxgdService;
+    @Resource
+    FYunFileConfig fYunFileConfig;
 
     @GetMapping("/getZxgd/{search}")
     public ResultData getZxgd(@PathVariable(value = "search") String search) throws Exception{
         LambdaQueryWrapper<SceneZxgd> wrapper = new LambdaQueryWrapper<>();
         wrapper.like(SceneZxgd::getTitle, search).or().like(SceneZxgd::getUnicode, search);
         List<SceneZxgd> list = sceneZxgdService.list(wrapper);
-        return ResultData.ok(list);
+        if(CollUtil.isEmpty(list)){
+            return ResultData.ok();
+        }
+        List<JSONObject> collect = list.stream().map(v -> {
+            JSONObject jsonObject = new JSONObject();
+            jsonObject.put("名称", v.getTitle());
+            jsonObject.put("unicode", v.getUnicode());
+            jsonObject.put("状态", CommonSuccessStatus.get(v.getStatus()).message());
+            jsonObject.put("url", fYunFileConfig.getHost() + "zxgd/" + v.getUnicode() + "/mesh.ply");
+            return jsonObject;
+        }).collect(Collectors.toList());
+        return ResultData.ok(collect);
     }
 
 }