test cron job

internal/tasks/processEvents.go

@@ -1,3 +1,76 @@
 package tasks
 
+import (
+	"context"
+	"log/slog"
+	"time"
+	"vctp/db"
+	"vctp/internal/vcenter"
+)
+
+// Handler handles requests.
+type CronTask struct {
+	Logger   *slog.Logger
+	Database db.Database
+}
+
 // use gocron to check events in the Events table
+func (c *CronTask) RunVmCheck(ctx context.Context, logger *slog.Logger) error {
+	var (
+		//unixTimestamp int64
+		numVcpus   int32
+		numRam     int32
+		datacenter string
+	)
+
+	logger.Debug("Started Events processing", "time", time.Now())
+
+	// Query events table
+	events, err := c.Database.Queries().ListUnprocessedEvents(ctx)
+	if err != nil {
+		logger.Error("Unable to query for unprocessed events", "error", err)
+		return nil // TODO - what to do with this error?
+	}
+
+	for _, evt := range events {
+		logger.Debug("Checking event", "event", evt)
+
+		c.Logger.Debug("connecting to vcenter")
+		vc := vcenter.New(c.Logger)
+		vc.Login(evt.Source)
+		//vmObject, err := vc.FindVMByName(vm.CloudEvent.Data.VM.Name)
+		//vmObject, err := vc.FindVMByID(vm.CloudEvent.Data.VM.VM.Value)
+		vmObject, err := vc.FindVMByIDWithDatacenter(evt.VmId.String, evt.DatacenterId.String)
+
+		if err != nil {
+			c.Logger.Error("Can't locate vm in vCenter", "vmID", evt.VmId.String, "error", err)
+		} else if vmObject == nil {
+			c.Logger.Debug("didn't find VM", "vm_id", evt.VmId.String)
+			numRam = 0
+			numVcpus = 0
+			datacenter = evt.DatacenterName.String
+		} else {
+			c.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
+			} else {
+				c.Logger.Error("Empty VM config")
+			}
+
+		}
+		err = vc.Logout()
+		if err != nil {
+			c.Logger.Error("unable to logout of vcenter", "error", err)
+		}
+
+		c.Logger.Debug("Simulate adding to Inventory", "vm_name", evt.VmName.String, "vcpus", numVcpus, "ram", numRam, "dc", datacenter)
+
+	}
+
+	//fmt.Printf("processing at %s", time.Now())
+	return nil
+}

internal/vcenter/vcenter.go

@@ -24,8 +24,8 @@ type Vcenter struct {
 }
 
 type VmProperties struct {
-	Vm         mo.VirtualMachine
-	Datacenter string
+	Vm mo.VirtualMachine
+	//Datacenter string
 }
 
 // New creates a new Vcenter with the given logger
@@ -159,14 +159,14 @@ func (v *Vcenter) FindVMByID(vmID string) (*VmProperties, error) {
 
 		if err == nil {
 			return &VmProperties{
-				Datacenter: dc.Name(),
-				Vm:         vm,
+				//Datacenter: dc.Name(),
+				Vm: vm,
 			}, nil
 		} else if _, ok := err.(*find.NotFoundError); !ok {
 			// If the error is not a NotFoundError, return it
 			//return nil, fmt.Errorf("failed to retrieve VM with ID %s in datacenter %s: %w", vmID, dc.Name(), err)
 			v.Logger.Debug("Couldn't find vm in datacenter", "vm_id", vmID, "datacenter_name", dc.Name())
-		} else if err != nil {
+		} else {
 			return nil, fmt.Errorf("failed to retrieve VM: %w", err)
 		}
 	}
@@ -175,7 +175,7 @@ func (v *Vcenter) FindVMByID(vmID string) (*VmProperties, error) {
 }
 
 func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmProperties, error) {
-	var dcName string
+	//var dcName string
 	var err error
 	v.Logger.Debug("searching for vm id", "vm_id", vmID, "datacenter_id", dcID)
 
@@ -196,11 +196,13 @@ func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmPropert
 	// Use finder.SetDatacenter to set the datacenter
 	finder.SetDatacenter(datacenter)
 
-	dcName, err = datacenter.ObjectName(v.ctx)
-	if err != nil {
-		v.Logger.Error("Couldn't find the name of the datacenter", "error", err)
-		dcName = ""
-	}
+	/*
+		dcName, err = datacenter.ObjectName(v.ctx)
+		if err != nil {
+			v.Logger.Error("Couldn't find the name of the datacenter", "error", err)
+			dcName = ""
+		}
+	*/
 
 	// Create a ManagedObjectReference for the VM
 	vmRef := types.ManagedObjectReference{
@@ -214,14 +216,14 @@ func (v *Vcenter) FindVMByIDWithDatacenter(vmID string, dcID string) (*VmPropert
 	if err == nil {
 		v.Logger.Debug("Found VM", "vm", vm)
 		return &VmProperties{
-			Datacenter: dcName,
-			Vm:         vm,
+			//Datacenter: dcName,
+			Vm: vm,
 		}, nil
 	} else if _, ok := err.(*find.NotFoundError); !ok {
 		// If the error is not a NotFoundError, return it
 		//return nil, fmt.Errorf("failed to retrieve VM with ID %s in datacenter %s: %w", vmID, dc.Name(), err)
-		v.Logger.Debug("Couldn't find vm in datacenter", "vm_id", vmID, "datacenter_name", dcName)
-	} else if err != nil {
+		v.Logger.Debug("Couldn't find vm in datacenter", "vm_id", vmID, "datacenter_id", dcID)
+	} else {
 		return nil, fmt.Errorf("failed to retrieve VM: %w", err)
 	}