51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package service
|
|
|
|
import (
|
|
"Portifolio/internal/model"
|
|
"database/sql"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func InsertCompany(db *sql.DB, input model.CompanyInput) (int, error) {
|
|
res, err := db.Exec(
|
|
`INSERT INTO companies (name, shares_outstanding, price, currency_id) VALUES (?, ?, ?, ?)`,
|
|
input.Name, input.SharesOutstanding, input.Price, input.CurrencyID,
|
|
)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, err := res.LastInsertId()
|
|
return int(id), err
|
|
}
|
|
|
|
func GetAllCompanies(db *sql.DB) ([]model.Company, error) {
|
|
rows, err := db.Query(`
|
|
SELECT c.id, c.name, c.shares_outstanding, c.price,
|
|
cu.id, cu.code, cu.name
|
|
FROM companies c
|
|
JOIN currencies cu ON c.currency_id = cu.id
|
|
ORDER BY c.name
|
|
`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var companies []model.Company
|
|
for rows.Next() {
|
|
var c model.Company
|
|
var cu model.Currency
|
|
if err := rows.Scan(
|
|
&c.ID, &c.Name, &c.SharesOutstanding, &c.Price,
|
|
&cu.ID, &cu.Code, &cu.Name,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
c.CurrencyID = cu.ID
|
|
c.Currency = &cu
|
|
companies = append(companies, c)
|
|
}
|
|
return companies, rows.Err()
|
|
}
|