package com.fdkankan.ucenter.service.impl;
import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.common.constant.SceneConstant;
import com.fdkankan.common.constant.SceneKind;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.util.*;
import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.ucenter.common.OssPath;
import com.fdkankan.ucenter.common.PageInfo;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.image.MatrixToImageWriterUtil;
import com.fdkankan.ucenter.common.constants.ConstantFilePath;
import com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.ucenter.common.utils.ShellUtil;
import com.fdkankan.ucenter.constant.CameraConstant;
import com.fdkankan.ucenter.constant.LoginConstant;
import com.fdkankan.ucenter.entity.*;
import com.fdkankan.ucenter.httpClient.service.LaserService;
import com.fdkankan.ucenter.mapper.ISceneProMapper;
import com.fdkankan.ucenter.mapper.ISceneUpgradeMapper;
import com.fdkankan.ucenter.service.*;
import com.fdkankan.ucenter.util.DateUserUtil;
import com.fdkankan.ucenter.vo.ResponseScene;
import com.fdkankan.ucenter.vo.request.RequestScene;
import com.fdkankan.ucenter.vo.request.SceneParam;
import com.fdkankan.ucenter.vo.response.GroupByCount;
import com.fdkankan.ucenter.vo.response.SceneInfoVo;
import com.fdkankan.ucenter.vo.response.SceneNumVo;
import com.fdkankan.ucenter.vo.response.SceneVo;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.locks.Condition;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.joda.time.DateTime;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;
/**
*
* pro场景表 服务实现类
*
*
* @author
* @since 2022-07-04
*/
@Service
@Slf4j
public class SceneProServiceImpl extends ServiceImpl implements ISceneProService {
@Autowired
private IUserService userService;
@Autowired
private ISceneCooperationService sceneCooperationService;
@Autowired
private IScenePlusService scenePlusService;
@Autowired
private ISceneService sceneService;
@Autowired
private LaserService fdkkLaserService;
@Autowired
private FYunFileServiceInterface fYunFileService;
@Autowired
private IUserIncrementService userIncrementService;
@Autowired
private IIncrementTypeService incrementTypeService;
@Autowired
private IScenePlusExtService scenePlusExtService;
@Autowired
private ISceneEditInfoService sceneEditInfoService;
@Autowired
private ICameraDetailService cameraDetailService;
@Autowired
ICameraService cameraService;
@Autowired
private IFolderService folderService;
@Autowired
private IFolderSceneService folderSceneService;
@Autowired
private ISceneCopyLogService sceneCopyLogService;
@Autowired
private IScene3dNumService scene3dNumService;
@Autowired
private ISceneProEditService sceneProEditService;
@Autowired
private RedisUtil redisUtil;
@Autowired
private ISceneUpgradeMapper sceneUpgradeMapper;
@Autowired
private IInnerService innerService;
@Autowired
LaserService laserService;
@Value("${main.url}")
private String mainUrl;
@Value("${scene.pro.url}")
private String sceneProUrl;
@Value("${scene.pro.new.url}")
private String sceneProNewUrl;
@Value("${queue.modeling.obj.modeling-pre}")
private String queueObjModelingPre;
@Autowired
private RabbitMqProducer mqProducer;
@Override
public HashMap findSceneNumber(String token) {
User user = userService.getByUserName(JwtUtil.getUsername(token));
SceneNumVo sceneNumVoKk = getSceneNumVoByType(Arrays.asList(1,2,12,13,14), user.getId());
Integer oldNum = sceneService.getCountByUserId(user.getId());
sceneNumVoKk.setSceneNum( oldNum + sceneNumVoKk.getSceneNum());
sceneNumVoKk.setTotalNum(oldNum + sceneNumVoKk.getTotalNum());
SceneNumVo sceneNumVoKj = getSceneNumVoByType(Collections.singletonList(3), user.getId());
SceneNumVo sceneNumVoSS = fdkkLaserService.getLaserSceneNumByUser(token);
SceneNumVo sceneNumVoSSObj = getSceneNumVoByObjType(Arrays.asList(4), user.getId());
HashMap hashMap = new HashMap<>();
hashMap.put("kk",sceneNumVoKk);
hashMap.put("kJ",sceneNumVoKj);
hashMap.put("SS",sceneNumVoSS);
hashMap.put("SS_OBJ",sceneNumVoSSObj);
return hashMap;
}
private SceneNumVo getSceneNumVoByType(List sceneSourceList, Long userId){
SceneNumVo sceneNumVo = new SceneNumVo();
LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(ScenePro::getUserId, userId);
queryWrapper.in(ScenePro::getSceneSource, sceneSourceList);
queryWrapper.eq(ScenePro::getIsUpgrade, 0);
Integer sceneNum = Math.toIntExact(this.count(queryWrapper));
Long count = scenePlusService.getCountByUserId(userId,sceneSourceList);
sceneNumVo.setCooperationSceneNum(sceneCooperationService.getCooperationSceneNum(userId,sceneSourceList));
sceneNumVo.setSceneNum(sceneNum + count);
sceneNumVo.setTotalNum(sceneNumVo.getSceneNum() + sceneNumVo.getCooperationSceneNum());
return sceneNumVo;
}
private SceneNumVo getSceneNumVoByObjType(List sceneSourceList, Long userId){
SceneNumVo sceneNumVo = new SceneNumVo();
LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(ScenePro::getUserId, userId);
queryWrapper.in(ScenePro::getSceneSource, sceneSourceList);
queryWrapper.eq(ScenePro::getIsUpgrade, 0);
queryWrapper.eq(ScenePro::getIsObj,1);
Integer sceneNum = Math.toIntExact(this.count(queryWrapper));
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.eq(ScenePlusExt::getIsObj,1);
List list = scenePlusExtService.list(wrapper);
List plusIds = list.stream().map(ScenePlusExt::getPlusId).collect(Collectors.toList());
long plusCount = 0L;
long cooperNum = 0L;
if(plusIds.size() >0){
LambdaQueryWrapper plusWr = new LambdaQueryWrapper<>();
plusWr.in(ScenePlus::getId,plusIds);
plusWr.in(ScenePlus::getSceneSource,sceneSourceList);
List plusList = scenePlusService.list(plusWr);
if(plusList.size() >0){
List userList = plusList.stream().filter(entity -> userId.equals(entity.getUserId())).collect(Collectors.toList());
plusCount = userList.size();
List numList = plusList.stream().map(ScenePlus::getNum).collect(Collectors.toList());
HashMap cooNumList = sceneCooperationService.getByNumList(numList);
for (String num : cooNumList.keySet()) {
if(userId.equals(cooNumList.get(num).getId())){
cooperNum ++;
}
}
}
}
sceneNumVo.setCooperationSceneNum(cooperNum);
sceneNumVo.setSceneNum(sceneNum + plusCount);
sceneNumVo.setTotalNum(sceneNumVo.getSceneNum() + sceneNumVo.getCooperationSceneNum());
return sceneNumVo;
}
@Override
public HashMap findSceneNumByCameraIds(List cameraIdList) {
HashMap map = new HashMap<>();
List result = this.getBaseMapper().findSceneProNumByCameraIds(cameraIdList);
List result2 = this.getBaseMapper().findScenePlusNumByCameraIds(cameraIdList);
result.forEach(entity ->map.put(entity.getId(),entity));
for (GroupByCount groupByCount : result2) {
if (map.get(groupByCount.getId()) != null) {
Long totalCount = map.get(groupByCount.getId()).getCount() + groupByCount.getCount();
String lastTime = DateUserUtil.getLastTime(map.get(groupByCount.getId()).getLastTime(), groupByCount.getLastTime());
groupByCount.setCount(totalCount);
groupByCount.setLastTime(lastTime);
}
map.put(groupByCount.getId(),groupByCount);
}
return map;
}
/**
* @param payStatus -2 封存,为 1 解封
*/
@Override
public void lockOrUnLockBySpace(CameraDetail cameraDetail, Long cameraId, Integer payStatus) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
LambdaQueryWrapper plusWr = new LambdaQueryWrapper<>();
Long totalSpace = cameraDetail.getTotalSpace();
UserIncrement userIncrement = userIncrementService.getByCameraId(cameraId);
if(userIncrement!=null){
IncrementType incrementType = incrementTypeService.getById(userIncrement.getIncrementTypeId());
if(incrementType!=null){
if(incrementType.getCameraCapacity() == -1){
totalSpace = -1L;
}else {
totalSpace = incrementType.getCameraCapacity() * 1024 * 1024L;
}
}
}
log.info("lockOrUnLockBySpace---cameraId:{},payStatus:{},totalSpace:{},useSpace{}",
cameraId,payStatus,totalSpace,cameraDetail.getUsedSpace());
if(payStatus == 1){ //解封,判断用户权益,用户会员权益无限容量
wrapper.orderByAsc(ScenePro::getCreateTime);
plusWr.orderByAsc(ScenePlus::getCreateTime);
wrapper.eq(ScenePro::getPayStatus,-2);
plusWr.eq(ScenePlus::getPayStatus,-2);
}else {
if (totalSpace >= cameraDetail.getUsedSpace()) {
// 总容量大于已使用容量,不予封存
return;
}
wrapper.orderByDesc(ScenePro::getCreateTime);
plusWr.orderByDesc(ScenePlus::getCreateTime);
wrapper.eq(ScenePro::getPayStatus,1);
plusWr.eq(ScenePlus::getPayStatus,1);
}
wrapper.eq(ScenePro::getCameraId,cameraId)
.eq(ScenePro::getIsUpgrade,0);
plusWr.eq(ScenePlus::getCameraId,cameraId);
List list = this.list(wrapper);
List plusList = scenePlusService.list(plusWr);
Long beyondSpace = Math.abs(cameraDetail.getUsedSpace() - totalSpace);
Long accumulateSpace = 0L;
List lockedIds = new ArrayList<>();
if(payStatus == 1){
Long doSpace = getScenePlusLockedIds(lockedIds, plusList, totalSpace, beyondSpace, accumulateSpace);
beyondSpace -= doSpace;
getSceneLockedIds(lockedIds,list,totalSpace,beyondSpace,accumulateSpace);
}else {
Long doSpace = getSceneLockedIds(lockedIds, list, totalSpace, beyondSpace, accumulateSpace);
beyondSpace -= doSpace;
getScenePlusLockedIds(lockedIds,plusList,totalSpace,beyondSpace,accumulateSpace);
}
lockOrUnLockScenes(lockedIds,payStatus);
}
private Long getSceneLockedIds(List lockedIds ,List list,Long count,Long beyondSpace,Long accumulateSpace){
if (list != null && list.size() > 0){
for (ScenePro scenePro : list){
accumulateSpace += scenePro.getSpace();
if (count ==-1 && accumulateSpace.compareTo(beyondSpace) > 0){
return accumulateSpace - scenePro.getSpace();
}
lockedIds.add(scenePro.getId());
}
}
return accumulateSpace;
}
private Long getScenePlusLockedIds(List lockedIds , List list, Long count, Long beyondSpace, Long accumulateSpace){
if (list != null && list.size() > 0){
List plusIds = list.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
HashMap byPlusIds = scenePlusExtService.getByPlusIds(plusIds);
for (ScenePlus scenePlus : list){
ScenePlusExt scenePlusExt = byPlusIds.get(scenePlus.getId());
accumulateSpace += scenePlusExt.getSpace();
if (count ==-1 && accumulateSpace.compareTo(beyondSpace) > 0){
return accumulateSpace - scenePlusExt.getSpace();
}
lockedIds.add(scenePlus.getId());
}
}
return accumulateSpace;
}
// payStatus 为 -2 封存,为 1 解封
private void lockOrUnLockScenes(List lockedIds,Integer payStatus) {
if (lockedIds == null || lockedIds.size() == 0){
return;
}
LambdaUpdateWrapper updateWrapper = new LambdaUpdateWrapper<>();
updateWrapper.set(ScenePro::getPayStatus,payStatus)
.eq(ScenePro::getIsUpgrade,0)
.in(ScenePro::getId,lockedIds);
this.update(updateWrapper);
LambdaUpdateWrapper updatePlusWrapper = new LambdaUpdateWrapper<>();
updatePlusWrapper.set(ScenePlus::getPayStatus,payStatus)
.in(ScenePlus::getId,lockedIds);
scenePlusService.update(updatePlusWrapper);
this.updateOssStatus(lockedIds,payStatus);
}
/**
* 修改oss status.json中 payStatus
*/
private void updateOssStatus(List lockedIds, Integer payStatus) {
LambdaQueryWrapper proWr = new LambdaQueryWrapper<>();
proWr.eq(ScenePro::getIsUpgrade,0);
proWr.in(ScenePro::getId,lockedIds);
List proList = this.list(proWr);
LambdaQueryWrapper pluWr = new LambdaQueryWrapper<>();
pluWr.in(ScenePlus::getId,lockedIds);
List plusList = scenePlusService.list(pluWr);
for (ScenePro scenePro : proList) {
this.updateOssStatus(String.format(OssPath.v3_statusPath,scenePro.getNum()),payStatus);
}
for (ScenePlus scenePlus : plusList) {
this.updateOssStatus(String.format(OssPath.v4_statusPath,scenePlus.getNum()),payStatus);
}
}
/**
* 从oss中获取文件,并重写,上传替换
*/
private void updateOssStatus(String path,Integer payStatus) {
try {
if(!fYunFileService.fileExist(path)){
return;
}
String data = fYunFileService.getFileContent(path);
if(StringUtils.isBlank(data)){
return;
}
JSONObject jsonObject = JSONObject.parseObject(data);
jsonObject.put("payStatus",payStatus);
String json = JSONUtil.toJsonStr(jsonObject);
FileUtils.writeFile(OssPath.localStatusPath ,json);
log.info("updateOssStatus--localPath:{},ossPath:{}",OssPath.localStatusPath,path);
fYunFileService.uploadFile(OssPath.localStatusPath,path);
}catch (Exception e){
e.printStackTrace();
}finally {
FileUtil.del(OssPath.localStatusPath);
}
}
@Override
public List getListByCameraId(Long cameraId) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.eq(ScenePro::getCameraId,cameraId)
.eq(ScenePro::getIsUpgrade,0);
return this.list(wrapper);
}
@Override
public void bindOrUnCamera(List cameraIds, Long userId) {
if(cameraIds.size() >0){
LambdaUpdateWrapper wrapper = new LambdaUpdateWrapper<>();
wrapper.in(ScenePro::getCameraId,cameraIds)
.eq(ScenePro::getIsUpgrade,0)
.set(ScenePro::getUserId,userId);
this.update(wrapper);
}
}
@Override
public List getListByCameraIds(List cameraIds) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.in(ScenePro::getCameraId,cameraIds)
.eq(ScenePro::getIsUpgrade,0);
return this.list(wrapper);
}
@Override
public List getListByNums(List numList) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.in(ScenePro::getNum,numList)
.eq(ScenePro::getIsUpgrade,0);
return this.list(wrapper);
}
@Override
public Long getCountByUserId(Long userId, List> resourceList) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.eq(ScenePro::getUserId,userId);
wrapper.eq(ScenePro::getIsUpgrade,0);
wrapper.in(ScenePro::getSceneSource,resourceList);
return this.count(wrapper);
}
@Override
public Long getCountByUserId(Long id, Integer cameraType) {
return this.getBaseMapper().getCountByUserId(id,cameraType);
}
@Override
public void deleteByIds(List sceneIds) {
LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
wrapper.in(ScenePro::getId,sceneIds)
.eq(ScenePro::getIsUpgrade,0);
List proList = this.list(wrapper);
LambdaQueryWrapper plusWr = new LambdaQueryWrapper<>();
plusWr.in(ScenePlus::getId,sceneIds);
List plusList = scenePlusService.list(plusWr);
this.deleteByList(proList,plusList);
}
@Override
public Page pageListAndFolder(Page