60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package gorm
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gitea.neatflow.kr/biosvos/seven-skies/internal/pkg/domain"
|
|
"gitea.neatflow.kr/biosvos/seven-skies/internal/pkg/repository"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var _ repository.Repository = (*Repository)(nil)
|
|
|
|
type Repository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewRepository(dsn string) (*Repository, error) {
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{}) //nolint:exhaustruct
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
|
|
err = db.AutoMigrate(&Weather{}) //nolint:exhaustruct
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to migrate database: %w", err)
|
|
}
|
|
|
|
return &Repository{
|
|
db: db,
|
|
}, nil
|
|
}
|
|
|
|
func (r *Repository) CreateWeather(ctx context.Context, weather *domain.Weather) error {
|
|
modelWeather := newModelWeather(weather)
|
|
|
|
err := gorm.G[Weather](r.db).Create(ctx, modelWeather)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create weather: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *Repository) ListWeathers(ctx context.Context, from, to time.Time) ([]*domain.Weather, error) {
|
|
weathers, err := gorm.G[*Weather](r.db).Where("target_date BETWEEN ? AND ?", from, to).Find(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to list weathers: %w", err)
|
|
}
|
|
|
|
var ret []*domain.Weather
|
|
for _, weather := range weathers {
|
|
ret = append(ret, weather.toDomain())
|
|
}
|
|
|
|
return ret, nil
|
|
}
|