gold-java/src/main/java/com/example/demo/controller/coin/GoldDetailController.java

package com.example.demo.controller.coin;

import com.example.demo.Util.BusinessException;
import com.example.demo.Util.JWTUtil;
import com.example.demo.Util.RedisLockUtil;
import com.example.demo.domain.DTO.GoldDetailDTO;
import com.example.demo.domain.DTO.GoldUserDTO;
import com.example.demo.domain.entity.Admin;
import com.example.demo.domain.entity.User;
import com.example.demo.domain.vo.coin.GoldDetail;
import com.example.demo.domain.vo.coin.Page;
import com.example.demo.domain.vo.coin.Result;
import com.example.demo.service.coin.GoldDetailService;
import com.example.demo.service.coin.MarketService;
import com.example.demo.serviceImpl.coin.AiEmotionServiceImpl;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.validation.Valid;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.Arrays;
import java.util.List;
import java.util.UUID;

/**
 * @program: GOLD
 * @ClassName GoldDetailController
 * @description:
 * @author: huangqizhen
 * @create: 2025−06-23 14:41
 * @Version 1.0
 **/

@RestController
@RequestMapping("/goldDetail")
@RequiredArgsConstructor
@Slf4j
@CrossOrigin
public class GoldDetailController {
    @Autowired
    private  GoldDetailService goldDetailService;


    @Autowired
    private RedisLockUtil redisLockUtil;
    @Autowired
    private AiEmotionServiceImpl aiEmotionServiceImpl;
    @Autowired
    MarketService marketService;

