1
0
mirror of https://github.com/mxpv/podsync.git synced 2024-05-11 05:55:04 +00:00

Rename models package

This commit is contained in:
Maksym Pavlenko
2017-11-02 21:41:52 -07:00
parent dcef9b2373
commit 2912b0f36a
4 changed files with 19 additions and 19 deletions

View File

@@ -1,4 +1,4 @@
package models
package model
import "time"

View File

@@ -8,7 +8,7 @@ import (
"github.com/go-pg/pg"
"github.com/mxpv/patreon-go"
"github.com/mxpv/podsync/pkg/api"
"github.com/mxpv/podsync/pkg/models"
"github.com/mxpv/podsync/pkg/model"
"github.com/pkg/errors"
)
@@ -20,7 +20,7 @@ type Patreon struct {
db *pg.DB
}
func (h Patreon) toModel(pledge *patreon.Pledge) (*models.Pledge, error) {
func (h Patreon) toModel(pledge *patreon.Pledge) (*model.Pledge, error) {
pledgeID, err := strconv.ParseInt(pledge.ID, 10, 64)
if err != nil {
return nil, errors.Wrapf(err, "failed to parse pledge id: %s", pledge.ID)
@@ -31,50 +31,50 @@ func (h Patreon) toModel(pledge *patreon.Pledge) (*models.Pledge, error) {
return nil, errors.Wrapf(err, "failed to parse patron id: %s", pledge.Relationships.Patron.Data.ID)
}
model := &models.Pledge{
m := &model.Pledge{
PledgeID: pledgeID,
PatronID: patronID,
AmountCents: pledge.Attributes.AmountCents,
}
if pledge.Attributes.CreatedAt.Valid {
model.CreatedAt = pledge.Attributes.CreatedAt.Time
m.CreatedAt = pledge.Attributes.CreatedAt.Time
}
if pledge.Attributes.DeclinedSince.Valid {
model.DeclinedSince = pledge.Attributes.DeclinedSince.Time
m.DeclinedSince = pledge.Attributes.DeclinedSince.Time
}
// Read optional fields
if pledge.Attributes.TotalHistoricalAmountCents != nil {
model.TotalHistoricalAmountCents = *pledge.Attributes.TotalHistoricalAmountCents
m.TotalHistoricalAmountCents = *pledge.Attributes.TotalHistoricalAmountCents
}
if pledge.Attributes.OutstandingPaymentAmountCents != nil {
model.OutstandingPaymentAmountCents = *pledge.Attributes.OutstandingPaymentAmountCents
m.OutstandingPaymentAmountCents = *pledge.Attributes.OutstandingPaymentAmountCents
}
if pledge.Attributes.IsPaused != nil {
model.IsPaused = *pledge.Attributes.IsPaused
m.IsPaused = *pledge.Attributes.IsPaused
}
return model, nil
return m, nil
}
func (h Patreon) Hook(pledge *patreon.Pledge, event string) error {
model, err := h.toModel(pledge)
obj, err := h.toModel(pledge)
if err != nil {
return err
}
switch event {
case patreon.EventCreatePledge:
return h.db.Insert(model)
return h.db.Insert(obj)
case patreon.EventUpdatePledge:
return h.db.Update(model)
return h.db.Update(obj)
case patreon.EventDeletePledge:
err := h.db.Delete(model)
err := h.db.Delete(obj)
if err == pg.ErrNoRows {
return nil
}
@@ -85,8 +85,8 @@ func (h Patreon) Hook(pledge *patreon.Pledge, event string) error {
}
}
func (h Patreon) FindPledge(patronID string) (*models.Pledge, error) {
p := &models.Pledge{}
func (h Patreon) FindPledge(patronID string) (*model.Pledge, error) {
p := &model.Pledge{}
return p, h.db.Model(p).Where("patron_id = ?", patronID).Limit(1).Select()
}

View File

@@ -18,7 +18,7 @@ func TestCreate(t *testing.T) {
err := hook.Hook(pledge, patreon.EventCreatePledge)
require.NoError(t, err)
model := &models.Pledge{PledgeID: 12345}
model := &model.Pledge{PledgeID: 12345}
err = hook.db.Select(model)
require.NoError(t, err)
require.Equal(t, pledge.Attributes.AmountCents, model.AmountCents)
@@ -36,7 +36,7 @@ func TestUpdate(t *testing.T) {
err = hook.Hook(pledge, patreon.EventUpdatePledge)
require.NoError(t, err)
model := &models.Pledge{PledgeID: 12345}
model := &model.Pledge{PledgeID: 12345}
err = hook.db.Select(model)
require.NoError(t, err)
require.Equal(t, 999, model.AmountCents)
@@ -85,7 +85,7 @@ func createHandler(t *testing.T) *Patreon {
db := pg.Connect(opts)
_, err = db.Model(&models.Pledge{}).Where("1=1").Delete()
_, err = db.Model(&model.Pledge{}).Where("1=1").Delete()
require.NoError(t, err)
return NewPatreon(db)