From c453f9e9b87381c127fd639aab2ebcdac7cc6b5f Mon Sep 17 00:00:00 2001 From: Leufolium Date: Tue, 24 Sep 2024 11:09:00 +0800 Subject: [PATCH] by Robin at 20240924 --- importfunc/import_user_id_map.go | 24 +++++++++++++++--------- 1 file changed, 15 insertions(+), 9 deletions(-) diff --git a/importfunc/import_user_id_map.go b/importfunc/import_user_id_map.go index 14c432c..6b434f5 100644 --- a/importfunc/import_user_id_map.go +++ b/importfunc/import_user_id_map.go @@ -37,9 +37,22 @@ func ImportUserIdMap() { if len(str) == 0 { continue } + if err == io.EOF { + err := mcli.CreateMappedUserIds(ctx, userIdMaps) + if err != nil { + fmt.Printf("CreateMappedUserIds err :%v", err) + } + break + } strs := strings.Split(str, ":") - seq, _ := strconv.Atoi(strs[0]) - userId, _ := strconv.Atoi(strs[1]) + seq, err := strconv.Atoi(strs[0]) + if err != nil { + fmt.Printf("Atoi err :%v", err) + } + userId, err := strconv.Atoi(strs[1]) + if err != nil { + fmt.Printf("Atoi err :%v", err) + } userIdMaps = append(userIdMaps, &dbstruct.UserIdMap{ Seq: int64(seq), UserId: int64(userId), @@ -51,13 +64,6 @@ func ImportUserIdMap() { } userIdMaps = make([]*dbstruct.UserIdMap, 0) } - if err == io.EOF { - err := mcli.CreateMappedUserIds(ctx, userIdMaps) - if err != nil { - fmt.Printf("CreateMappedUserIds err :%v", err) - } - break - } } infile.Close() fmt.Printf("%dth imported", i)