85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package database
|
|
|
|
import (
|
|
"Portifolio/internal/model"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func GetPeriodByID(db *sql.DB, periodID int) (model.Period, error) {
|
|
var p model.Period
|
|
var start, end string
|
|
err := db.QueryRow(
|
|
`SELECT type, year, idx, start_date, end_date FROM periods WHERE id = ?`,
|
|
periodID,
|
|
).Scan(&p.Type, &p.Year, &p.Index, &start, &end)
|
|
if err == sql.ErrNoRows {
|
|
return p, fmt.Errorf("period %d not found", periodID)
|
|
}
|
|
if err != nil {
|
|
return p, fmt.Errorf("get period by id: %w", err)
|
|
}
|
|
p.Start, _ = time.Parse("2006-01-02", start)
|
|
p.End, _ = time.Parse("2006-01-02", end)
|
|
return p, nil
|
|
}
|
|
|
|
func GetRevenueByPeriod(db *sql.DB, companyID int, periodID int) ([]model.Revenue, error) {
|
|
rows, err := db.Query(
|
|
`SELECT id, company_id, currency_id, period_id, value FROM revenue_entries WHERE company_id = ? AND period_id = ?`,
|
|
companyID, periodID,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("query revenue by period: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var revenues []model.Revenue
|
|
for rows.Next() {
|
|
var rc model.Revenue
|
|
if err := rows.Scan(&rc.ID, &rc.Company, &rc.Currency, &rc.Period, &rc.Value); err != nil {
|
|
return nil, fmt.Errorf("scan revenue row: %w", err)
|
|
}
|
|
revenues = append(revenues, rc)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, fmt.Errorf("iterate revenue rows: %w", err)
|
|
}
|
|
if len(revenues) == 0 {
|
|
return nil, fmt.Errorf("revenue by company %d and period_id %d not found", companyID, periodID)
|
|
}
|
|
|
|
return revenues, nil
|
|
}
|
|
|
|
func GetRevenueByCategory(db *sql.DB, companyID int, categoryID int) ([]model.Revenue, error) {
|
|
rows, err := db.Query(
|
|
`SELECT id, company_id, currency_id, category_id, value FROM revenue_entries WHERE company_id = ? AND category_id = ?`,
|
|
companyID, categoryID,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("query revenue by category: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var revenues []model.Revenue
|
|
for rows.Next() {
|
|
var rc model.Revenue
|
|
if err := rows.Scan(&rc.ID, &rc.Company, &rc.Currency, &rc.Category, &rc.Value); err != nil {
|
|
return nil, fmt.Errorf("scan revenue row: %w", err)
|
|
}
|
|
revenues = append(revenues, rc)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, fmt.Errorf("iterate revenue rows: %w", err)
|
|
}
|
|
if len(revenues) == 0 {
|
|
return nil, fmt.Errorf("revenue by company %d and category_id %d not found", companyID, categoryID)
|
|
}
|
|
|
|
return revenues, nil
|
|
}
|