athlete amount endpoint

controller/athlete_controller.go

@@ -12,6 +12,9 @@ import (
 
 func athleteController() {
 	router.GET("/athlete", getAthletes)
+
+	router.GET("/athlete/amount", getAthletesAmount)
+
 	router.GET("/athlete/:id", getAthlete)
 	router.GET("/athlete/name_year", getAthleteByNameAndYear)
 	router.GET("/athlete/alias_year", getAthleteByAliasAndYear)
@@ -39,6 +42,16 @@ func getAthletes(c *gin.Context) {
 	c.IndentedJSON(http.StatusOK, athletes)
 }
 
+func getAthletesAmount(c *gin.Context) {
+	starts, err := service.GetAthletesAmount()
+	if err != nil {
+		c.IndentedJSON(http.StatusInternalServerError, gin.H{"message": err.Error()})
+		return
+	}
+
+	c.IndentedJSON(http.StatusOK, starts)
+}
+
 func getAthletesByMeeting(c *gin.Context) {
 	id := c.Param("meet_id")
 	if id == "" {

service/athlete_service.go

@@ -60,6 +60,18 @@ func getAthletesByBsonDocumentWithOptions(d interface{}, fOps options.FindOption
 	return athletes, nil
 }
 
+func GetAthletesAmount() (int, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	opts := options.Count().SetHint("_id_")
+	count, err := athleteCollection.CountDocuments(ctx, bson.D{}, opts)
+	if err != nil {
+		return 0, err
+	}
+	return int(count), nil
+}
+
 func GetAthletes(paging Paging) ([]model.Athlete, error) {
 	return getAthletesByBsonDocumentWithOptions(
 		bson.M{