Move prices to store

internal/store/db/db.go

@@ -39,11 +39,11 @@ func GetDbMap(db *sql.DB, dbtype config.DbType) (*gorp.DbMap, error) {
 	dbmap := &gorp.DbMap{Db: db, Dialect: dialect}
 	dbmap.AddTableWithName(models.User{}, "users").SetKeys(true, "UserId")
 	dbmap.AddTableWithName(models.Session{}, "sessions").SetKeys(true, "SessionId")
-	dbmap.AddTableWithName(models.Account{}, "accounts").SetKeys(true, "AccountId")
 	dbmap.AddTableWithName(models.Security{}, "securities").SetKeys(true, "SecurityId")
+	dbmap.AddTableWithName(models.Price{}, "prices").SetKeys(true, "PriceId")
+	dbmap.AddTableWithName(models.Account{}, "accounts").SetKeys(true, "AccountId")
 	dbmap.AddTableWithName(models.Transaction{}, "transactions").SetKeys(true, "TransactionId")
 	dbmap.AddTableWithName(models.Split{}, "splits").SetKeys(true, "SplitId")
-	dbmap.AddTableWithName(models.Price{}, "prices").SetKeys(true, "PriceId")
 	rtable := dbmap.AddTableWithName(models.Report{}, "reports").SetKeys(true, "ReportId")
 	rtable.ColMap("Lua").SetMaxSize(models.LuaMaxLength + luaMaxLengthBuffer)
 

internal/store/db/prices.go

@@ -0,0 +1,78 @@
+package db
+
+import (
+	"fmt"
+	"github.com/aclindsa/moneygo/internal/models"
+	"time"
+)
+
+func (tx *Tx) PriceExists(price *models.Price) (bool, error) {
+	var prices []*models.Price
+	_, err := tx.Select(&prices, "SELECT * from prices where SecurityId=? AND CurrencyId=? AND Date=? AND Value=?", price.SecurityId, price.CurrencyId, price.Date, price.Value)
+	return len(prices) > 0, err
+}
+
+func (tx *Tx) InsertPrice(price *models.Price) error {
+	return tx.Insert(price)
+}
+
+func (tx *Tx) GetPrice(priceid, securityid int64) (*models.Price, error) {
+	var price models.Price
+	err := tx.SelectOne(&price, "SELECT * from prices where PriceId=? AND SecurityId=?", priceid, securityid)
+	if err != nil {
+		return nil, err
+	}
+	return &price, nil
+}
+
+func (tx *Tx) GetPrices(securityid int64) (*[]*models.Price, error) {
+	var prices []*models.Price
+
+	_, err := tx.Select(&prices, "SELECT * from prices where SecurityId=?", securityid)
+	if err != nil {
+		return nil, err
+	}
+	return &prices, nil
+}
+
+// Return the latest price for security in currency units before date
+func (tx *Tx) GetLatestPrice(security, currency *models.Security, date *time.Time) (*models.Price, error) {
+	var price models.Price
+	err := tx.SelectOne(&price, "SELECT * from prices where SecurityId=? AND CurrencyId=? AND Date <= ? ORDER BY Date DESC LIMIT 1", security.SecurityId, currency.SecurityId, date)
+	if err != nil {
+		return nil, err
+	}
+	return &price, nil
+}
+
+// Return the earliest price for security in currency units after date
+func (tx *Tx) GetEarliestPrice(security, currency *models.Security, date *time.Time) (*models.Price, error) {
+	var price models.Price
+	err := tx.SelectOne(&price, "SELECT * from prices where SecurityId=? AND CurrencyId=? AND Date >= ? ORDER BY Date ASC LIMIT 1", security.SecurityId, currency.SecurityId, date)
+	if err != nil {
+		return nil, err
+	}
+	return &price, nil
+}
+
+func (tx *Tx) UpdatePrice(price *models.Price) error {
+	count, err := tx.Update(price)
+	if err != nil {
+		return err
+	}
+	if count != 1 {
+		return fmt.Errorf("Expected to update 1 price, was going to update %d", count)
+	}
+	return nil
+}
+
+func (tx *Tx) DeletePrice(price *models.Price) error {
+	count, err := tx.Delete(price)
+	if err != nil {
+		return err
+	}
+	if count != 1 {
+		return fmt.Errorf("Expected to delete 1 price, was going to delete %d", count)
+	}
+	return nil
+}