diff --git a/pkg/plug/algorithm.go b/pkg/plug/algorithm.go index 10b5bd4..1950c54 100644 --- a/pkg/plug/algorithm.go +++ b/pkg/plug/algorithm.go @@ -10,15 +10,15 @@ import ( ) type User struct { - userFavoriteCount int64 - userName string - userID string + UserFavoriteCount int64 + UserName string + UserID string } type Favorites struct { - posts []models.Post - nextPage string - lastPage string + Posts []models.Post + NextPage string + LastPage string } type Plug interface { @@ -70,7 +70,7 @@ func Algorithm(ctx context.Context, plugInterface Plug, db *gorm.DB, userSource nextPage := "" outer: - for anthroveUserFavCount < profile.userFavoriteCount { + for anthroveUserFavCount < profile.UserFavoriteCount { select { case <-ctx.Done(): break outer @@ -87,13 +87,13 @@ outer: return taskSummery, err } - if len(favorites.posts) <= 0 { + if len(favorites.Posts) <= 0 { span.AddEvent("No more favorites found") log.WithContext(ctx).WithFields(basicLoggingInfo).Info("No more favorites found") break outer } - summery, err := BatchPostProcessingWithSummery(ctx, userSource, favorites.posts) + summery, err := BatchPostProcessingWithSummery(ctx, userSource, favorites.Posts) if err != nil { span.RecordError(err) span.SetStatus(codes.Error, err.Error()) @@ -101,12 +101,12 @@ outer: return taskSummery, err } - if summery.AddedFavorites != int64(len(favorites.posts)) { + if summery.AddedFavorites != int64(len(favorites.Posts)) { span.AddEvent("user has no more favorites to add") break outer } - nextPage = favorites.nextPage + nextPage = favorites.NextPage taskSummery.AddedPosts += int(summery.AddedFavorites) } }