diff --git a/example_batch_insert_test.go b/example_batch_insert_test.go index 6ec5199d..547d1fef 100644 --- a/example_batch_insert_test.go +++ b/example_batch_insert_test.go @@ -62,10 +62,6 @@ func Example_batchInsert() { subscribeChan, subscribeCancel := riverClient.Subscribe(river.EventKindJobCompleted) defer subscribeCancel() - if err := riverClient.Start(ctx); err != nil { - panic(err) - } - results, err := riverClient.InsertMany(ctx, []river.InsertManyParams{ {Args: BatchInsertArgs{}}, {Args: BatchInsertArgs{}}, @@ -78,6 +74,12 @@ func Example_batchInsert() { } fmt.Printf("Inserted %d jobs\n", len(results)) + // Start the client after inserting and printing so that the "Inserted" + // message is guaranteed to appear before any "Worked" messages. + if err := riverClient.Start(ctx); err != nil { + panic(err) + } + // Wait for jobs to complete. Only needed for purposes of the example test. riversharedtest.WaitOrTimeoutN(testutil.PanicTB(), subscribeChan, 5)