thermokarst/bactdb
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ember data: convey strain/meas. relationships

Browse source
This commit is contained in:
Matthew Dillon 2015-01-26 11:39:26 -09:00
parent 4be150f897
commit af515e3e3c
8 changed files with 55 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

4
datastore/measurements.go
View file

@ -1,7 +1,6 @@
package datastore
import (
"fmt"
"strings"
"time"
@ -59,8 +58,7 @@ func (s *measurementsStore) List(opt *models.MeasurementListOptions) ([]*models.
sql += " WHERE (" + strings.Join(conds, ") AND (") + ")"
}
sql += fmt.Sprintf(" LIMIT $%v OFFSET $%v;", len(conds)+1, len(conds)+2)
vals = append(vals, opt.PerPageOrDefault(), opt.Offset())
sql += ";"
var measurements []*models.Measurement
err := s.dbh.Select(&measurements, sql, vals...)

22
datastore/strains.go
View file

@ -1,7 +1,6 @@
package datastore
import (
"fmt"
"strings"
"time"
@ -9,7 +8,7 @@ import (
)
func init() {
DB.AddTableWithName(models.Strain{}, "strains").SetKeys(true, "Id")
DB.AddTableWithName(models.StrainBase{}, "strains").SetKeys(true, "Id")
}
type strainsStore struct {
@ -18,7 +17,8 @@ type strainsStore struct {
func (s *strainsStore) Get(id int64) (*models.Strain, error) {
var strain models.Strain
if err := s.dbh.SelectOne(&strain, `SELECT * FROM strains WHERE id=$1;`, id); err != nil {
err := s.dbh.SelectOne(&strain, `SELECT s.*, array_agg(m.id) AS measurements FROM strains s LEFT OUTER JOIN measurements m ON m.strain_id=s.id WHERE s.id=$1 GROUP BY s.id;`, id)
if err != nil {
return nil, err
}
if &strain == nil {
@ -31,9 +31,11 @@ func (s *strainsStore) Create(strain *models.Strain) (bool, error) {
currentTime := time.Now()
strain.CreatedAt = currentTime
strain.UpdatedAt = currentTime
if err := s.dbh.Insert(strain); err != nil {
base := strain.StrainBase
if err := s.dbh.Insert(base); err != nil {
return false, err
}
strain.Id = base.Id
return true, nil
}
@ -42,13 +44,13 @@ func (s *strainsStore) List(opt *models.StrainListOptions) ([]*models.Strain, er
opt = &models.StrainListOptions{}
}
sql := `SELECT * FROM strains`
sql := `SELECT s.*, array_agg(m.id) AS measurements FROM strains s LEFT OUTER JOIN measurements m ON m.strain_id=s.id`
var conds []string
var vals []interface{}
if opt.Genus != "" {
conds = append(conds, `species_id IN (SELECT s.id FROM species s
conds = append(conds, `s.species_id IN (SELECT s.id FROM species s
INNER JOIN genera g ON g.id = s.genus_id WHERE lower(g.genus_name) = $1)`)
vals = append(vals, opt.Genus)
}
@ -57,8 +59,7 @@ func (s *strainsStore) List(opt *models.StrainListOptions) ([]*models.Strain, er
sql += " WHERE (" + strings.Join(conds, ") AND (") + ")"
}
sql += fmt.Sprintf(" LIMIT $%v OFFSET $%v;", len(conds)+1, len(conds)+2)
vals = append(vals, opt.PerPageOrDefault(), opt.Offset())
sql += " GROUP BY s.id;"
var strains []*models.Strain
err := s.dbh.Select(&strains, sql, vals...)
@ -79,7 +80,8 @@ func (s *strainsStore) Update(id int64, strain *models.Strain) (bool, error) {
}
strain.UpdatedAt = time.Now()
changed, err := s.dbh.Update(strain)
changed, err := s.dbh.Update(strain.StrainBase)
if err != nil {
return false, err
}
@ -97,7 +99,7 @@ func (s *strainsStore) Delete(id int64) (bool, error) {
return false, err
}
deleted, err := s.dbh.Delete(strain)
deleted, err := s.dbh.Delete(strain.StrainBase)
if err != nil {
return false, err
}

15
datastore/strains_test.go
View file

@ -12,17 +12,17 @@ import (
func insertStrain(t *testing.T, tx *modl.Transaction) *models.Strain {
// clean up our target table
tx.Exec(`DELETE FROM strains;`)
strain := newStrain(t, tx)
if err := tx.Insert(strain); err != nil {
s := newStrain(t, tx)
if err := tx.Insert(s); err != nil {
t.Fatal(err)
}
return strain
return &models.Strain{s, []int64(nil)}
}
func newStrain(t *testing.T, tx *modl.Transaction) *models.Strain {
func newStrain(t *testing.T, tx *modl.Transaction) *models.StrainBase {
// we want to create and insert a species (and genus) record too
species := insertSpecies(t, tx)
return &models.Strain{
return &models.StrainBase{
SpeciesId: species.Id,
StrainName: "Test Strain",
StrainType: "Test Type",
@ -67,11 +67,12 @@ func TestStrainsStore_Create_db(t *testing.T) {
tx, _ := DB.Begin()
defer tx.Rollback()
strain := newStrain(t, tx)
base_strain := newStrain(t, tx)
strain := models.Strain{base_strain, []int64(nil)}
d := NewDatastore(tx)
created, err := d.Strains.Create(strain)
created, err := d.Strains.Create(&strain)
if err != nil {
t.Fatal(err)
}