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)