kirby/infra-dashboard
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding deletion for os and server

Browse Source
This commit is contained in:
kirby
2024-08-09 14:43:32 +02:00
parent b04387d29f
commit e98031135f
5 changed files with 126 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
Database/os.go
View File

@@ -84,14 +84,49 @@ func GetVersionsByDistributionList(db *sql.DB, d string) ([]null.String, error)
return list, nil
}
func checkIfOsExists(os OS, db *sql.DB) bool {
row := db.QueryRow("Select distribution, version from dashboard_os where distribution = ? and version = ?", os.Distribution, os.Version)
err := row.Scan(&os.Distribution, &os.Version)
if !errors.Is(err, sql.ErrNoRows) {
log.Println("OS exists")
return true
}
return false
}
func CreateOS(os OS, db *sql.DB) error {
if checkIfOsExists(os, db) {
log.Println("OS already exists")
return nil
}
q, err := db.Prepare("INSERT INTO `dashboard_os` (`distribution`,`version`,`end_of_support`) VALUES (?,?,?)")
if err != nil {
log.Println("Error creating query", err)
return err
}
_, err = q.Exec(&os.Distribution, &os.Version, &os.EndOfSupport)
if err != nil {
log.Println("Error inserting OS", err)
return err
}
return err
return nil
}
func DeleteOS(os OS, db *sql.DB) (int, error) {
if !checkIfOsExists(os, db) {
log.Println("OS does not exist, nothing to delete")
return 404, nil
}
q, err := db.Prepare("DELETE FROM dashboard_os WHERE distribution = ? AND version = ?")
if err != nil {
log.Println("Error deleting OS", err)
return 500, err
}
_, err = q.Exec(&os.Distribution, &os.Version)
if err != nil {
log.Println("Error deleting OS", err)
return 500, err
}
log.Println("OS deleted")
return 200, nil
}

38
Database/server.go
View File

@@ -2,6 +2,7 @@ package database
import (
"database/sql"
"errors"
"gopkg.in/guregu/null.v4"
"log"
)
@@ -70,14 +71,49 @@ func GetServersbyOS(db *sql.DB, id int64) ([]Server, error) {
return servers, err
}
func checkIfServerExists(server Server, db *sql.DB) bool {
row := db.QueryRow("Select hostname, os_id from dashboard_server where hostname = ? and os_id = ?", server.Hostname, server.OsId)
err := row.Scan(&server.Hostname, &server.OsId)
if !errors.Is(err, sql.ErrNoRows) {
log.Println("Server exists")
return true
}
return false
}
func CreateServer(server Server, db *sql.DB) error {
if checkIfServerExists(server, db) {
log.Println("Server already exists")
return nil
}
q, err := db.Prepare("INSERT INTO `dashboard_server` (hostname, os_id) VALUES (?,?)")
if err != nil {
log.Println("Error creating query", err)
return err
}
_, err = q.Exec(server.Hostname, server.OsId)
if err != nil {
log.Println("Error inserting Server", err)
return err
}
return err
return nil
}
func DeleteServer(server Server, db *sql.DB) (int, error) {
if !checkIfServerExists(server, db) {
log.Println("Server does not exist, nothing to delete")
return 404, nil
}
q, err := db.Prepare("DELETE FROM dashboard_server WHERE hostname = ? AND os_id = ?")
if err != nil {
log.Println("Error deleting Server", err)
return 500, err
}
_, err = q.Exec(server.Hostname, server.OsId)
if err != nil {
log.Println("Error deleting Server", err)
return 500, err
}
log.Println("Server deleted")
return 200, nil
}