100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/thermokarst/bactdb/router"
|
|
)
|
|
|
|
// A Species is a high-level classifier in bactdb.
|
|
type Species struct {
|
|
Id int64 `json:"id,omitempty"`
|
|
GenusId int64 `db:"genus_id" json:"genus_id"`
|
|
SpeciesName string `db:"species_name" json:"species_name"`
|
|
CreatedAt time.Time `db:"created_at" json:"created_at"`
|
|
UpdatedAt time.Time `db:"updated_at" json:"updated_at"`
|
|
DeletedAt time.Time `db:"deleted_at" json:"deleted_at"`
|
|
}
|
|
|
|
// SpeciesService interacts with the species-related endpoints in bactdb's API.
|
|
type SpeciesService interface {
|
|
// Get a species
|
|
Get(id int64) (*Species, error)
|
|
|
|
// Create a species record
|
|
Create(species *Species) (bool, error)
|
|
}
|
|
|
|
var (
|
|
ErrSpeciesNotFound = errors.New("species not found")
|
|
)
|
|
|
|
type speciesService struct {
|
|
client *Client
|
|
}
|
|
|
|
func (s *speciesService) Get(id int64) (*Species, error) {
|
|
// Pass in key value pairs as strings, sp that the gorilla mux URL generation is happy
|
|
strId := strconv.FormatInt(id, 10)
|
|
|
|
url, err := s.client.url(router.Species, map[string]string{"Id": strId}, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
req, err := s.client.NewRequest("GET", url.String(), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var species *Species
|
|
_, err = s.client.Do(req, &species)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return species, nil
|
|
}
|
|
|
|
func (s *speciesService) Create(species *Species) (bool, error) {
|
|
url, err := s.client.url(router.CreateSpecies, nil, nil)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
req, err := s.client.NewRequest("POST", url.String(), species)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
resp, err := s.client.Do(req, &species)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return resp.StatusCode == http.StatusCreated, nil
|
|
}
|
|
|
|
type MockSpeciesService struct {
|
|
Get_ func(id int64) (*Species, error)
|
|
Create_ func(species *Species) (bool, error)
|
|
}
|
|
|
|
var _ SpeciesService = &MockSpeciesService{}
|
|
|
|
func (s *MockSpeciesService) Get(id int64) (*Species, error) {
|
|
if s.Get_ == nil {
|
|
return nil, nil
|
|
}
|
|
return s.Get_(id)
|
|
}
|
|
|
|
func (s *MockSpeciesService) Create(species *Species) (bool, error) {
|
|
if s.Create_ == nil {
|
|
return false, nil
|
|
}
|
|
return s.Create_(species)
|
|
}
|