package com.iplatform.file.controller;

import com.iplatform.base.SystemController;
import com.iplatform.base.config.ApiProperties;
import com.walker.file.FileInfo;
import com.walker.infrastructure.utils.FileCopyUtils;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.web.ResponseValue;
import com.walker.web.util.ServletUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@RestController
@RequestMapping("/oss")
public class OssFileApi extends SystemController {

    private ApiProperties apiProperties;

    @Autowired
    public OssFileApi(ApiProperties apiProperties){
        this.apiProperties = apiProperties;
    }

    @RequestMapping(value = "/{id}", method = RequestMethod.GET)
    @ResponseBody
    public void downloadOssFile(@PathVariable(name = "id") String id){
        if(StringUtils.isEmpty(id)){
            throw new IllegalArgumentException("文件参数为空，无法下载文件");
        }
        FileInfo fileInfo = this.acquireFileOperateSpi().getFileInfo(Long.parseLong(id));
        if(fileInfo == null){
            throw new IllegalArgumentException("文件不存在，id=" + id);
        }

        // 2024-03-08 为压测提供支持，
        try {
            byte[] fileContent = null;
            if(this.apiProperties.isTimeEnabled()){
                // 压测模式，记录接口调用时间
                fileContent = this.acquirePressureTestFile(fileInfo);
            } else {
                // 正常模式
                fileContent = this.getOssFileData(id);
            }
            this.downloadSimpleFile(fileContent, fileInfo.getFileName());

        } catch (IOException e) {
            logger.error("下载oss错误:" + e.getMessage() + ", id=" + id, e);
            ServletUtils.renderString(getResponse(), "下载oss错误:" + e.getMessage() + ", id=" + id);
        }
    }

    /**
     * 压测时尝试从缓存查找文件，如果未缓存会把默认64个文件缓存。如果超过64个，先清除缓存，并重新开始缓存后续64个，循环往复。
     * @param fileInfo
     * @return
     * @author 时克英
     * @date 2024-03-08
     */
    private byte[] acquirePressureTestFile(FileInfo fileInfo){
        byte[] fileData = null;
        long fileSize = fileInfo.getFileSize();
        if(fileSize <= maxCacheFileSize){
            // 当缓存数量达到上限时，清空重新开始缓存新的
            if(fileContentCache.size() >= maxCacheFileCount){
                fileContentCache.clear();
            }
            // 当文件小于等于1M，并且缓存数量未达到上限时，缓存
            fileData = this.fileContentCache.get(fileInfo.getId());
            if(fileData == null){
                fileData = this.getOssFileData(fileInfo.getId());
                this.fileContentCache.put(fileInfo.getId(), fileData);
                logger.debug("缓存一个oss文件：{}，大小：{}", fileInfo.getFileName(), fileSize);
            } else {
                logger.debug("从缓存加载了oss文件：{}", fileInfo.getId());
            }
        } else {
            fileData = this.getOssFileData(fileInfo.getId());
        }
        return fileData;
    }

    private Map<String, byte[]> fileContentCache = new ConcurrentHashMap<>(128);

    private int maxCacheFileCount = 64;
    private long maxCacheFileSize = 1024 * 1024;

    /**
     * 模拟测试。
     */
    @RequestMapping(value = "/test/demo", method = RequestMethod.GET)
    public void downloadOssDemo(){
        String file = "F:/app_ocr_demo/ocr_e_005.jpg";
        try {
            this.downloadSimpleFile(FileCopyUtils.copyToByteArray(new File(file)), "测试图片");
        } catch (IOException e) {
            logger.error("下载oss demo错误:" + e.getMessage() + ", id=" + file, e);
            ServletUtils.renderString(getResponse(), "下载oss demo错误:" + e.getMessage() + ", id=" + file);
        }
    }

    @RequestMapping(value = "/test/upload", method = RequestMethod.GET)
    public ResponseValue demoUploadOss() throws Exception{
        File file = new File("F:/app_ocr_demo/ocr_e_005.jpg");
        FileInputStream inputStream = new FileInputStream(file);
        FileInfo fileInfo = this.uploadFileToRemote(inputStream, "demo.jpg", null, file.length(), 0, "-1");
        logger.info("fileInfo = {}", fileInfo);
        return ResponseValue.success("上传文件成功，" + fileInfo.getUrl());
    }
}