improve performance of a couple loops

cmd/cloudexec/clean.go

@@ -15,13 +15,13 @@ func ConfirmDeleteDroplets(config config.Config, dropletName string, instanceToJ
 	if err != nil {
 		return confirmedToDelete, fmt.Errorf("Failed to get droplets by name: %w", err)
 	}
+	if instanceToJobs == nil {
+		return confirmedToDelete, fmt.Errorf("Given instanceToJobs argument must not be nil")
+	}
 	if len(instances) > 0 {
 		fmt.Printf("Existing %s instance(s) found:\n", dropletName)
 		for _, instance := range instances {
 			// get a pretty string describing the jobs associated with this instance
-			if instanceToJobs == nil {
-				return confirmedToDelete, fmt.Errorf("Given instanceToJobs argument must not be nil")
-			}
 			jobs := instanceToJobs[int64(instance.ID)]
 			var prettyJobs string
 			if len(jobs) == 0 {

pkg/state/state.go

@@ -178,16 +178,19 @@ func (s *State) DeleteJob(jobID int64) {
 	}
 }
 
-func (s *State) CancelRunningJobs(config config.Config, bucketName string, toCancel []int64) error {
+func (s *State) CancelRunningJobs(config config.Config, bucketName string, toCancelSlice []int64) error {
+	// Gather list of ids to mark as cancelled
+	toCancel := make(map[int64]bool)
+	for _, id := range toCancelSlice {
+		toCancel[id] = true
+	}
+
 	// Mark any running jobs as cancelled
 	for i, job := range s.Jobs {
 		if job.Status == Running || job.Status == Provisioning {
-			for _, id := range toCancel {
-				if id == job.ID {
-					fmt.Printf("Setting status of job %d to 'Cancelled'\n", job.ID)
-					s.Jobs[i].Status = Cancelled
-					break
-				}
+			if _, exists := toCancel[job.ID]; exists {
+				fmt.Printf("Setting status of job %d to 'Cancelled'\n", job.ID)
+				s.Jobs[i].Status = Cancelled
 			}
 		}
 	}