change endpoint to store cloudevent in events table

2024-09-13 12:58:22 +10:00
parent 446cd30d8e
commit eecf227cd8
8 changed files with 270 additions and 53 deletions
--- a/server/handler/vmCreate.go
+++ b/server/handler/vmCreate.go
@@ -11,7 +11,6 @@ import (
 	"strconv"
 	"time"
 	queries "vctp/db/queries"
-	"vctp/internal/vcenter"
 	models "vctp/server/models"
 )

@@ -35,14 +34,14 @@ func (h *Handler) VmCreate(w http.ResponseWriter, r *http.Request) {
 	}

 	// Decode the JSON body into vmModel struct
-	var vm models.CloudEventReceived
-	if err := json.Unmarshal(reqBody, &vm); err != nil {
+	var event models.CloudEventReceived
+	if err := json.Unmarshal(reqBody, &event); err != nil {
 		h.Logger.Error("unable to decode json", "error", err)
 		http.Error(w, "Invalid JSON body", http.StatusBadRequest)
 		return
 	} else {
 		h.Logger.Debug("successfully decoded JSON")
-		prettyPrint(vm)
+		prettyPrint(event)
 	}

 	// Convert vmModel to CreateInventoryParams using the utility function
@@ -50,7 +49,7 @@ func (h *Handler) VmCreate(w http.ResponseWriter, r *http.Request) {
 	//db.ConvertToSQLParams(&vm, &params)

 	// Parse the datetime string to a time.Time object
-	eventTime, err := time.Parse(time.RFC3339, vm.CloudEvent.Time)
+	eventTime, err := time.Parse(time.RFC3339, event.CloudEvent.Time)
 	if err != nil {
 		h.Logger.Warn("unable to convert cloud event time to timestamp", "error", err)
 		unixTimestamp = time.Now().Unix()
@@ -59,58 +58,74 @@ func (h *Handler) VmCreate(w http.ResponseWriter, r *http.Request) {
 		unixTimestamp = eventTime.Unix()
 	}

-	// TODO - initiate govmomi query of source vcenter to discover related data
-	h.Logger.Debug("connecting to vcenter")
-	vc := vcenter.New(h.Logger)
-	vc.Login(vm.CloudEvent.Source)
-	//vmObject, err := vc.FindVMByName(vm.CloudEvent.Data.VM.Name)
-	//vmObject, err := vc.FindVMByID(vm.CloudEvent.Data.VM.VM.Value)
-	vmObject, err := vc.FindVMByIDWithDatacenter(vm.CloudEvent.Data.VM.VM.Value, vm.CloudEvent.Data.Datacenter.Datacenter.Value)
+	/*
+		// TODO - initiate govmomi query of source vcenter to discover related data
+		h.Logger.Debug("connecting to vcenter")
+		vc := vcenter.New(h.Logger)
+		vc.Login(event.CloudEvent.Source)
+		//vmObject, err := vc.FindVMByName(vm.CloudEvent.Data.VM.Name)
+		//vmObject, err := vc.FindVMByID(vm.CloudEvent.Data.VM.VM.Value)
+		vmObject, err := vc.FindVMByIDWithDatacenter(event.CloudEvent.Data.VM.VM.Value, event.CloudEvent.Data.Datacenter.Datacenter.Value)

-	if err != nil {
-		h.Logger.Error("Can't locate vm in vCenter", "vmID", vm.CloudEvent.Data.VM.VM.Value, "error", err)
-	} else if vmObject == nil {
-		h.Logger.Debug("didn't find VM", "vm_id", vm.CloudEvent.Data.VM.VM.Value)
-		numRam = 0
-		numVcpus = 0
-		datacenter = vm.CloudEvent.Data.Datacenter.Name
-	} else {
-		h.Logger.Debug("found VM")
-		//prettyPrint(vmObject)
-
-		// calculate VM properties we want to store
-		if vmObject.Vm.Config != nil {
-			numRam = vmObject.Vm.Config.Hardware.MemoryMB
-			numVcpus = vmObject.Vm.Config.Hardware.NumCPU * vmObject.Vm.Config.Hardware.NumCoresPerSocket
+		if err != nil {
+			h.Logger.Error("Can't locate vm in vCenter", "vmID", event.CloudEvent.Data.VM.VM.Value, "error", err)
+		} else if vmObject == nil {
+			h.Logger.Debug("didn't find VM", "vm_id", event.CloudEvent.Data.VM.VM.Value)
+			numRam = 0
+			numVcpus = 0
+			datacenter = event.CloudEvent.Data.Datacenter.Name
 		} else {
-			h.Logger.Error("Empty VM config")
-		}
+			h.Logger.Debug("found VM")
+			//prettyPrint(vmObject)

-	}
-	err = vc.Logout()
-	if err != nil {
-		h.Logger.Error("unable to logout of vcenter", "error", err)
-	}
+			// calculate VM properties we want to store
+			if vmObject.Vm.Config != nil {
+				numRam = vmObject.Vm.Config.Hardware.MemoryMB
+				numVcpus = vmObject.Vm.Config.Hardware.NumCPU * vmObject.Vm.Config.Hardware.NumCoresPerSocket
+			} else {
+				h.Logger.Error("Empty VM config")
+			}
+
+		}
+		err = vc.Logout()
+		if err != nil {
+			h.Logger.Error("unable to logout of vcenter", "error", err)
+		}
+	*/

 	// Create an instance of CreateInventoryParams
 	h.Logger.Debug("Creating database parameters")
-	params := queries.CreateInventoryParams{
-		Name:         vm.CloudEvent.Data.VM.Name,
-		Vcenter:      vm.CloudEvent.Source,
-		EventId:      sql.NullString{String: vm.CloudEvent.ID, Valid: vm.CloudEvent.ID != ""},
-		EventKey:     sql.NullString{String: strconv.Itoa(vm.CloudEvent.Data.Key), Valid: strconv.Itoa(vm.CloudEvent.Data.Key) != ""},
-		VmId:         sql.NullString{String: vm.CloudEvent.Data.VM.VM.Value, Valid: vm.CloudEvent.Data.VM.VM.Value != ""},
-		Datacenter:   sql.NullString{String: datacenter, Valid: datacenter != ""},
-		Cluster:      sql.NullString{String: vm.CloudEvent.Data.ComputeResource.Name, Valid: vm.CloudEvent.Data.ComputeResource.Name != ""},
-		CreationTime: sql.NullInt64{Int64: unixTimestamp, Valid: unixTimestamp > 0},
-		InitialVcpus: sql.NullInt64{Int64: int64(numVcpus), Valid: numVcpus > 0},
-		InitialRam:   sql.NullInt64{Int64: int64(numRam), Valid: numRam > 0},
+	/*
+		params := queries.CreateInventoryParams{
+			Name:         event.CloudEvent.Data.VM.Name,
+			Vcenter:      event.CloudEvent.Source,
+			EventId:      sql.NullString{String: event.CloudEvent.ID, Valid: event.CloudEvent.ID != ""},
+			EventKey:     sql.NullString{String: strconv.Itoa(event.CloudEvent.Data.Key), Valid: strconv.Itoa(event.CloudEvent.Data.Key) != ""},
+			VmId:         sql.NullString{String: event.CloudEvent.Data.VM.VM.Value, Valid: event.CloudEvent.Data.VM.VM.Value != ""},
+			Datacenter:   sql.NullString{String: datacenter, Valid: datacenter != ""},
+			Cluster:      sql.NullString{String: event.CloudEvent.Data.ComputeResource.Name, Valid: event.CloudEvent.Data.ComputeResource.Name != ""},
+			CreationTime: sql.NullInt64{Int64: unixTimestamp, Valid: unixTimestamp > 0},
+			InitialVcpus: sql.NullInt64{Int64: int64(numVcpus), Valid: numVcpus > 0},
+			InitialRam:   sql.NullInt64{Int64: int64(numRam), Valid: numRam > 0},
+		}
+	*/
+
+	params2 := queries.CreateEventParams{
+		Source:          event.CloudEvent.Source,
+		CloudId:         event.CloudEvent.ID,
+		EventTime:       sql.NullInt64{Int64: unixTimestamp, Valid: unixTimestamp > 0},
+		ChainId:         strconv.Itoa(event.CloudEvent.Data.ChainID),
+		VmId:            sql.NullString{String: event.CloudEvent.Data.VM.VM.Value, Valid: event.CloudEvent.Data.VM.VM.Value != ""},
+		EventKey:        sql.NullString{String: strconv.Itoa(event.CloudEvent.Data.Key), Valid: strconv.Itoa(event.CloudEvent.Data.Key) != ""},
+		Datacenter:      sql.NullString{String: event.CloudEvent.Data.Datacenter.Name, Valid: event.CloudEvent.Data.Datacenter.Name != ""},
+		ComputeResource: sql.NullString{String: event.CloudEvent.Data.ComputeResource.Name, Valid: event.CloudEvent.Data.ComputeResource.Name != ""},
+		UserName:        sql.NullString{String: event.CloudEvent.Data.UserName, Valid: event.CloudEvent.Data.UserName != ""},
 	}

-	h.Logger.Debug("database params", "params", params)
+	h.Logger.Debug("database params", "params", params2)

 	// Insert the new inventory record into the database
-	result, err := h.Database.Queries().CreateInventory(context.Background(), params)
+	result, err := h.Database.Queries().CreateEvent(context.Background(), params2)
 	if err != nil {
 		h.Logger.Error("unable to perform database insert", "error", err)
 		w.WriteHeader(http.StatusInternalServerError)