@@ -120,4 +120,6 @@ public class User extends BaseEntity implements Serializable {
private String invitorName;
private String invitorPhone;
+
+ private String inviteCode;
}
@@ -40,4 +40,6 @@ public class UserRegister {
@@ -163,4 +163,6 @@ public interface UserRepo extends JpaRepository<User, Long>, JpaSpecificationExe
@Modifying
@Query("update User u set u.sales = COALESCE(u.sales, 0) + ?2 where u.id = ?1")
public void increaseSales(Long id, int num);
+ List<User> findByInviteCode(String inviteCode);
@@ -145,6 +145,7 @@ public class UserService {
.phone(phone)
.invitorPhone(Optional.ofNullable(invite).map(Invite::getPhone).orElse(null))
.invitorName(Optional.ofNullable(invite).map(Invite::getName).orElse(null))
+ .inviteCode(Optional.ofNullable(invite).map(Invite::getCode).orElse(null))
.build());
if (invite != null) {
inviteRepo.increaseNum(invite.getId());