Adding support for packages
This commit is contained in:
44
Database/package.go
Normal file
44
Database/package.go
Normal file
@@ -0,0 +1,44 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
|
||||
"gopkg.in/guregu/null.v4"
|
||||
)
|
||||
|
||||
type Package struct {
|
||||
ID int `json:"id"`
|
||||
Name null.String `json:"name"`
|
||||
Active int `json:"active"`
|
||||
}
|
||||
|
||||
func GetPackage(db *sql.DB) ([]Package, error) {
|
||||
var packages []Package
|
||||
rows, err := db.Query("SELECT * FROM dashboard_package")
|
||||
if err != nil {
|
||||
log.Println("Error querying list of OS", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
var pkg Package
|
||||
err := rows.Scan(&pkg.ID, &pkg.Name, &pkg.Active)
|
||||
if err != nil {
|
||||
log.Println("Error building list", err)
|
||||
}
|
||||
packages = append(packages, pkg)
|
||||
}
|
||||
return packages, err
|
||||
}
|
||||
|
||||
func GetPackagebyID(db *sql.DB, id int64) (Package, error) {
|
||||
var pkg Package
|
||||
if err := db.QueryRow("SELECT * FROM dashboard_package where id = ?", id).Scan(&pkg.ID, &pkg.Name, &pkg.Active); err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
return pkg, nil
|
||||
}
|
||||
log.Println("Error getting OS by ID", err)
|
||||
}
|
||||
return pkg, nil
|
||||
}
|
||||
@@ -37,7 +37,7 @@ func GetServersbyID(db *sql.DB, id int64) Server {
|
||||
return s
|
||||
}
|
||||
|
||||
func GetServersbyOS(db *sql.DB, id int) ([]Server, error) {
|
||||
func GetServersbyOS(db *sql.DB, id int64) ([]Server, error) {
|
||||
var servers []Server
|
||||
rows, err := db.Query("SELECT * FROM dashboard_server WHERE os_id = ?", id)
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user