2024-03-22 09:42:32 +01:00

92 lines
2.3 KiB
Go

package database
import (
"database/sql"
"log"
"gopkg.in/guregu/null.v4"
)
type OS struct {
ID int `json:"id"`
Distribution null.String `json:"distribution"`
Version null.String `json:"version"`
End_of_support null.String `json:"end_of_support"`
}
func GetOS(db *sql.DB) ([]OS, error) {
var oses []OS
rows, err := db.Query("SELECT * FROM dashboard_os")
if err != nil {
log.Println("Error querying list of OS", err)
}
defer rows.Close()
for rows.Next() {
var os OS
err := rows.Scan(&os.ID, &os.Distribution, &os.Version, &os.End_of_support)
if err != nil {
log.Println("Error building list", err)
}
oses = append(oses, os)
}
return oses, err
}
func GetOSbyID(db *sql.DB, id string) (OS, error) {
var os OS
if err := db.QueryRow("SELECT * FROM dashboard_os where id = ?", id).Scan(&os.ID, &os.Distribution, &os.Version, &os.End_of_support); err != nil {
if err == sql.ErrNoRows {
return os, nil
}
log.Println("Error getting OS by ID", err)
}
return os, nil
}
func GetDistributionList(db *sql.DB) ([]null.String, error) {
var list []null.String
rows, err := db.Query("SELECT DISTINCT distribution FROM dashboard_os")
if err != nil {
log.Println("Error getting distribution list", err)
}
for rows.Next() {
var os OS
err := rows.Scan(&os.Distribution)
if err != nil {
log.Println("Error building distribution list", err)
}
list = append(list, os.Distribution)
}
return list, nil
}
func GetVersionsByDistributionList(db *sql.DB, d string) ([]null.String, error) {
var list []null.String
rows, err := db.Query("SELECT DISTINCT version FROM dashboard_os where distribution = ?", d)
if err != nil {
log.Println("Error getting distribution list", err)
}
for rows.Next() {
var os OS
err := rows.Scan(&os.Distribution)
if err != nil {
log.Println("Error building distribution list", err)
}
list = append(list, os.Distribution)
}
return list, nil
}
func CreateOS(os OS, db *sql.DB) error {
q, err := db.Prepare("INSERT INTO `dashboard_os` (`distribution`,`version`,`end_of_support`) VALUES (?,?,?)")
if err != nil {
log.Println("Error creating query", err)
}
_, err = q.Exec(&os.Distribution, &os.Version, &os.End_of_support)
if err != nil {
log.Println("Error inserting OS", err)
}
return err
}