moving insert to another model and done so in database internal

This commit is contained in:
samantha42
2026-03-25 21:58:08 +01:00
parent 2dccca5da1
commit d491b9c14c
11 changed files with 85 additions and 83 deletions

View File

@@ -36,7 +36,7 @@ func GetTrades(db *sql.DB) ([]model.Trade, error) {
if err != nil {
return nil, err
}
t.Currency = currency
t.Currency = currency.Name
t.Ticker = *company
switch TypeInt {
case 0:
@@ -83,7 +83,7 @@ func InsertTrade(db *sql.DB, trade model.Trade) error {
_, err := db.Exec(
"INSERT INTO trades (company_id, currency_id, shares, product, type, price, traded_at) VALUES (?, ?, ?, ?, ?, ?, ?)",
trade.Ticker.ID,
trade.Currency.ID,
trade.Currency,
trade.Shares,
trade.Product,
trade.Type,

View File

@@ -82,3 +82,52 @@ func GetRevenueByCategory(db *sql.DB, companyID int, categoryID int) ([]model.Re
return revenues, nil
}
func InsertRevenue(db *sql.DB, rev model.RevenueInsert) error {
_, err := GetCompanyByID(db, rev.CompanyID)
if err != nil {
return err
}
_, err = GetCurrencyByID(db, rev.CurrencyID)
if err != nil {
return err
}
// checking if period is in db, in case not will insert
_, err = GetPeriodByID(db, rev.Period.ID)
if err != nil {
err = rev.Period.Insert(db)
if err != nil {
return err
}
}
// Getting Category, if error, trying to insert the category with the company.
category, err := GetCategoryByName(db, rev.CompanyID, rev.CategoryName)
if err != nil {
err := InsertCategory(db, model.RevenueCategory{
CompanyID: rev.CompanyID,
ParentID: &rev.ParentID,
Name: rev.CategoryName,
})
if err != nil {
return err
}
}
category, err = GetCategoryByName(db, rev.CompanyID, rev.CategoryName)
if err != nil {
return err
}
_, err = db.Exec(
`INSERT INTO revenue_entries (company_id, currency_id, category_id, period_id, value)
VALUES (?, ?, ?, ?, ?)`,
rev.CompanyID, rev.CurrencyID, category.ID, rev.Period.ID, rev.Value,
)
if err != nil {
return fmt.Errorf("insert revenue_entries: %w", err)
}
return nil
}