feat: implement events

Author: pan93412

internal/events/README.md

@@ -0,0 +1,3 @@
+# Events
+
+負責觸發事件和加減點數的 service。

internal/events/constants.go

@@ -0,0 +1,7 @@
+package events
+
+type EventType string
+
+const (
+	EventTypeLogin EventType = "login"
+)

internal/events/events.go

@@ -0,0 +1,55 @@
+package events
+
+import (
+	"context"
+	"log/slog"
+	"time"
+
+	"github.com/database-playground/backend-v2/ent"
+)
+
+// EventService is the service for triggering events.
+type EventService struct {
+	entClient     *ent.Client
+	pointsGranter *PointsGranter
+}
+
+// NewEventService creates a new EventService.
+func NewEventService(entClient *ent.Client) *EventService {
+	return &EventService{
+		entClient:     entClient,
+		pointsGranter: NewPointsGranter(entClient),
+	}
+}
+
+// Event is the event to be triggered.
+type Event struct {
+	Type    EventType
+	Payload map[string]any
+	UserID  int
+}
+
+// TriggerEvent triggers an event.
+func (s *EventService) TriggerEvent(ctx context.Context, event Event) error {
+	err := s.entClient.Events.Create().
+		SetType(string(event.Type)).
+		SetPayload(event.Payload).
+		SetUserID(event.UserID).
+		SetTriggeredAt(time.Now()).
+		Exec(ctx)
+	if err != nil {
+		return err
+	}
+
+	if event.Type == EventTypeLogin {
+		ok, err := s.pointsGranter.GrantDailyLoginPoints(ctx, event.UserID)
+		if err != nil {
+			return err
+		}
+		if ok {
+			slog.Info("granted daily login points", "user_id", event.UserID)
+		}
+	}
+
+	return nil
+}

internal/events/points.go

@@ -0,0 +1,118 @@
+package events
+
+import (
+	"context"
+	"time"
+
+	"github.com/database-playground/backend-v2/ent"
+	"github.com/database-playground/backend-v2/ent/events"
+	"github.com/database-playground/backend-v2/ent/points"
+	"github.com/database-playground/backend-v2/ent/user"
+)
+
+const (
+	PointDescriptionDailyLogin  = "daily login"
+	PointDescriptionWeeklyLogin = "weekly login"
+)
+
+const (
+	PointValueDailyLogin  = 20
+	PointValueWeeklyLogin = 50
+)
+
+// PointsGranter determines if the criteria is met to grant points to a user.
+type PointsGranter struct {
+	entClient *ent.Client
+}
+
+// NewPointsGranter creates a new PointsGranter.
+func NewPointsGranter(entClient *ent.Client) *PointsGranter {
+	return &PointsGranter{
+		entClient: entClient,
+	}
+}
+
+// GrantDailyLoginPoints grants the "daily login" points to a user.
+func (d *PointsGranter) GrantDailyLoginPoints(ctx context.Context, userID int) (bool, error) {
+	// Check if we have granted the "daily login" points for this user today.
+	hasPointsRecord, err := d.entClient.Points.Query().
+		Where(points.HasUserWith(user.ID(userID))).
+		Where(points.DescriptionEQ(PointDescriptionDailyLogin)).
+		Where(points.CreatedAtGTE(time.Now().AddDate(0, 0, -1))).Exist(ctx)
+	if err != nil {
+		return false, err
+	}
+	if hasPointsRecord {
+		return false, nil
+	}
+
+	// Check if the user has logged in today.
+	hasTodayLoginRecord, err := d.entClient.Events.Query().
+		Where(events.Type(string(EventTypeLogin))).
+		Where(events.UserID(userID)).
+		Where(events.TriggeredAtGTE(time.Now().AddDate(0, 0, -1))).
+		Exist(ctx)
+	if err != nil {
+		return false, err
+	}
+	if !hasTodayLoginRecord {
+		return false, nil
+	}
+
+	// Grant the "daily login" points to the user.
+	err = d.entClient.Points.Create().
+		SetUserID(userID).
+		SetDescription(PointDescriptionDailyLogin).
+		SetPoints(PointValueDailyLogin).
+		Exec(ctx)
+	if err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
+// GrantWeeklyLoginPoints grants the "weekly login" points to a user.
+func (d *PointsGranter) GrantWeeklyLoginPoints(ctx context.Context, userID int) (bool, error) {
+	// Check if we have granted the "weekly login" points for this user this week.
+	hasPointsRecord, err := d.entClient.Points.Query().
+		Where(points.HasUserWith(user.ID(userID))).
+		Where(points.DescriptionEQ(PointDescriptionWeeklyLogin)).
+		Where(points.CreatedAtGTE(time.Now().AddDate(0, 0, -7))).Exist(ctx)
+	if err != nil {
+		return false, err
+	}
+	if hasPointsRecord {
+		return false, nil
+	}
+
+	// Check if the user has logged in every day this week.
+	weekLoginRecords, err := d.entClient.Events.Query().
+		Where(events.Type(string(EventTypeLogin))).
+		Where(events.UserID(userID)).
+		All(ctx)
+	if err != nil {
+		return false, err
+	}
+
+	// aggreated by day
+	weekLoginRecordsByDay := make(map[time.Time]int)
+	for _, record := range weekLoginRecords {
+		weekLoginRecordsByDay[record.TriggeredAt.Truncate(24*time.Hour)]++
+	}
+
+	if len(weekLoginRecordsByDay) != 7 {
+		return false, nil
+	}
+
+	// Grant the "weekly login" points to the user.
+	err = d.entClient.Points.Create().
+		SetUserID(userID).
+		SetDescription(PointDescriptionWeeklyLogin).
+		SetPoints(PointValueWeeklyLogin).
+		Exec(ctx)
+	if err != nil {
+		return false, err
+	}
+	return true, nil
+}