4dkankan-center-manage/src/main/java/com/fdkankan/manage/test/TestController.java

package com.fdkankan.manage.test;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.fdkankan.elasticsearch.service.DocumentService;
import com.fdkankan.elasticsearch.service.IndexService;
import com.fdkankan.elasticsearch.service.QueryDataService;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.manage.common.*;
import com.fdkankan.manage.entity.Camera;
import com.fdkankan.manage.entity.CameraDetail;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.manage.service.ICameraDetailService;
import com.fdkankan.manage.service.ICameraService;
import com.fdkankan.manage.service.IScenePlusService;
import com.fdkankan.manage.service.ISceneProService;
import com.fdkankan.manage.util.ExcelUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;

import static com.amazonaws.util.CredentialUtils.getCredentialsProvider;

@Slf4j
@RestController
@RequestMapping("/test")
public class TestController {

    @Autowired
    ISceneProService sceneProService;
    @Autowired
    IScenePlusService scenePlusService;
    @Autowired
    FYunFileServiceInterface fYunFileServiceInterface;
    @Autowired
    ICameraService cameraService;
    @Autowired
    ICameraDetailService cameraDetailService;
    @Autowired
    IndexService indexService;
    @Autowired
    QueryDataService queryDataService;
    @Autowired
    DocumentService documentService;

    @RequestMapping("/test")
    public ResultData test(@RequestParam(required = false) String oldSnCode,
                           @RequestParam(required = false)String newSnCode,
                           @RequestParam(required = false)String dataSource) throws Exception {

//        RestStatus restStatus = documentService.addDocument("abc-test1", null, "1", "{\n" +
//                "    \"appId\": \"7b5958d5-1ae6-4ad5-8a87-5fc8a4b92999\",\n" +
//                "    \"module\": \"module\",\n" +
//                "    \"userId\": null,\n" +
//                "    \"time\": 1662519289298,\n" +
//                "    \"url\": \"/#/roam\",\n" +
//                "    \"urlParam\": null,\n" +
//                "    \"requestData\": {\n" +
//                "      \"screen\": \"1920x1080\",\n" +
//                "      \"click\": \"filter\",\n" +
//                "      \"user\": \"\",\n" +
//                "      \"version\": \"1\"\n" +
//                "    }}");
//        log.info("{}",restStatus);
        List<String> strings = queryDataService.boolQuery("abc-test1", String.class);

        return ResultData.ok(JSONObject.toJSONString(strings));
    }

    /**
     * 重置相机容量
     */
    @RequestMapping("/testCameraSpace")
    public ResultData testCameraSpace(@RequestParam(required = false) Long userId){
        cameraService.initAllCameraSpace(userId);
        return ResultData.ok();
    }

    @RequestMapping("/testCameraSpaceId")
    public ResultData testCameraSpaceId(@RequestParam(required = false) Long cameraId){
        cameraDetailService.initSpace(cameraId);
        return ResultData.ok();
    }


    /**
     * 导入excel
     * type 0 入库模板， 1出库模板 ，2 客户关联模板
     */
    @PostMapping("/uploadExcel")
    public ResultData uploadExcel(@RequestParam(required = false) MultipartFile file) throws IOException {
        String originalFilename = file.getOriginalFilename();
        assert originalFilename != null;
        String fileType=originalFilename.substring(originalFilename.lastIndexOf(".")+1);
        if (!fileType.equalsIgnoreCase("xlsx")) {
            throw new BusinessException(ResultCode.FILE_TYPE_ERROR);
        }
        List<HashMap<Integer, String>> excelRowList = new ArrayList<>();
        try {
            excelRowList = ExcelUtil.getExcelRowList(file);
        }catch (Exception e){
            throw new BusinessException(ResultCode.TEMPLATE_TYPE_ERROR);
        }
        System.out.println(excelRowList);
        return ResultData.ok();
    }

