bactdb/models/species.go

package models

import (
	"errors"
	"net/http"
	"strconv"
	"time"

	"github.com/lib/pq"
	"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:"genusId"`
	SpeciesName string      `db:"species_name" json:"speciesName"`
	CreatedAt   time.Time   `db:"created_at" json:"createdAt"`
	UpdatedAt   time.Time   `db:"updated_at" json:"updatedAt"`
	DeletedAt   pq.NullTime `db:"deleted_at" json:"deletedAt"`
}

func NewSpecies() *Species {
	return &Species{SpeciesName: "Test Species"}
}

// SpeciesService interacts with the species-related endpoints in bactdb's API.
type SpeciesService interface {
	// Get a species
	Get(id int64) (*Species, error)

	// List all species
	List(opt *SpeciesListOptions) ([]*Species, error)

	// Create a species record
	Create(species *Species) (bool, error)

	// Update an existing species
	Update(id int64, species *Species) (updated bool, err error)

	// Delete an existing species
	Delete(id int64) (deleted bool, err 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 SpeciesListOptions struct {
	ListOptions
}

func (s *speciesService) List(opt *SpeciesListOptions) ([]*Species, error) {
	url, err := s.client.url(router.SpeciesList, nil, opt)
	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) Update(id int64, species *Species) (bool, error) {
	strId := strconv.FormatInt(id, 10)

	url, err := s.client.url(router.UpdateSpecies, map[string]string{"Id": strId}, nil)
	if err != nil {
		return false, err
	}

	req, err := s.client.NewRequest("PUT", 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.StatusOK, nil
}

func (s *speciesService) Delete(id int64) (bool, error) {
	strId := strconv.FormatInt(id, 10)

	url, err := s.client.url(router.DeleteSpecies, map[string]string{"Id": strId}, nil)
	if err != nil {
		return false, err
	}

	req, err := s.client.NewRequest("DELETE", url.String(), nil)
	if err != nil {
		return false, err
	}

	var species *Species
	resp, err := s.client.Do(req, &species)
	if err != nil {
		return false, err
	}

	return resp.StatusCode == http.StatusOK, nil
}

type MockSpeciesService struct {
	Get_    func(id int64) (*Species, error)
	List_   func(opt *SpeciesListOptions) ([]*Species, error)
	Create_ func(species *Species) (bool, error)
	Update_ func(id int64, species *Species) (bool, error)
	Delete_ func(id int64) (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)
}

func (s *MockSpeciesService) List(opt *SpeciesListOptions) ([]*Species, error) {
	if s.List_ == nil {
		return nil, nil
	}
	return s.List_(opt)
}

func (s *MockSpeciesService) Update(id int64, species *Species) (bool, error) {
	if s.Update_ == nil {
		return false, nil
	}
	return s.Update_(id, species)
}

func (s *MockSpeciesService) Delete(id int64) (bool, error) {
	if s.Delete_ == nil {
		return false, nil
	}
	return s.Delete_(id)
}