BCE临时使用

src/main/java/com/ycwl/basic/controller/extern/AioDeviceController.java

@@ -1,6 +1,9 @@
 package com.ycwl.basic.controller.extern;
 
+import cn.hutool.http.HttpUtil;
 import com.ycwl.basic.annotation.IgnoreToken;
+import com.ycwl.basic.image.enhancer.adapter.BceImageEnhancer;
+import com.ycwl.basic.image.enhancer.entity.BceEnhancerConfig;
 import com.ycwl.basic.mapper.AioDeviceMapper;
 import com.ycwl.basic.mapper.MemberMapper;
 import com.ycwl.basic.model.aio.entity.AioDeviceBannerEntity;
@@ -21,13 +24,19 @@ import com.ycwl.basic.service.aio.AioDeviceService;
 import com.ycwl.basic.service.mobile.GoodsService;
 import com.ycwl.basic.service.pc.FaceService;
 import com.ycwl.basic.service.pc.OrderService;
+import com.ycwl.basic.service.pc.ScenicService;
+import com.ycwl.basic.service.pc.SourceService;
+import com.ycwl.basic.storage.adapters.IStorageAdapter;
 import com.ycwl.basic.utils.ApiResponse;
 import com.ycwl.basic.utils.JwtTokenUtil;
 import com.ycwl.basic.utils.SnowFlakeUtil;
 import jakarta.servlet.ServletRequest;
 import jakarta.servlet.http.HttpServletRequest;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.commons.lang3.Strings;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -37,8 +46,10 @@ import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.io.File;
 import java.util.Date;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 @Slf4j
 @IgnoreToken
@@ -57,6 +68,10 @@ public class AioDeviceController {
     private AioDeviceService aioDeviceService;
     @Autowired
     private OrderService orderService;
+    @Autowired
+    private RedisTemplate<String, String> redisTemplate;
+    @Autowired
+    private SourceService sourceService;
 
     @GetMapping("/info")
     public ApiResponse<AioDeviceInfoResp> getDeviceInfo(HttpServletRequest request) {
@@ -105,6 +120,47 @@ public class AioDeviceController {
         memberEntity.setNickname("用户");
         memberMapper.add(memberEntity);
         FaceRecognizeResp resp = faceService.faceUpload(file, aioDevice.getScenicId(), memberEntity.getId());
+        // 尝试超分
+        new Thread(() -> {
+            try {
+                Thread.sleep(2000L);
+            } catch (InterruptedException e) {
+                return;
+            }
+            GoodsReqQuery query = new GoodsReqQuery();
+            query.setSourceType(2);
+            query.setFaceId(resp.getFaceId());
+            List<GoodsDetailVO> sourcePhotoList = goodsService.sourceGoodsList(query);
+            if (sourcePhotoList == null || sourcePhotoList.isEmpty()) {
+                log.info("无源图片");
+                redisTemplate.opsForValue().set("aio:faceId:"+resp.getFaceId().toString()+":pass", "1", 1, TimeUnit.DAYS);
+                return;
+            }
+            log.info("超分开始！");
+            sourcePhotoList.forEach(photo -> {
+                if (StringUtils.contains(photo.getUrl(), "_q_")) {
+                    return;
+                }
+                try {
+                    File dstFile = new File(photo.getGoodsId()+".jpg");
+                    long fileSize = HttpUtil.downloadFile(photo.getUrl(), dstFile);
+                    log.info("超分开始：{}", fileSize);
+                    BceImageEnhancer enhancer = getEnhancer();
+                    MultipartFile enhancedFile = enhancer.enhance(dstFile.getName());
+                    log.info("超分结束：{}", photo.getUrl());
+                    String url = sourceService.uploadAndUpdateUrl(photo.getGoodsId(), enhancedFile);
+                    log.info("上传结束：->{}", url);
+                } catch (Exception e) {
+                    log.error("超分失败：{}", photo.getGoodsId(), e);
+                } finally {
+                    File _file = new File(photo.getGoodsId()+".jpg");
+                    if (_file.exists()) {
+                        _file.delete();
+                    }
+                }
+            });
+            redisTemplate.opsForValue().set("aio:faceId:"+sourcePhotoList.getFirst().getFaceId().toString()+":pass", "1", 1, TimeUnit.DAYS);
+        }).start();
         return ApiResponse.success(resp);
     }
 
@@ -113,6 +169,14 @@ public class AioDeviceController {
     public ApiResponse<FaceRespVO> faceInfo(@PathVariable Long faceId) {
         return faceService.getById(faceId);
     }
+    @GetMapping("/face/{faceId}/check")
+    public ApiResponse<Boolean> faceCheck(@PathVariable Long faceId) {
+        if (redisTemplate.hasKey("aio:faceId:"+faceId.toString()+":pass")) {
+            return ApiResponse.success(true);
+        } else {
+            return ApiResponse.success(false);
+        }
+    }
     // 照片商品列表
     @GetMapping("/{faceId}/photo")
     public ApiResponse<List<GoodsDetailVO>> sourceGoodsList(@PathVariable Long faceId) {
@@ -144,4 +208,15 @@ public class AioDeviceController {
         }
         return ApiResponse.success(orderService.queryOrder(orderId));
     }
+
+    private BceImageEnhancer getEnhancer() {
+        BceImageEnhancer enhancer = new BceImageEnhancer();
+        BceEnhancerConfig config = new BceEnhancerConfig();
+        config.setQps(1);
+        config.setAppId("119554288");
+        config.setApiKey("OX6QoijgKio3eVtA0PiUVf7f");
+        config.setSecretKey("dYatXReVriPeiktTjUblhfubpcmYfuMk");
+        enhancer.setConfig(config);
+        return enhancer;
+    }
 }