package cc.mrbird.febs.user.service.Impl; import cc.mrbird.febs.common.entity.FebsResponse; import cc.mrbird.febs.common.exception.FebsException; import cc.mrbird.febs.common.utils.ShareCodeUtil; import cc.mrbird.febs.common.utils.ValidateEntityUtils; import cc.mrbird.febs.mall.service.impl.CommonService; import cc.mrbird.febs.user.contant.MemberConstant; import cc.mrbird.febs.user.entity.ChatMember; import cc.mrbird.febs.user.mapper.ChatMemberMapper; import cc.mrbird.febs.user.req.ApiRegisterDto; import cc.mrbird.febs.user.service.AsyncService; import cc.mrbird.febs.user.service.IChatMemberService; import cn.hutool.core.date.DateUtil; import cn.hutool.core.util.StrUtil; import cn.hutool.crypto.SecureUtil; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; @Slf4j @Service @RequiredArgsConstructor @Transactional public class ChatMemberServiceImpl extends ServiceImpl implements IChatMemberService { private final CommonService commonService; private final AsyncService asyncService; @Override public FebsResponse register(ApiRegisterDto apiRegisterDto) { String account = apiRegisterDto.getAccount(); String password = apiRegisterDto.getPassword(); String passwordAgain = apiRegisterDto.getPasswordAgain(); String refInviteId = apiRegisterDto.getInviteId(); String code = apiRegisterDto.getCode(); //后台添加用户,不需要验证验证码 Integer addType = apiRegisterDto.getAddType(); if (MemberConstant.ADD_TYPE_NORMAL == addType) { boolean flags = commonService.verifyCode(account, code); if (!flags) { throw new FebsException("验证码错误"); } } ValidateEntityUtils.ensureEqual(password, passwordAgain, "两次密码不一致"); ValidateEntityUtils.ensureUnique(account, ChatMember::getEmail, this.baseMapper::selectCount, "邮箱已使用"); ChatMember chatMember = new ChatMember(); chatMember.setEmail(account); chatMember.setPassword(SecureUtil.md5(password)); chatMember.setName(account); chatMember.setSex(MemberConstant.UNKNOWN); //对于邀请码的验证和上级联系人的验证 Integer count = this.baseMapper.selectCount(null); if (count != null && count != 0) { if(StrUtil.isNotEmpty(refInviteId)){ ChatMember chatMemberInvite = ValidateEntityUtils.ensureColumnReturnEntity(refInviteId, ChatMember::getInviteId, this.baseMapper::selectOne, "邀请码不存在"); chatMember.setReferrerId(chatMemberInvite.getInviteId()); } } chatMember.setAccountStatus(MemberConstant.ACCOUNT_STATUS_NORMAL); chatMember.setAccountType(MemberConstant.ACCOUNT_TYPE_NORMAL); // --todo 默认头像为空 // chatMember.setAvatar("https://res.runstep.cc/rslogo.png"); chatMember.setDirector(MemberConstant.DIRECTER_YES); chatMember.setLastLoginTime(DateUtil.date()); // -- todo 默认主账号的角色设置没有设置 chatMember.setRoleId(0L); this.baseMapper.insert(chatMember); //公司ID默认为主账号的会员ID chatMember.setCompanyId(chatMember.getId()); chatMember.setInviteId(ShareCodeUtil.toSerialCode(chatMember.getId())); //推荐人链 asyncService.initMemberReferrerIds(chatMember.getId(),chatMember.getReferrerId()); asyncService.initWallet(chatMember.getId()); return new FebsResponse().success().message("注册成功"); } }