Rough in species

This commit is contained in:
Matthew Dillon 2015-06-02 12:27:42 -08:00
parent e2ba99c511
commit 0ceaccf0e4
3 changed files with 133 additions and 1 deletions

View file

@ -92,6 +92,8 @@ func Handler() http.Handler {
r{handleGetter(MeasurementService{}), "GET", "/measurements/{Id:.+}"}, r{handleGetter(MeasurementService{}), "GET", "/measurements/{Id:.+}"},
r{handleLister(CharacteristicService{}), "GET", "/characteristics"}, r{handleLister(CharacteristicService{}), "GET", "/characteristics"},
r{handleGetter(CharacteristicService{}), "GET", "/characteristics/{Id:.+}"}, r{handleGetter(CharacteristicService{}), "GET", "/characteristics/{Id:.+}"},
r{handleLister(SpeciesService{}), "GET", "/species"},
r{handleGetter(SpeciesService{}), "GET", "/species/{Id:.+}"},
} }
for _, route := range routes { for _, route := range routes {

View file

@ -13,9 +13,16 @@ CREATE TABLE species (
updated_at TIMESTAMP WITH TIME ZONE NOT NULL, updated_at TIMESTAMP WITH TIME ZONE NOT NULL,
deleted_at TIMESTAMP WITH TIME ZONE NULL, deleted_at TIMESTAMP WITH TIME ZONE NULL,
created_by BIGINT NOT NULL,
updated_by BIGINT NOT NULL,
deleted_by BIGINT NULL,
CONSTRAINT species_pkey PRIMARY KEY (id), CONSTRAINT species_pkey PRIMARY KEY (id),
FOREIGN KEY (genus_id) REFERENCES genera(id), FOREIGN KEY (genus_id) REFERENCES genera(id),
FOREIGN KEY (subspecies_species_id) REFERENCES species(id) FOREIGN KEY (subspecies_species_id) REFERENCES species(id),
FOREIGN KEY (created_by) REFERENCES users(id),
FOREIGN KEY (updated_by) REFERENCES users(id),
FOREIGN KEY (deleted_by) REFERENCES users(id)
); );
CREATE INDEX genus_id_idx ON species (genus_id); CREATE INDEX genus_id_idx ON species (genus_id);

123
species.go Normal file
View file

@ -0,0 +1,123 @@
package main
import (
"database/sql"
"encoding/json"
"errors"
"fmt"
"strings"
)
var (
ErrSpeciesNotFound = errors.New("species not found")
ErrSpeciesNotUpdated = errors.New("species not updated")
)
func init() {
DB.AddTableWithName(SpeciesBase{}, "species").SetKeys(true, "Id")
}
type SpeciesService struct{}
// SpeciesBase is what the DB expects to see for inserts/updates
type SpeciesBase struct {
Id int64 `db:"id" json:"id"`
SpeciesName string `db:"species_name" json:"speciesName"`
TypeSpecies NullBool `db:"type_species" json:"typeSpecies"`
Etymology NullString `db:"etymology" json:"etymology"`
CreatedAt NullTime `db:"created_at" json:"createdAt"`
UpdatedAt NullTime `db:"updated_at" json:"updatedAt"`
DeletedAt NullTime `db:"deleted_at" json:"deletedAt"`
CreatedBy int64 `db:"created_by" json:"createdBy"`
UpdatedBy int64 `db:"updated_by" json:"updatedBy"`
DeletedBy NullInt64 `db:"deleted_by" json:"deletedBy"`
}
// Species & SpeciesJSON(s) are what ember expects to see
type Species struct {
*SpeciesBase
GenusName string `db:"genus_name" json:"genusName"`
Strains NullSliceInt64 `db:"strains" json:"strains"`
TotalStrains int64 `db:"total_strains" json:"totalStrains"`
}
type ManySpecies []*Species
type SpeciesJSON struct {
Species *Species `json:"species"`
}
type ManySpeciesJSON struct {
ManySpecies *ManySpecies `json:"species"`
}
func (s *Species) marshal() ([]byte, error) {
return json.Marshal(&SpeciesJSON{Species: s})
}
func (s *ManySpecies) marshal() ([]byte, error) {
return json.Marshal(&ManySpeciesJSON{ManySpecies: s})
}
func (s SpeciesService) unmarshal(b []byte) (entity, error) {
var sj SpeciesJSON
err := json.Unmarshal(b, &sj)
return sj.Species, err
}
func (s SpeciesService) list(opt *ListOptions) (entity, error) {
if opt == nil {
return nil, errors.New("must provide options")
}
var vals []interface{}
sql := `SELECT sp.id, sp.species_name, sp.type_species, sp.etymology,
sp.created_at, sp.created_by, sp.updated_at, sp.updated_by,
sp.deleted_at, sp.deleted_by, g.genus_name, array_agg(st.id) AS strains,
COUNT(st) AS total_strains
FROM species sp
INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=$1
LEFT OUTER JOIN strains st ON st.species_id=sp.id`
vals = append(vals, opt.Genus)
if len(opt.Ids) != 0 {
var conds []string
s := "st.id IN ("
for i, id := range opt.Ids {
s = s + fmt.Sprintf("$%v,", i+2) // start param index at 2
vals = append(vals, id)
}
s = s[:len(s)-1] + ")"
conds = append(conds, s)
sql += " WHERE (" + strings.Join(conds, ") AND (") + ")"
}
sql += " GROUP BY sp.id, g.genus_name;"
var species ManySpecies
err := DBH.Select(&species, sql, vals...)
if err != nil {
return nil, err
}
return &species, nil
}
func (s SpeciesService) get(id int64, genus string) (entity, error) {
var species Species
q := `SELECT sp.id, sp.species_name, sp.type_species, sp.etymology,
sp.created_at, sp.created_by, sp.updated_at, sp.updated_by, sp.deleted_at,
sp.deleted_by, g.genus_name, array_agg(st.id) AS strains,
COUNT(st) AS total_strains
FROM species sp
INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=$1
LEFT OUTER JOIN strains st ON st.species_id=sp.id
WHERE sp.id=$2
GROUP BY sp.id, g.genus_name;`
if err := DBH.SelectOne(&species, q, genus, id); err != nil {
if err == sql.ErrNoRows {
return nil, ErrStrainNotFound
}
return nil, err
}
return &species, nil
}