Nathan Coad
9a561f3b07
All checks were successful
continuous-integration/drone/push Build is passing
89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
package handler
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"vctp/db"
|
|
queries "vctp/db/queries"
|
|
models "vctp/server/models"
|
|
)
|
|
|
|
// VmImport ingests a bulk VM import payload.
|
|
// @Summary Import VMs (deprecated)
|
|
// @Description Deprecated: Imports existing VM inventory data in bulk.
|
|
// @Tags inventory
|
|
// @Deprecated
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param import body models.ImportReceived true "Bulk import payload"
|
|
// @Success 200 {object} models.StatusMessageResponse "Import processed"
|
|
// @Failure 500 {object} models.ErrorResponse "Server error"
|
|
// @Router /api/import/vm [post]
|
|
func (h *Handler) VmImport(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
writeJSONError(w, http.StatusMethodNotAllowed, "method not allowed")
|
|
return
|
|
}
|
|
|
|
if h.denyLegacyAPI(w, "/api/import/vm") {
|
|
return
|
|
}
|
|
|
|
var inData models.ImportReceived
|
|
if err := decodeJSONBody(w, r, &inData); err != nil {
|
|
h.Logger.Error("Unable to decode json request body", "error", err)
|
|
writeJSONError(w, http.StatusBadRequest, fmt.Sprintf("Unable to decode json request body: '%s'", err))
|
|
return
|
|
} else {
|
|
//h.Logger.Debug("successfully decoded JSON")
|
|
//prettyPrint(inData)
|
|
}
|
|
|
|
if strings.HasPrefix(inData.Name, "vCLS-") {
|
|
h.Logger.Info("Skipping internal vCLS VM import", "vm_name", inData.Name)
|
|
writeJSONOKMessage(w, fmt.Sprintf("Skipped internal VM '%s'", inData.Name))
|
|
return
|
|
}
|
|
|
|
ctx, cancel := withRequestTimeout(r, defaultRequestTimeout)
|
|
defer cancel()
|
|
|
|
// Query Inventory table for this VM before adding it
|
|
h.Logger.Debug("Checking inventory table for VM record")
|
|
invParams := queries.GetInventoryVmIdParams{
|
|
VmId: sql.NullString{String: inData.VmId, Valid: inData.VmId != ""},
|
|
DatacenterName: sql.NullString{String: inData.Datacenter, Valid: inData.Datacenter != ""},
|
|
}
|
|
_, err := h.Database.Queries().GetInventoryVmId(ctx, invParams)
|
|
|
|
if err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
// do the insert
|
|
// Create an instance of CreateInventoryParams
|
|
var params queries.CreateInventoryParams
|
|
|
|
// Convert vmModel to CreateInventoryParams using the utility function
|
|
db.ConvertToSQLParams(&inData, ¶ms)
|
|
//prettyPrint(params)
|
|
|
|
// Insert the new inventory record into the database
|
|
result, err := h.Database.Queries().CreateInventory(ctx, params)
|
|
if err != nil {
|
|
h.Logger.Error("unable to perform database insert", "error", err)
|
|
} else {
|
|
h.Logger.Debug("created database record", "insert_result", result)
|
|
}
|
|
|
|
} else {
|
|
h.Logger.Error("unable to check inventory for vm", "error", err, "vm_id", inData.VmId, "datacenter_name", inData.Datacenter)
|
|
}
|
|
} else {
|
|
h.Logger.Info("not adding vm to inventory table since record alraedy exists", "vm_id", inData.VmId, "datacenter_name", inData.Datacenter)
|
|
}
|
|
|
|
writeJSONOKMessage(w, fmt.Sprintf("Successfully processed import request for VM '%s'", inData.Name))
|
|
}
|