From 57eb2a5a67f2da322e8f428d199fad21483421b5 Mon Sep 17 00:00:00 2001
From: howell <2827207845@qq.com>
Date: Tue, 15 Feb 2022 15:59:54 +0800
Subject: [PATCH] fix: improve

---
 internal/service/app/register.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/internal/service/app/register.go b/internal/service/app/register.go
index 6069773..fd62498 100644
--- a/internal/service/app/register.go
+++ b/internal/service/app/register.go
@@ -35,6 +35,7 @@ func (r *registerUser) UserAdd(c *gin.Context, login *user.Login, info *user.Inf
 		}
 		level := 0
 		var parentID uint = 0
+		var rootID uint = 0
 		if info.InvitationNo != "" {
 			var shareUser user.Information
 			// uid := baseCode.Decode(info.InvitationNo)
@@ -47,6 +48,10 @@ func (r *registerUser) UserAdd(c *gin.Context, login *user.Login, info *user.Inf
 					level = 0
 					parentID = p.ID
 				}
+				if p.Level == 10 {
+					level = 2
+					rootID = p.ID
+				}
 			}
 
 		}
@@ -63,6 +68,7 @@ func (r *registerUser) UserAdd(c *gin.Context, login *user.Login, info *user.Inf
 			Identifier:   time.Now().Format("060102") + baseCode.Encode(uint64(lastId)),
 			Level:        level,
 			ParentID:     parentID,
+			RootID:       rootID,
 			InvitationNo: baseCode.Encode(uint64(lastId)),
 		}