From 67879cf7b0091d60e0d1cd76ca7506cf9a3c4468 Mon Sep 17 00:00:00 2001
From: Bel LaPointe <153096461+breel-render@users.noreply.github.com>
Date: Mon, 8 Dec 2025 16:35:16 -0700
Subject: [PATCH] fix loop

---
 cmd/pg-fill/main.go | 26 +++++++-------------------
 1 file changed, 7 insertions(+), 19 deletions(-)

diff --git a/cmd/pg-fill/main.go b/cmd/pg-fill/main.go
index 7a066fe..d406802 100644
--- a/cmd/pg-fill/main.go
+++ b/cmd/pg-fill/main.go
@@ -57,25 +57,13 @@ func run(ctx context.Context) error {
 
 		// https://gist.github.com/ololobus/5b25c432f208d7eb31051a5f238dffff
 		// 2e6=1GB, so 2e6/8=12MB
-		for {
-			if presently() >= *n {
-				break
-			}
-			for ctx.Err() == nil {
-				if _, err := pg.ExecContext(ctx, `
-			   	INSERT INTO fill_with_data (x, y, z)
-			   	SELECT ROUND(RANDOM()), RANDOM(), RANDOM()
-			   	FROM generate_series(1, 2e6/8)
-			   `); err != nil {
-					return fmt.Errorf("failed lo_from_bytea: %w", err)
-				} else {
-					break
-				}
-				select {
-				case <-ctx.Done():
-					return ctx.Err()
-				default:
-				}
+		for ctx.Err() == nil && presently() < *n {
+			if _, err := pg.ExecContext(ctx, `
+		   	INSERT INTO fill_with_data (x, y, z)
+		   	SELECT ROUND(RANDOM()), RANDOM(), RANDOM()
+		   	FROM generate_series(1, 2e6/8)
+		   `); err != nil {
+				return fmt.Errorf("failed lo_from_bytea: %w", err)
 			}
 		}