    @GetMapping("/initCamera")
    public ResultData initCamera(){
        String[] snCodes = this.getSnCodes();
        List<String> snCodeList = Arrays.asList(snCodes);
        LambdaQueryWrapper<Camera> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(Camera::getSnCode,snCodeList);
        List<Camera> list = cameraService.list(wrapper);
        for (Camera camera : list) {
            if(camera == null){
                continue;
            }
            CameraDetail cameraDetail = cameraDetailService.getByCameraId(camera.getId());
            if(cameraDetail == null){
                continue;
            }
            Long proSpace = sceneProService.getCountByCameraId(cameraDetail.getCameraId());
            Long plusSpace = scenePlusService.getCountByCameraId(cameraDetail.getCameraId());
            LambdaUpdateWrapper<CameraDetail> wrapper1 = new LambdaUpdateWrapper<>();
            wrapper1.eq(CameraDetail::getId,cameraDetail.getId());
            wrapper1.set(CameraDetail::getUnit,"SP");
            wrapper1.set(CameraDetail::getTotalSpace,5);
            wrapper1.set(CameraDetail::getUsedSpace,(plusSpace == null ?0L:plusSpace )+( proSpace == null ?0L:proSpace));
            cameraDetailService.update(wrapper1);
            sceneProService.lockOrUnLockBySpace(cameraDetail.getCameraId());
        }

        return ResultData.ok();
    }

    private String[] getSnCodes(){
        String res [] = {
                "067E15841",
                "07C3DA2E1",
                "07EC6E0DC",
                "06FFCBDA1",
                "061D18D73",
                "030285552",
                "06EE53219",
                "02452945D",
                "060C4CB6A",
                "06FAE7E46",
                "06F860091",
                "072D80097",
                "073BD9E33",
                "04AE6E2F5",
                "071123F88",
                "071256BC7",
                "071210632",
                "071265E8B",
                "070D743D2",
                "0700FA837",
                "0703A543A",
                "070B42A10",
                "070BE22BA",
                "0711305CA",
                "0694087DB",
                "070273265",
                "071624ACD",
                "072FAE15E",
                "07F3DD250",
                "0707D2A2C",
                "0266F55AE",
                "021DD5E87",
                "02A80E87F",
                "02C6209A2",
                "02FE298D8",
                "067B891A9",
                "0408E93F5",
                "0650ABFFB",
                "070DE64AF",
                "02D9E2F29",
                "06B38A34F",
                "063DE2E04",
                "072ACF5F2",
                "07288B5BF",
                "02FFAB589",
                "072D27B03",
                "075BB0B0A",
                "033869CC0",
                "06EA790F7",
                "06105B947",
                "04FC2C13C",
                "065E1EA54",
                "06050CCD2",
                "065DAAB57",
                "06376454A",
                "069CE5379",
                "06FE7A53F",
                "06DA9EADA",
                "062B959FB",
                "07058CC1C",
                "0721ADBCF",
                "0708E15A3",
                "073B6BE96",
                "07757C452",
                "076A4A2C2",
                "07467B94D",
                "07F838850",
                "076AAAA05",
                "0766855A9",
                "0765692AA",
                "077A0D967",
                "0707D67D9",
                "07676CC0C",
                "074C7BE97",
                "0772F50DA",
                "0712A8FA0",
                "0768CD82B",
                "0764A39B5",
                "0771B69E4",
                "07F66673D",
                "077705FEE",
                "07696266D",
                "076F8D814",
                "077EBF4FE",
                "07109D1B0",
                "06F83452B",
                "0300B6F0A",
                "0774EA21B",
                "070D890E8",
                "077E7B417",
                "07DE244F1",
                "070591078",
                "03451435F",
                "0775B3B58",
                "07239FFFB",
                "070E51723",
                "076F1650D",
                "07199F0E6",
                "0703C265E",
                "07451A77B",
                "0711ABE58",
                "07BE0B836",
                "070AFD6DD",
                "070A40E42",
                "07143F457",
                "0706CE0EC",
                "0713DCCC0",
                "073AFACA6",
                "025D3928F",
                "02D874C48",
                "06042AB08",
                "04CA21B73",
                "02F7D171B",
                "02C17AE5E",
                "024696CB5",
                "4DKKP01B000127",
                "4DKKP01B000114",
                "4DKKP01B000138",
                "4DKKP01B000160",
                "4DKKP01B000163",
        };
        return res;
    }
}