    @PostMapping("/getGoldDetail")
    public Result getGoldDetail(@RequestBody Page page) throws Exception {

        // 获取当前请求对象
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String token = request.getHeader("token");

// 解析 token 获取用户信息
        Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
        List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
        List<String> markets = marketService.getMarketIds(userMarkets);

// 校验分页参数
        if (ObjectUtils.isEmpty(page.getPageNum())) {
            return Result.error("页码数为空！");
        }
        if (ObjectUtils.isEmpty(page.getPageSize())) {
            return Result.error("页大小为空！");
        }

// 获取传入的市场列表
        List<String> requestedMarkets = page.getGoldDetail() != null ? page.getGoldDetail().getMarkets() : null;

// 权限校验
        if (requestedMarkets != null && !requestedMarkets.isEmpty()) {
            if (!markets.containsAll(requestedMarkets)) {
                // 如果 markets 不包含 requestedMarkets 中的所有元素，则返回无权限
                return Result.error("无权限！");
            }
        } else {
            // 如果传入的市场为空，则设置默认市场
            if (markets.contains("9") || markets.contains("9999")) {
                page.getGoldDetail().setMarkets(null); // 包含 9 或 9999，传 null
            } else {
                page.getGoldDetail().setMarkets(markets); // 不包含，传原值
            }
        }

// 返回详情数据
        return Result.success(goldDetailService.getGoldDetail(
                page.getPageNum(),
                page.getPageSize(),
                page.getGoldDetail()
        ));
    }
    @PostMapping("/getTotal")
    public Result getTotal(@RequestBody Page page) throws Exception {
        Integer pageNum = page.getPageNum();
        Integer pageSize = page.getPageSize();
        GoldDetail goldDetail = page.getGoldDetail();
        // 获取当前请求对象
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String token = request.getHeader("token");

// 解析 token 获取用户信息
        Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
        List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
        List<String> markets = marketService.getMarketIds(userMarkets);
// 获取传入的市场列表
        List<String> requestedMarkets = page.getGoldDetail() != null ? page.getGoldDetail().getMarkets() : null;

// 权限校验
        if (requestedMarkets != null && !requestedMarkets.isEmpty()) {
            if (!markets.containsAll(requestedMarkets)) {
                // 如果 markets 不包含 requestedMarkets 中的所有元素，则返回无权限
                return Result.error("无权限！");
            }
        } else {
            // 如果传入的市场为空，则设置默认市场
            if (markets.contains("9") || markets.contains("9999")) {
                page.getGoldDetail().setMarkets(null); // 包含 9 或 9999，传 null
            } else {
                page.getGoldDetail().setMarkets(markets); // 不包含，传原值
            }
        }
        return Result.success(goldDetailService.sumGold(pageNum,pageSize,goldDetail));
    }
    @PostMapping("/goldTotal")
    public Result GoldTotal(@RequestBody User user) throws Exception {
        // 获取当前请求对象
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String token = request.getHeader("token");

// 解析 token 获取用户信息
        Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
        List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
        List<String> markets = marketService.getMarketIds(userMarkets);


// 获取传入的市场列表
        List<String> requestedMarkets = user != null ? user.getMarkets() : null;

// 权限校验
        if (requestedMarkets != null && !requestedMarkets.isEmpty()) {
            if (!markets.containsAll(requestedMarkets)) {
                // 如果 markets 不包含 requestedMarkets 中的所有元素，则返回无权限
                return Result.error("无权限！");
            }
        } else {
            // 如果传入的市场为空，则设置默认市场
            if (markets.contains("9") || markets.contains("9999")) {
                user.setMarkets(null); // 包含 9 或 9999，传 null
            } else {
                user.setMarkets(markets); // 不包含，传原值
            }
        }
        return Result.success(goldDetailService.GoldTotal(user));
    }
    @PostMapping("/getGold")
    public Result getGold(@RequestBody Page  page) throws  Exception {
  // 获取当前请求对象
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String token = request.getHeader("token");

// 解析 token 获取用户信息
        Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
        List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
        List<String> markets = marketService.getMarketIds(userMarkets);

// 校验分页参数
        if (ObjectUtils.isEmpty(page.getPageNum())) {
            return Result.error("页码数为空！");
        }
        if (ObjectUtils.isEmpty(page.getPageSize())) {
            return Result.error("页大小为空！");
        }

// 获取传入的市场列表
        List<String> requestedMarkets = page.getUser() != null ? page.getUser().getMarkets() : null;

// 权限校验
        if (requestedMarkets != null && !requestedMarkets.isEmpty()) {
            if (!markets.containsAll(requestedMarkets)) {
                // 如果 markets 不包含 requestedMarkets 中的所有元素，则返回无权限
                return Result.error("无权限！");
            }
        } else {
            // 如果传入的市场为空，则设置默认市场
            if (markets.contains("9") || markets.contains("9999")) {
                page.getUser().setMarkets(null); // 包含 9 或 9999，传 null
            } else {
                page.getUser().setMarkets(markets); // 不包含，传原值
            }
        }
        return Result.success(goldDetailService.getGold(page.getPageNum(), page.getPageSize(), page.getUser()));
    }
    @PostMapping("/export")
    public Result export(@Valid @RequestBody GoldDetailDTO dto) {
        String lockKey = "export:lock:" + dto.getToken(); // 锁的 Key（可按用户/业务区分）
        String requestId = UUID.randomUUID().toString();  // 请求 ID（防止误删锁）
        long expireTime = 5000; // 锁过期时间（5秒）s
        try {
            // 尝试获取锁
            if (!redisLockUtil.tryLock(lockKey, requestId, expireTime)) {
                throw new BusinessException("操作太频繁，请稍后重试");
            }
            // 执行业务逻辑
            return goldDetailService.addExportRecord(dto);
        } finally {
            // 释放锁
            redisLockUtil.unlock(lockKey, requestId);
        }
    }
    @PostMapping("/exportGold")
    public Result export(@Valid @RequestBody GoldUserDTO dto) {
        String lockKey = "export:lock:" + dto.getToken(); // 锁的 Key（可按用户/业务区分）
        String requestId = UUID.randomUUID().toString();  // 请求 ID（防止误删锁）
        long expireTime = 5000; // 锁过期时间（5秒）s
        try {
            // 尝试获取锁
            if (!redisLockUtil.tryLock(lockKey, requestId, expireTime)) {
                throw new BusinessException("操作太频繁，请稍后重试");
            }
            // 执行业务逻辑
            return goldDetailService.addExportRecordGold(dto);
        } finally {
            // 释放锁
            redisLockUtil.unlock(lockKey, requestId);
        }
    }
    @PostMapping("/exportqqq")
    public Result ExcelGoldDetail(@RequestBody Page page) throws Exception {

        if(ObjectUtils.isEmpty(page.getPageNum())){
            return Result.error("页码数为空！");
        }
        if(ObjectUtils.isEmpty(page.getPageSize())){
            return Result.error("页大小为空！");
        }
        else{

            return Result.success(goldDetailService.getGoldDetail(page.getPageNum(), page.getPageSize(), page.getGoldDetail()));
        }
    }

    public Result ExcelGold(@RequestBody Page  page) throws  Exception {
//        System.out.println( page);
        if(ObjectUtils.isEmpty(page.getPageNum())){
            return Result.error("页码数为空！");
        }
        if(ObjectUtils.isEmpty(page.getPageSize())){
            return Result.error("页大小为空！");
        }

        return Result.success(goldDetailService.getGold(page.getPageNum(), page.getPageSize(), page.getUser()));
    }
}