mirror of
https://github.com/ForFarmTeam/ForFarm.git
synced 2025-12-19 22:14:08 +01:00
86 lines
2.8 KiB
Go
86 lines
2.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/forfarm/backend/internal/domain"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type postgresPlantRepository struct {
|
|
conn Connection
|
|
}
|
|
|
|
func NewPostgresPlant(conn Connection) domain.PlantRepository {
|
|
return &postgresPlantRepository{conn: conn}
|
|
}
|
|
|
|
func (p *postgresPlantRepository) fetch(ctx context.Context, query string, args ...interface{}) ([]domain.Plant, error) {
|
|
rows, err := p.conn.Query(ctx, query, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var plants []domain.Plant
|
|
for rows.Next() {
|
|
var plant domain.Plant
|
|
if err := rows.Scan(
|
|
&plant.UUID, &plant.Name, &plant.Variety,
|
|
&plant.RowSpacing, &plant.OptimalTemp, &plant.PlantingDepth,
|
|
&plant.AverageHeight, &plant.LightProfileID, &plant.SoilConditionID,
|
|
&plant.PlantingDetail, &plant.IsPerennial, &plant.DaysToEmerge,
|
|
&plant.DaysToFlower, &plant.DaysToMaturity, &plant.HarvestWindow,
|
|
&plant.PHValue, &plant.EstimateLossRate, &plant.EstimateRevenuePerHU,
|
|
&plant.HarvestUnitID, &plant.WaterNeeds, &plant.CreatedAt, &plant.UpdatedAt,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
plants = append(plants, plant)
|
|
}
|
|
return plants, nil
|
|
}
|
|
|
|
func (p *postgresPlantRepository) GetByUUID(ctx context.Context, uuid string) (domain.Plant, error) {
|
|
query := `SELECT * FROM plants WHERE uuid = $1`
|
|
plants, err := p.fetch(ctx, query, uuid)
|
|
if err != nil || len(plants) == 0 {
|
|
return domain.Plant{}, domain.ErrNotFound
|
|
}
|
|
return plants[0], nil
|
|
}
|
|
|
|
func (p *postgresPlantRepository) GetAll(ctx context.Context) ([]domain.Plant, error) {
|
|
query := `SELECT * FROM plants`
|
|
return p.fetch(ctx, query)
|
|
}
|
|
|
|
func (p *postgresPlantRepository) Create(ctx context.Context, plant *domain.Plant) error {
|
|
if strings.TrimSpace(plant.UUID) == "" {
|
|
plant.UUID = uuid.New().String()
|
|
}
|
|
if err := plant.Validate(); err != nil {
|
|
return err
|
|
}
|
|
query := `INSERT INTO plants (uuid, name, light_profile_id, soil_condition_id, harvest_unit_id, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, NOW(), NOW()) RETURNING created_at, updated_at`
|
|
return p.conn.QueryRow(ctx, query, plant.UUID, plant.Name, plant.LightProfileID, plant.SoilConditionID, plant.HarvestUnitID).Scan(&plant.CreatedAt, &plant.UpdatedAt)
|
|
}
|
|
|
|
func (p *postgresPlantRepository) Update(ctx context.Context, plant *domain.Plant) error {
|
|
if err := plant.Validate(); err != nil {
|
|
return err
|
|
}
|
|
query := `UPDATE plants SET name = $2, light_profile_id = $3, soil_condition_id = $4,
|
|
harvest_unit_id = $5, updated_at = NOW() WHERE uuid = $1`
|
|
_, err := p.conn.Exec(ctx, query, plant.UUID, plant.Name, plant.LightProfileID, plant.SoilConditionID, plant.HarvestUnitID)
|
|
return err
|
|
}
|
|
|
|
func (p *postgresPlantRepository) Delete(ctx context.Context, uuid string) error {
|
|
query := `DELETE FROM plants WHERE uuid = $1`
|
|
_, err := p.conn.Exec(ctx, query, uuid)
|
|
return err
|
|
}
|