66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
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
|
|
}
|