initial commit

database/accountStore.go

@@ -0,0 +1,66 @@
+package database
+
+import (
+	"github.com/dhax/go-base/models"
+	"github.com/go-pg/pg"
+)
+
+type AccountStore struct {
+	db *pg.DB
+}
+
+func NewAccountStore(db *pg.DB) *AccountStore {
+	return &AccountStore{
+		db: db,
+	}
+}
+
+func (s *AccountStore) Get(id int) (*models.Account, error) {
+	a := models.Account{ID: id}
+	err := s.db.Model(&a).
+		Where("account.id = ?id").
+		Column("account.*", "Profile", "Token").
+		First()
+	return &a, err
+}
+
+func (s *AccountStore) Update(a *models.Account) error {
+	_, err := s.db.Model(a).
+		Column("email", "name").
+		Update()
+	return err
+}
+
+func (s *AccountStore) Delete(a *models.Account) error {
+	err := s.db.RunInTransaction(func(tx *pg.Tx) error {
+		if _, err := tx.Model(&models.Token{}).
+			Where("account_id = ?", a.ID).
+			Delete(); err != nil {
+			return err
+		}
+		if _, err := tx.Model(&models.Profile{}).
+			Where("account_id = ?", a.ID).
+			Delete(); err != nil {
+			return err
+		}
+		return tx.Delete(a)
+	})
+	return err
+}
+
+func (s *AccountStore) UpdateToken(t *models.Token) error {
+	_, err := s.db.Model(t).
+		Column("identifier").
+		Update()
+	return err
+}
+
+func (s *AccountStore) DeleteToken(t *models.Token) error {
+	err := s.db.Delete(t)
+	return err
+}
+
+func (s *AccountStore) UpdateProfile(p *models.Profile) error {
+	err := s.db.Update(p)
+	return err
+}