implement count

Author: miladrahimi

internal/http/server/handlers/v1/feed.go

@@ -6,6 +6,7 @@ import (
 	"github.com/neatplex/nightel-core/internal/services/container"
 	"github.com/neatplex/nightel-core/internal/utils"
 	"net/http"
+	"strconv"
 )
 
 func Feed(ctr *container.Container) echo.HandlerFunc {
@@ -24,7 +25,16 @@ func Feed(ctr *container.Container) echo.HandlerFunc {
 			}
 		}
 
-		stories, err := ctr.StoryService.Feed(user.ID, lastId)
+		count := 10
+		requestCount := ctx.QueryParams().Get("count")
+		if requestCount != "" {
+			parsedRequestCount, _ := strconv.Atoi(requestCount)
+			if parsedRequestCount > 0 && parsedRequestCount < 100 {
+				count = parsedRequestCount
+			}
+		}
+
+		stories, err := ctr.StoryService.Feed(user.ID, lastId, count)
 		if err != nil {
 			return err
 		}

internal/services/story/service.go

@@ -29,10 +29,10 @@ func (s *Service) Index(userId uint64) ([]*models.Story, error) {
 	return stories, nil
 }
 
-func (s *Service) Feed(userId uint64, lastId uint64) ([]*models.Story, error) {
+func (s *Service) Feed(userId uint64, lastId uint64, count int) ([]*models.Story, error) {
 	var stories []*models.Story
 	r := s.database.Handler().
-		Where("id < ? ORDER BY id DESC LIMIT 3", lastId).
+		Where("id < ? ORDER BY id DESC LIMIT ?", lastId, count).
 		Preload("Audio").
 		Preload("Image").
 		Find(&stories)