From c01721e3baf978f50fd72144e1cca06c14abe479 Mon Sep 17 00:00:00 2001
From: WGH <wgh@torlan.ru>
Date: Fri, 21 Feb 2020 22:28:34 +0300
Subject: [PATCH] Add Len() method to *pgx.Batch

This makes the API slightly easier to use when number of calls to
Queue() cannot be trivially computed.

For example, if the program contains the loop like the following,
a separate variable counting the iterations is needed:

    numHeaders := 0
    for _, header := range prepareHeadersForInsert(*res.Headers) {
        headerBatch.Queue("INSERT ...", ...)
        numHeaders++
    }

    headerBatchResult := tx.SendBatch(ctx, headerBatch)

    for i := 0; i < numHeaders; i++ {
        _, err := headerBatchResult.Exec()
        // ...
    }

With method Len(), this extra variable can be eliminated.
---
 batch.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/batch.go b/batch.go
index 2f85b852..d46a478e 100644
--- a/batch.go
+++ b/batch.go
@@ -26,6 +26,11 @@ func (b *Batch) Queue(query string, arguments ...interface{}) {
 	})
 }
 
+// Len returns number of queries that have been queued so far.
+func (b *Batch) Len() int {
+	return len(b.items)
+}
+
 type BatchResults interface {
 	// Exec reads the results from the next query in the batch as if the query has been sent with Conn.Exec.
 	Exec() (pgconn.CommandTag, error)