xiongzhu před 4 roky
rodič
revize
adc182b887

+ 2 - 0
src/main/java/com/izouma/nineth/domain/User.java

@@ -120,4 +120,6 @@ public class User extends BaseEntity implements Serializable {
     private String invitorName;
 
     private String invitorPhone;
+
+    private String inviteCode;
 }

+ 2 - 0
src/main/java/com/izouma/nineth/dto/UserRegister.java

@@ -40,4 +40,6 @@ public class UserRegister {
     private String invitorName;
 
     private String invitorPhone;
+
+    private String inviteCode;
 }

+ 2 - 0
src/main/java/com/izouma/nineth/repo/UserRepo.java

@@ -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);
 }

+ 1 - 0
src/main/java/com/izouma/nineth/service/UserService.java

@@ -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());