initial commit

database/admAccountStore.go

@@ -0,0 +1,84 @@
+package database
+
+import (
+	"errors"
+
+	"github.com/dhax/go-base/models"
+	"github.com/go-pg/pg"
+)
+
+var (
+	ErrUniqueEmailConstraint = errors.New("email already registered")
+)
+
+type AdmAccountStore struct {
+	db *pg.DB
+}
+
+func NewAdmAccountStore(db *pg.DB) *AdmAccountStore {
+	return &AdmAccountStore{
+		db: db,
+	}
+}
+
+func (s *AdmAccountStore) List(f models.AccountFilter) (*[]models.Account, int, error) {
+	var a []models.Account
+	count, err := s.db.Model(&a).
+		Apply(f.Filter).
+		SelectAndCount()
+	if err != nil {
+		return nil, 0, err
+	}
+	return &a, count, nil
+}
+
+func (s *AdmAccountStore) Create(a *models.Account) error {
+	count, _ := s.db.Model(a).
+		Where("email = ?email").
+		Count()
+
+	if count != 0 {
+		return ErrUniqueEmailConstraint
+	}
+
+	err := s.db.RunInTransaction(func(tx *pg.Tx) error {
+		err := tx.Insert(a)
+		if err != nil {
+			return err
+		}
+		p := &models.Profile{
+			AccountID: a.ID,
+		}
+		return tx.Insert(p)
+	})
+
+	return err
+}
+
+func (s *AdmAccountStore) Get(id int) (*models.Account, error) {
+	a := models.Account{ID: id}
+	err := s.db.Select(&a)
+	return &a, err
+}
+
+func (s *AdmAccountStore) Update(a *models.Account) error {
+	err := s.db.Update(a)
+	return err
+}
+
+func (s *AdmAccountStore) 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
+}