Helius
2022-04-02 06d334585a2174726e3574ea939e10beec620bae
src/main/java/cc/mrbird/febs/dapp/service/impl/DappUserServiceImpl.java
@@ -48,9 +48,22 @@
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void add(User user) {
        User currentUser = FebsUtil.getCurrentUser();
        DappMemberEntity member = dappMemberDao.selectMemberInfoByInviteId(user.getInviteId());
        if(member == null) {
            throw new FebsException("邀请码错误");
        }
        if (currentUser.getDeptId() == null) {
            DappUserMemberRelateEntity relate = dappUserMemberRelateDao.selectByUserId(currentUser.getUserId());
            if (!relate.getInviteId().equals(member.getRefererId())) {
                throw new FebsException("该用户不是直推");
            }
        }
        DappUserMemberRelateEntity userMemberRelate = dappUserMemberRelateDao.selectByInviteId(user.getInviteId());
        if (userMemberRelate != null) {
            throw new FebsException("邀请码用户已存在");
        }
        user.setCreateTime(new Date());
@@ -66,7 +79,6 @@
        userRole.setUserId(user.getUserId());
        userRoleMapper.insert(userRole);
        User currentUser = FebsUtil.getCurrentUser();
        DappUserMemberRelateEntity relate = new DappUserMemberRelateEntity();
        relate.setMemberId(member.getId());
        relate.setUserId(user.getUserId());