feat(user): 重构用户注册逻辑并添加钱包功能

KKSU

2025-02-18 0140571d040178c6a1c4c6f66f8df0dcb46ab5ac

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
package cc.mrbird.febs.user.service.Impl;
 
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.ValidateEntityUtils;
import cc.mrbird.febs.user.entity.ChatWallet;
import cc.mrbird.febs.user.mapper.ChatWalletMapper;
import cc.mrbird.febs.user.service.IChatWalletService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
 
import java.math.BigDecimal;
 
@Slf4j
@Service
@RequiredArgsConstructor
public class ChatWalletServiceImpl extends ServiceImpl<ChatWalletMapper, ChatWallet> implements IChatWalletService {
 
    @Override
    public void addBalance(BigDecimal amount, Long memberId) {
        if (BigDecimal.ZERO.compareTo(amount) > 0) {
            throw new FebsException("增加余额异常");
        }
        int i = 0;
        boolean flag = true;
 
        while (flag) {
            i++;
            ChatWallet wallet = ValidateEntityUtils.ensureColumnReturnEntity(memberId, ChatWallet::getMemberId, this.baseMapper::selectOne, "钱包不存在");
            wallet.setBalance(wallet.getBalance().add(amount));
            int result = this.baseMapper.updateBalanceWithVersion(wallet);
            if (result > 0) {
                flag = false;
            } else {
                if (i > 2) {
                    throw new FebsException("操作失败");
                }
            }
        }
    }
 
    @Override
    public void reduceBalance(BigDecimal amount, Long memberId) {
        int i = 0;
        boolean flag = true;
 
        while (flag) {
            i++;
            ChatWallet wallet = ValidateEntityUtils.ensureColumnReturnEntity(memberId, ChatWallet::getMemberId, this.baseMapper::selectOne, "钱包不存在");
            if (amount.compareTo(wallet.getBalance()) > 0) {
                throw new FebsException("余额不足");
            }
 
            wallet.setBalance(wallet.getBalance().subtract(amount));
 
            int result = this.baseMapper.updateBalanceWithVersion(wallet);
            if (result > 0) {
                flag = false;
            } else {
                if (i > 2) {
                    throw new FebsException("操作失败");
                }
            }
        }
    }
 
 
}