Trim the view count to 1024 to reduce traffic volume (#43)

Author: YaroslavPodorvanov

internal/controllers/stats_controller.go

@@ -16,6 +16,13 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+const (
+	// language=SVG
+	pixel = `<svg xmlns="http://www.w3.org/2000/svg" width="1" height="1">
+  <rect width="1" height="1" fill="white" />
+</svg>`
+)
+
 type StatsController struct {
 	userService  *services.UserService
 	statsService *services.StatsService
@@ -113,17 +120,17 @@ func (c *StatsController) GitHubDayWeekMonthTotalCountBadge(ctx *gin.Context) {
 		return
 	}
 
+	// Temporary solution to reduce traffic volume
+	if statsCount.HourCount > 1024 {
+		c.renderImage(ctx, []byte(pixel))
+
+		return
+	}
+
 	c.renderImage(ctx, []byte(tmv2.Badge(statsCount)))
 }
 
 func (c *StatsController) Pixel(ctx *gin.Context) {
-	const (
-		// language=SVG
-		pixel = `<svg xmlns="http://www.w3.org/2000/svg" width="1" height="1">
-  <rect width="1" height="1" fill="white" />
-</svg>`
-	)
-
 	done := c.increment(ctx, dbs.SocialProviderGithub)
 	if done {
 		return

internal/controllers/web_controller.go

@@ -75,6 +75,7 @@ func (c *WebController) Index(ctx *gin.Context) {
 				Name:                 user.Name,
 			},
 			ProfileViewsStats: tmv2.ProfileViewsStats{
+				HourCount:  stats.HourCount,
 				DayCount:   stats.DayCount,
 				WeekCount:  stats.WeekCount,
 				MonthCount: stats.MonthCount,

internal/models/stats.go

@@ -1,19 +1,22 @@
 package models
 
 type ProfileViewsStats struct {
+	HourCount  int64
 	DayCount   int64
 	WeekCount  int64
 	MonthCount int64
 	TotalCount int64
 }
 
 type DayWeekMonthViewsStats struct {
+	HourCount  int64
 	DayCount   int64
 	WeekCount  int64
 	MonthCount int64
 }
 
 func (s *DayWeekMonthViewsStats) Inc() {
+	s.HourCount += 1
 	s.DayCount += 1
 	s.WeekCount += 1
 	s.MonthCount += 1

internal/services/stats_service.go

@@ -76,6 +76,7 @@ func (s *StatsService) StatsCount(ctx context.Context, userID int64, increment b
 	}
 
 	return ProfileViewsStats{
+		HourCount:  stats.HourCount,
 		DayCount:   stats.DayCount,
 		WeekCount:  stats.WeekCount,
 		MonthCount: stats.MonthCount,
@@ -98,6 +99,7 @@ func (s *StatsService) Increment(ctx context.Context, userID int64) (err error)
 
 func (s *StatsService) UserDayWeekMonthViewsStatsMap(ctx context.Context, userIDs []int64, now time.Time) (map[int64]models.DayWeekMonthViewsStats, error) {
 	rows, err := s.repository.Queries().ProfileHourlyViewsStats(ctx, dbs.ProfileHourlyViewsStatsParams{
+		Hour:    now,
 		Day:     now.AddDate(0, 0, -1),
 		Week:    now.AddDate(0, 0, -7),
 		Month:   now.AddDate(0, -1, 0),
@@ -110,6 +112,7 @@ func (s *StatsService) UserDayWeekMonthViewsStatsMap(ctx context.Context, userID
 	result := make(map[int64]models.DayWeekMonthViewsStats, len(rows))
 	for _, row := range rows {
 		result[row.UserID] = models.DayWeekMonthViewsStats{
+			HourCount:  row.HourCount,
 			DayCount:   row.DayCount,
 			WeekCount:  row.WeekCount,
 			MonthCount: row.MonthCount,