Working docker containerization
This commit is contained in:
131
Database/os.go
131
Database/os.go
@@ -1,131 +0,0 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"errors"
|
||||
"log"
|
||||
|
||||
"gopkg.in/guregu/null.v4"
|
||||
)
|
||||
|
||||
type OS struct {
|
||||
ID int `json:"id"`
|
||||
Distribution null.String `json:"distribution"`
|
||||
Version null.String `json:"version"`
|
||||
EndOfSupport null.String `json:"EndOfSupport"`
|
||||
}
|
||||
|
||||
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 func(rows *sql.Rows) {
|
||||
err := rows.Close()
|
||||
if err != nil {
|
||||
|
||||
}
|
||||
}(rows)
|
||||
|
||||
for rows.Next() {
|
||||
var os OS
|
||||
err := rows.Scan(&os.ID, &os.Distribution, &os.Version, &os.EndOfSupport)
|
||||
if err != nil {
|
||||
log.Println("Error building list", err)
|
||||
}
|
||||
oses = append(oses, os)
|
||||
}
|
||||
return oses, err
|
||||
}
|
||||
|
||||
func GetOSbyID(db *sql.DB, id int64) (OS, error) {
|
||||
var os OS
|
||||
if err := db.QueryRow("SELECT * FROM dashboard_os where id = ?", id).Scan(&os.ID, &os.Distribution, &os.Version, &os.EndOfSupport); err != nil {
|
||||
if errors.Is(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 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) {
|
||||
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 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
|
||||
}
|
||||
@@ -1,151 +0,0 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"errors"
|
||||
"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 errors.Is(err, sql.ErrNoRows) {
|
||||
return pkg, nil
|
||||
}
|
||||
log.Println("Error getting OS by ID", err)
|
||||
}
|
||||
return pkg, nil
|
||||
}
|
||||
|
||||
func GetPackagebyName(db *sql.DB, id int64) (Package, error) {
|
||||
var pkg Package
|
||||
if err := db.QueryRow("SELECT * FROM dashboard_package where name = ?", id).Scan(&pkg.ID, &pkg.Name, &pkg.Active); err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return pkg, nil
|
||||
}
|
||||
log.Println("Error getting OS by ID", err)
|
||||
}
|
||||
return pkg, nil
|
||||
}
|
||||
|
||||
func checkIfPackageExists(pkg Package, db *sql.DB) bool {
|
||||
if err := db.QueryRow("Select name from dashboard_package where name = ?", pkg.Name).Scan(&pkg.Name); err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func CreatePackage(pkg Package, db *sql.DB) error {
|
||||
if checkIfPackageExists(pkg, db) {
|
||||
/* If package exists but is not active we activate it */
|
||||
row := db.QueryRow("Select name, active from dashboard_package where name = ? and active = 1", pkg.Name)
|
||||
err := row.Scan(&pkg.Name, &pkg.Active)
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
q, err := db.Prepare("UPDATE dashboard_package set active = 1 where name = ?")
|
||||
if err != nil {
|
||||
log.Println("Error creating query", err)
|
||||
}
|
||||
_, err = q.Exec(pkg.Name)
|
||||
if err != nil {
|
||||
log.Println("Error activating Package", err)
|
||||
}
|
||||
log.Println("Activating new package", pkg.Name)
|
||||
return nil
|
||||
}
|
||||
return nil
|
||||
}
|
||||
q, err := db.Prepare("INSERT INTO dashboard_package (name, active) values (?, 1)")
|
||||
if err != nil {
|
||||
log.Println("Error creating query", err)
|
||||
return err
|
||||
}
|
||||
_, err = q.Exec(pkg.Name)
|
||||
if err != nil {
|
||||
log.Println("Error creating Package", err)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func DisablePackage(pkg Package, db *sql.DB) (int, error) {
|
||||
if !checkIfPackageExists(pkg, db) {
|
||||
log.Println("Package does not exist", pkg.Name)
|
||||
return 404, nil
|
||||
}
|
||||
q, err := db.Prepare("UPDATE dashboard_package set active = 0 where name = ?")
|
||||
if err != nil {
|
||||
log.Println("Error creating query", err)
|
||||
return 500, err
|
||||
}
|
||||
_, err = q.Exec(pkg.Name)
|
||||
if err != nil {
|
||||
log.Println("Error disabling Package", err)
|
||||
return 500, err
|
||||
}
|
||||
return 200, nil
|
||||
}
|
||||
|
||||
func EnablePackage(pkg Package, db *sql.DB) (int, error) {
|
||||
if !checkIfPackageExists(pkg, db) {
|
||||
log.Println("Package does not exist", pkg.Name)
|
||||
return 404, nil
|
||||
}
|
||||
q, err := db.Prepare("UPDATE dashboard_package set active = 1 where name = ?")
|
||||
if err != nil {
|
||||
log.Println("Error creating query", err)
|
||||
return 500, err
|
||||
}
|
||||
_, err = q.Exec(pkg.Name)
|
||||
if err != nil {
|
||||
log.Println("Error disabling Package", err)
|
||||
return 500, err
|
||||
}
|
||||
return 200, nil
|
||||
}
|
||||
|
||||
func DeletePackage(pkg Package, db *sql.DB) (int, error) {
|
||||
if !checkIfPackageExists(pkg, db) {
|
||||
log.Println("Package does not exist", pkg.Name)
|
||||
return 404, nil
|
||||
}
|
||||
q, err := db.Prepare("DELETE FROM dashboard_package where name = ?")
|
||||
if err != nil {
|
||||
log.Println("Error creating query", err)
|
||||
return 500, err
|
||||
}
|
||||
_, err = q.Exec(pkg.Name)
|
||||
if err != nil {
|
||||
log.Println("Error deleting Package", err)
|
||||
return 500, err
|
||||
}
|
||||
return 200, nil
|
||||
}
|
||||
@@ -1,118 +0,0 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"errors"
|
||||
"gopkg.in/guregu/null.v4"
|
||||
"log"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
ID int `json:"id"`
|
||||
Hostname null.String `json:"hostname"`
|
||||
OsId null.String `json:"OsId"`
|
||||
}
|
||||
|
||||
func GetServersList(db *sql.DB) ([]Server, error) {
|
||||
var servers []Server
|
||||
rows, err := db.Query("SELECT * FROM dashboard_server")
|
||||
if err != nil {
|
||||
log.Println("Error querying servers list", err)
|
||||
}
|
||||
defer func(rows *sql.Rows) {
|
||||
err := rows.Close()
|
||||
if err != nil {
|
||||
|
||||
}
|
||||
}(rows)
|
||||
|
||||
for rows.Next() {
|
||||
var s Server
|
||||
err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
|
||||
if err != nil {
|
||||
log.Println("Error building servers list", err)
|
||||
}
|
||||
servers = append(servers, s)
|
||||
}
|
||||
return servers, err
|
||||
}
|
||||
|
||||
func GetServersbyID(db *sql.DB, id int64) Server {
|
||||
var s Server
|
||||
rows := db.QueryRow("SELECT * FROM dashboard_server WHERE id = ?", id)
|
||||
err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
|
||||
if err != nil {
|
||||
return Server{}
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
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 {
|
||||
log.Println("Error querying servers list by os_id", err)
|
||||
}
|
||||
defer func(rows *sql.Rows) {
|
||||
err := rows.Close()
|
||||
if err != nil {
|
||||
|
||||
}
|
||||
}(rows)
|
||||
|
||||
for rows.Next() {
|
||||
var s Server
|
||||
err := rows.Scan(&s.ID, &s.Hostname, &s.OsId)
|
||||
if err != nil {
|
||||
log.Println("Error building servers list", err)
|
||||
}
|
||||
servers = append(servers, s)
|
||||
}
|
||||
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) {
|
||||
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 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
|
||||
}
|
||||
@@ -1,25 +0,0 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/go-sql-driver/mysql"
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
)
|
||||
|
||||
func GetDatabaseConnection() *sql.DB {
|
||||
cfg := mysql.Config{
|
||||
User: os.Getenv("DATABASE_USER"),
|
||||
Passwd: os.Getenv("DATABASE_PASSWORD"),
|
||||
Net: "tcp",
|
||||
Addr: os.Getenv("DATABASE_HOST"),
|
||||
DBName: os.Getenv("DATABASE_NAME"),
|
||||
}
|
||||
DbConn, err := sql.Open("mysql", cfg.FormatDSN())
|
||||
if err != nil {
|
||||
log.Println("Database connection error : ", err)
|
||||
}
|
||||
return DbConn
|
||||
}
|
||||
Reference in New Issue
Block a user