prepare for team follow

SwimResults · Apr 17, 2024 · 5d0452e · 5d0452e
1 parent 0db72f3
commit 5d0452e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/model/following.go b/model/following.go
@@ -5,7 +5,19 @@ import (
 	"time"
 )
 
-type Following struct {
+type FollowingAthlete struct {
 	AthleteId primitive.ObjectID `json:"athlete_id" bson:"athlete_id"`
 	AddedAt   time.Time          `json:"added_at,omitempty" bson:"added_at,omitempty"`
 }
+
+// FollowingTeam TODO make teams followable
+type FollowingTeam struct {
+	TeamId  primitive.ObjectID `json:"team_id" bson:"team_id"`
+	AddedAt time.Time          `json:"added_at,omitempty" bson:"added_at,omitempty"`
+}
+
+// Following TODO use merged struct
+type Following struct {
+	FollowingAthletes []FollowingAthlete `json:"following_athletes,omitempty" bson:"following_athletes,omitempty"`
+	FollowingTeams    []FollowingTeam    `json:"following_teams,omitempty" bson:"following_teams,omitempty"`
+}
diff --git a/model/user.go b/model/user.go
@@ -7,7 +7,7 @@ import (
 type User struct {
 	Identifier   primitive.ObjectID  `json:"_id,omitempty" bson:"_id,omitempty"`
 	KeycloakId   string              `json:"keycloak_id,omitempty" bson:"keycloak_id,omitempty"`
-	Following    []Following         `json:"following,omitempty" bson:"following,omitempty"`
+	Following    []FollowingAthlete  `json:"following,omitempty" bson:"following,omitempty"`
 	OwnAthleteId *primitive.ObjectID `json:"own_athlete_id,omitempty" bson:"own_athlete_id,omitempty"`
 	Settings     Settings            `json:"settings,omitempty" bson:"settings,omitempty"`
 	Meetings     []string            `json:"meetings,omitempty" bson:"meetings,omitempty"`

diff --git a/service/user_service.go b/service/user_service.go
@@ -127,7 +127,7 @@ func ModifyFollowForUser(id uuid.UUID, athleteId primitive.ObjectID, follow bool
 	}
 
 	if follow {
-		user.Following = append(user.Following, model.Following{
+		user.Following = append(user.Following, model.FollowingAthlete{
 			AthleteId: athleteId,
 			AddedAt:   time.Now(),
 		})