You've already forked FrameTour-BE
227 lines
9.7 KiB
Java
227 lines
9.7 KiB
Java
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;
|
|
import com.ycwl.basic.model.aio.entity.AioDeviceEntity;
|
|
import com.ycwl.basic.model.aio.entity.AioDevicePriceConfigEntity;
|
|
import com.ycwl.basic.model.aio.req.AioDeviceCreateOrderReq;
|
|
import com.ycwl.basic.model.aio.resp.AioDeviceCreateOrderResp;
|
|
import com.ycwl.basic.model.aio.resp.AioDeviceInfoResp;
|
|
import com.ycwl.basic.model.jwt.JwtInfo;
|
|
import com.ycwl.basic.model.mobile.face.FaceRecognizeResp;
|
|
import com.ycwl.basic.model.mobile.goods.GoodsDetailVO;
|
|
import com.ycwl.basic.model.mobile.goods.GoodsReqQuery;
|
|
import com.ycwl.basic.model.pc.face.resp.FaceRespVO;
|
|
import com.ycwl.basic.model.pc.member.entity.MemberEntity;
|
|
import com.ycwl.basic.model.pc.member.resp.MemberRespVO;
|
|
import com.ycwl.basic.pay.entity.PayResponse;
|
|
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;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
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
|
|
@RestController
|
|
@RequestMapping("/api/aio")
|
|
public class AioDeviceController {
|
|
@Autowired
|
|
private GoodsService goodsService;
|
|
@Autowired
|
|
private FaceService faceService;
|
|
@Autowired
|
|
private MemberMapper memberMapper;
|
|
@Autowired
|
|
private AioDeviceMapper aioDeviceMapper;
|
|
@Autowired
|
|
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) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
List<AioDeviceBannerEntity> banners = aioDeviceMapper.getBannerByDeviceId(aioDevice.getId());
|
|
return ApiResponse.success(new AioDeviceInfoResp(aioDevice, banners));
|
|
}
|
|
|
|
@GetMapping("/banners")
|
|
public ApiResponse<List<AioDeviceBannerEntity>> getBanners(HttpServletRequest request) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
List<AioDeviceBannerEntity> banners = aioDeviceMapper.getBannerByDeviceId(aioDevice.getId());
|
|
return ApiResponse.success(banners);
|
|
}
|
|
|
|
@GetMapping("/config")
|
|
public ApiResponse<AioDevicePriceConfigEntity> getPriceConfig(HttpServletRequest request) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
AioDevicePriceConfigEntity config = aioDeviceMapper.getPriceConfigByDeviceId(aioDevice.getId());
|
|
return ApiResponse.success(config);
|
|
}
|
|
|
|
@PostMapping("/faceUpload")
|
|
public ApiResponse<FaceRecognizeResp> faceUpload(@RequestParam("file") MultipartFile file, HttpServletRequest request) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
MemberEntity memberEntity = new MemberEntity();
|
|
memberEntity.setScenicId(aioDevice.getScenicId());
|
|
memberEntity.setCreateDate(new Date());
|
|
memberEntity.setId(SnowFlakeUtil.getLongId());
|
|
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);
|
|
}
|
|
|
|
// 人脸信息
|
|
@GetMapping("/{faceId}")
|
|
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("/face/{faceId}/bindWxaCode")
|
|
public ApiResponse<String> bindWxaCode(@PathVariable Long faceId) {
|
|
return ApiResponse.success(faceService.bindWxaCode(faceId));
|
|
}
|
|
// 照片商品列表
|
|
@GetMapping("/{faceId}/photo")
|
|
public ApiResponse<List<GoodsDetailVO>> sourceGoodsList(@PathVariable Long faceId) {
|
|
GoodsReqQuery query = new GoodsReqQuery();
|
|
query.setSourceType(2);
|
|
query.setFaceId(faceId);
|
|
List<GoodsDetailVO> goodsDetailVOS = goodsService.sourceGoodsList(query);
|
|
return ApiResponse.success(goodsDetailVOS);
|
|
}
|
|
|
|
// 创建订单
|
|
@PostMapping("/order")
|
|
public ApiResponse<AioDeviceCreateOrderResp> createOrder(HttpServletRequest request, @RequestBody AioDeviceCreateOrderReq req) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
return ApiResponse.success(aioDeviceService.createOrder(aioDevice, req));
|
|
}
|
|
|
|
// 查询订单
|
|
@GetMapping("/order/{orderId}")
|
|
public ApiResponse<PayResponse> queryOrder(HttpServletRequest request, @PathVariable("orderId") Long orderId) {
|
|
String deviceId = request.getHeader("X-DeviceId");
|
|
AioDeviceEntity aioDevice = aioDeviceMapper.getByKey(deviceId);
|
|
if (aioDevice == null) {
|
|
return ApiResponse.fail("设备不存在");
|
|
}
|
|
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;
|
|
}
|
|
}
|