ref: batch size percent CLI configuration

this makes the loadCert batch size configuration via the
`--batch-size-percent` variable with a default value of 1%. I found that
to be pretty good on my large Vault installation

cmd/main.go

@@ -2,12 +2,13 @@ package main
 
 import (
 	"fmt"
+	"time"
+
 	log "github.com/aarnaud/vault-pki-exporter/pkg/logger"
 	"github.com/aarnaud/vault-pki-exporter/pkg/vault"
 	vaultMon "github.com/aarnaud/vault-pki-exporter/pkg/vault-mon"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
-	"time"
 )
 
 var version string
@@ -60,6 +61,11 @@ func init() {
 	if err := viper.BindPFlag("refresh_interval", flags.Lookup("refresh-interval")); err != nil {
 		log.Fatal(err)
 	}
+
+	flags.Float64("batch-size-percent", 1, "loadCerts batch size percentage")
+	if err := viper.BindPFlag("batch_size_percent", flags.Lookup("batch-size-percent")); err != nil {
+		log.Fatal(err)
+	}
 }
 
 func main() {

pkg/vault-mon/pki.go

@@ -154,9 +154,11 @@ func (pki *PKI) loadCerts(loadCertsDuration prometheus.Histogram) error {
 	// reset expired certs to avoid counter creep
 	pki.expiredCertsCounter = 0
 
-	// Determine batch size based on the length of serialsList.Keys.
-	// todo make batch size CLI configurable
-	batchSize := len(serialsList.Keys) / 20 // 5% of the total certificates in each batch
+	// determine batch size dynamically based on the length of serialsList.Keys
+	batchSizePercentage := viper.GetFloat64("batch_size_percent")
+
+	// Calculate the batch size using floating-point division, then round to the nearest integer
+	batchSize := int(float64(len(serialsList.Keys)) * (batchSizePercentage / 100.0))
 	if batchSize < 1 {
 		batchSize = 1
 	}