58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"go-server/internal/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type SupplementRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewSupplementRepository(db *gorm.DB) *SupplementRepository {
|
|
return &SupplementRepository{db: db}
|
|
}
|
|
|
|
func (r *SupplementRepository) GetAll(ctx context.Context) ([]*models.Supplement, error) {
|
|
var supplements []*models.Supplement
|
|
err := r.db.WithContext(ctx).
|
|
Preload("Nutrients").
|
|
Preload("SupplementNutrients").
|
|
Find(&supplements).
|
|
Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return supplements, nil
|
|
}
|
|
|
|
func (r *SupplementRepository) GetById(ctx context.Context, id string) (*models.Supplement, error) {
|
|
var supplement *models.Supplement
|
|
err := r.db.WithContext(ctx).
|
|
Preload("Nutrients").
|
|
Preload("SupplementNutrients").
|
|
First(&supplement, "id = ?", id).
|
|
Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return supplement, nil
|
|
}
|
|
|
|
func (r *SupplementRepository) GetDailySupplementsOverview(ctx context.Context) ([]*models.SupplementNutrientOverview, error) {
|
|
var supplementNutrientOverview []*models.SupplementNutrientOverview
|
|
err := r.db.WithContext(ctx).
|
|
Table("supplements").
|
|
Select("supplements.id as supplement_id, supplements.name as supplement_name, supplements.description as supplement_description, supplement_nutrients.serving_size as serving_size, supplement_nutrients.per_serving as per_serving, supplement_nutrients.per_serving_reference_intake as per_serving_reference_intake, nutrients.id as nutrient_id, nutrients.name as nutrient_name, nutrients.description as nutrient_description").
|
|
Joins("INNER JOIN supplement_nutrients ON supplements.id = supplement_nutrients.supplement_id").
|
|
Joins("JOIN nutrients ON supplement_nutrients.nutrient_id = nutrients.id").
|
|
Find(&supplementNutrientOverview).
|
|
Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return supplementNutrientOverview, nil
|
|
}
|