44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
models "vibeStonk/server/models/v1"
|
|
)
|
|
|
|
type HoldingRepo interface {
|
|
// GetAll returns all holdings (purchases with sold quantity information)
|
|
GetAll() ([]*models.Holding, error)
|
|
|
|
// GetUnsold returns holdings that are not fully sold (including partially sold)
|
|
GetUnsold() ([]*models.Holding, error)
|
|
|
|
// GetSold returns holdings that are fully sold
|
|
GetSold() ([]*models.Holding, error)
|
|
|
|
// GetByID returns a specific holding by purchaseID
|
|
GetByID(purchaseID int64) (*models.Holding, error)
|
|
|
|
// GetByStockID returns holdings for a specific stock
|
|
GetByStockID(stockID int64) ([]*models.Holding, error)
|
|
|
|
// GetUnsoldByStockID returns unsold holdings for a specific stock ID
|
|
GetUnsoldByStockID(stockID int64) ([]*models.Holding, error)
|
|
|
|
// GetSoldByStockID returns sold holdings for a specific stock ID
|
|
GetSoldByStockID(stockID int64) ([]*models.Holding, error)
|
|
}
|
|
|
|
func NewHoldingRepo(config *Config, db *sql.DB) (HoldingRepo, error) {
|
|
switch config.DBEngine {
|
|
case Sqlite:
|
|
repo, err := newSqliteHoldingRepo(db)
|
|
if err != nil {
|
|
return nil, fmt.Errorf(errMsgFailed, "Sqlite", err)
|
|
}
|
|
return repo, nil
|
|
default:
|
|
return nil, ErrBadEngine
|
|
}
|
|
}
|