56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package service
|
|
|
|
import (
|
|
"Portifolio/internal/database"
|
|
"Portifolio/internal/model"
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func InsertRevenue(db *sql.DB, companyID int, currencyID int, categoryName string, parentID *int, value float64, period model.Period) error {
|
|
_, err := database.GetCompanyByID(db, companyID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = database.GetCurrencyByID(db, currencyID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// checking if period is in db, in case not will insert
|
|
_, err = database.GetPeriodByID(db, period.ID)
|
|
if err != nil {
|
|
err = period.Insert(db)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
// Getting Category, if error, trying to insert the category with the company.
|
|
category, err := database.GetCategoryByName(db, companyID, categoryName)
|
|
if err != nil {
|
|
category = model.RevenueCategory{
|
|
CompanyID: companyID,
|
|
ParentID: parentID,
|
|
Name: categoryName,
|
|
}
|
|
err := database.InsertCategory(db, category)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
_, err = db.Exec(
|
|
`INSERT INTO revenue_entries (company_id, currency_id, category_id, period_id, value)
|
|
VALUES (?, ?, ?, ?, ?)`,
|
|
companyID, currencyID, category.ID, period.ID, value,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("insert revenue_entries: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|