新版仿ok交易所-后端
zyy
2025-09-01 69b5fdc593fa73424448df0c911c5177f7497e7c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package com.yami.trading.api.controller;
 
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yami.trading.bean.ico.domain.Ico;
import com.yami.trading.bean.ico.domain.UserSubscription;
import com.yami.trading.bean.ico.dto.UserSubscriptionDTO;
import com.yami.trading.bean.ico.query.IcoQuery;
import com.yami.trading.bean.item.query.ItemQuery;
import com.yami.trading.common.domain.Result;
import com.yami.trading.common.exception.YamiShopBindException;
import com.yami.trading.common.query.QueryWrapperGenerator;
import com.yami.trading.security.common.util.SecurityUtils;
import com.yami.trading.service.ico.IcoService;
import com.yami.trading.service.ico.UserSubscriptionService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
 
import javax.validation.Valid;
import java.util.concurrent.TimeUnit;
 
@RestController
@RequestMapping("api/ico")
@Api(tags = "新币申购")
@Slf4j
public class ApiIcoController {
 
    @Autowired
    IcoService icoService;
 
    @Autowired
    UserSubscriptionService userSubscriptionService;
 
    @Autowired
    private RedissonClient redissonClient;
 
    @ApiOperation(value = "新币列表,配置列表")
    @GetMapping("list")
    public Result<Page<Ico>> list(ItemQuery itemQuery, Page<Ico> page) throws Exception {
        QueryWrapper queryWrapper = QueryWrapperGenerator.buildQueryCondition (itemQuery, ItemQuery.class);
        Page<Ico> result = icoService.page(page, queryWrapper);
        return Result.ok(result);
    }
 
    @ApiOperation(value = "新币申购")
    @PostMapping("subscribe")
    @ResponseBody
    public Result<String> subscribe(@Valid UserSubscription model) {
        String partyId = SecurityUtils.getUser().getUserId();
        if (StringUtils.isEmpty(partyId)) {
            throw new YamiShopBindException("请重新登录");
        }
        RLock rLock = redissonClient.getLock("subscribe" + partyId);
        boolean lockAcquired = false;
        try {
            // 尝试获取锁,最多等待5秒
            lockAcquired = rLock.tryLock(5, TimeUnit.SECONDS);
            if (!lockAcquired) {
                log.warn("无法获取锁: subscribe{}", partyId);
                throw new YamiShopBindException("请稍后再试");
            }
            model.setUserId(partyId);
            return icoService.subscribe(model);
        } catch (YamiShopBindException e) {
            log.error("错误信息: {}", e.getMessage(), e);
            throw e; // 重新抛出自定义异常
        } catch (Exception e) {
            log.error("系统异常: {}", e.getMessage(), e);
            throw new YamiShopBindException("操作失败,请稍后再试");
        } finally {
            // 确保释放锁
            if (lockAcquired && rLock.isHeldByCurrentThread()) {
                rLock.unlock();
            }
        }
    }
 
    @ApiOperation(value = "新币申购记录列表")
    @GetMapping("recordList")
    public Result<Page<UserSubscriptionDTO>> recordList(IcoQuery icoQuery, Page<UserSubscriptionDTO> page) throws Exception {
        QueryWrapper queryWrapper = QueryWrapperGenerator.buildQueryCondition (icoQuery, IcoQuery.class);
        String partyId = SecurityUtils.getUser().getUserId();
        if (StringUtils.isEmpty(partyId)) {
            throw new YamiShopBindException("请重新登录");
        }
        queryWrapper.eq("u.user_id", partyId);
        Page<UserSubscriptionDTO> result = userSubscriptionService.findPage(page, queryWrapper);
        return Result.ok(result);
    }
 
}