samantha/Portfolio-Engine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rev cat getter by company id

Browse Source
This commit is contained in:
zipfriis
2026-03-25 19:07:57 +01:00
parent e056d8a37e
commit 206954fb75
6 changed files with 79 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
internal/database/category.go
View File

@@ -52,3 +52,54 @@ func InsertCategory(db *sql.DB, rc model.RevenueCategory) error {
}
return nil
}
func GetCategoriesByCompanyID(db *sql.DB, companyID int) ([]string, error) {
rows, err := db.Query("SELECT name FROM category WHERE company_id = ?", companyID)
if err != nil {
return []string{}, err
}
defer rows.Close()
var list []string
for rows.Next() {
var Name string
err := rows.Scan(&Name)
if err != nil {
return []string{}, err
}
list = append(list, Name)
}
return list, nil
}
func GetCategoryByID(db *sql.DB, companyID int, ID int) (*model.RevenueCategory, error) {
var rc model.RevenueCategory
err := db.QueryRow(
`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND id = ?`,
companyID, ID,
).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
if err == sql.ErrNoRows {
return &rc, fmt.Errorf("category %q not found for company %d", ID, companyID)
}
if err != nil {
return &rc, fmt.Errorf("get category by name: %w", err)
}
return &rc, nil
}
func GetCategoryByName(db *sql.DB, companyID int, name string) (model.RevenueCategory, error) {
var rc model.RevenueCategory
err := db.QueryRow(
`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND name = ?`,
companyID, name,
).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
if err == sql.ErrNoRows {
return rc, fmt.Errorf("category %q not found for company %d", name, companyID)
}
if err != nil {
return rc, fmt.Errorf("get category by name: %w", err)
}
return rc, nil
}

31
internal/database/revenue.go
View File

@@ -9,37 +9,6 @@ import (
_ "github.com/mattn/go-sqlite3"
)
func GetCategoryByID(db *sql.DB, companyID int, ID int) (*model.RevenueCategory, error) {
var rc model.RevenueCategory
err := db.QueryRow(
`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND id = ?`,
companyID, ID,
).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
if err == sql.ErrNoRows {
return &rc, fmt.Errorf("category %q not found for company %d", ID, companyID)
}
if err != nil {
return &rc, fmt.Errorf("get category by name: %w", err)
}
return &rc, nil
}
func GetCategoryByName(db *sql.DB, companyID int, name string) (model.RevenueCategory, error) {
var rc model.RevenueCategory
err := db.QueryRow(
`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND name = ?`,
companyID, name,
).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
if err == sql.ErrNoRows {
return rc, fmt.Errorf("category %q not found for company %d", name, companyID)
}
if err != nil {
return rc, fmt.Errorf("get category by name: %w", err)
}
return rc, nil
}
func GetPeriodByID(db *sql.DB, periodID int) (model.Period, error) {
var p model.Period
var start, end string