修改 用户表更新邀请人id

ch-wx-api/src/main/java/com/wwdz/ch/wx/impl/InviteRecordServiceImpl.java

@@ -4,8 +4,10 @@ import com.wwdz.ch.core.consts.InvitationCodeStateEnum;
 import com.wwdz.ch.core.type.Result;
 import com.wwdz.ch.db.dao.InvitationCodeDao;
 import com.wwdz.ch.db.dao.InviteRecordDao;
+import com.wwdz.ch.db.dao.UserDao;
 import com.wwdz.ch.db.domain.InvitationCode;
 import com.wwdz.ch.db.domain.InviteRecord;
+import com.wwdz.ch.db.domain.User;
 import com.wwdz.ch.db.dto.request.InvitationCodeRequestDto;
 import com.wwdz.ch.db.dto.request.InviteRecordRequestDto;
 import com.wwdz.ch.wx.service.InviteRecordService;
@@ -17,6 +19,7 @@ import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 import java.util.Date;
 import java.util.List;
+import java.util.Objects;
 
 @Service
 public class InviteRecordServiceImpl implements InviteRecordService {
@@ -29,6 +32,9 @@ public class InviteRecordServiceImpl implements InviteRecordService {
     @Autowired
     InvitationCodeDao invitationCodeDao;
 
+    @Autowired
+    UserDao userDao;
+
     @Override
     public Result create(InviteRecordRequestDto dto) {
         try {
@@ -74,6 +80,13 @@ public class InviteRecordServiceImpl implements InviteRecordService {
             } else {
                 invitationCodeDao.updateState(dto.getInvitationCode(), InvitationCodeStateEnum.PART_USED.getCode());
             }
+
+            // 更新user表
+            User user = userDao.queryById(dto.getInvitee());
+            if (Objects.nonNull(user)) {
+                user.setShareUserId(invitationCode.getApplicant());
+                userDao.updateById(user);
+            }
             return Result.success();
         } catch (Exception e) {
             logger.error("邀请码使用失败:{}", e);