58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package leads
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// GormStore : Store adossé à PostgreSQL via GORM.
|
|
type GormStore struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewGormStore(db *gorm.DB) *GormStore {
|
|
return &GormStore{db: db}
|
|
}
|
|
|
|
func (s *GormStore) Create(l Lead) (Lead, error) {
|
|
l.ID = uuid.NewString()
|
|
l.Status = StatusNew
|
|
l.CreatedAt = time.Now().UTC()
|
|
if err := s.db.Create(&l).Error; err != nil {
|
|
return Lead{}, err
|
|
}
|
|
return l, nil
|
|
}
|
|
|
|
func (s *GormStore) List() ([]Lead, error) {
|
|
var out []Lead
|
|
if err := s.db.Order("created_at DESC").Find(&out).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (s *GormStore) Get(id string) (Lead, bool) {
|
|
var l Lead
|
|
err := s.db.Where("id = ?", id).First(&l).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) || err != nil {
|
|
return Lead{}, false
|
|
}
|
|
return l, true
|
|
}
|
|
|
|
func (s *GormStore) SetStatus(id string, status Status) (Lead, bool) {
|
|
var l Lead
|
|
if err := s.db.Where("id = ?", id).First(&l).Error; err != nil {
|
|
return Lead{}, false
|
|
}
|
|
l.Status = status
|
|
if err := s.db.Save(&l).Error; err != nil {
|
|
return Lead{}, false
|
|
}
|
|
return l, true
|
|
}
|