From cff32af15371c0165ad73a4209e216fa318a5163 Mon Sep 17 00:00:00 2001
From: Chris Gianelloni <wolf31o2@blinklabs.io>
Date: Fri, 4 Oct 2024 10:23:54 -0400
Subject: [PATCH] feat: pass logger to automaxprocs (#242)

Signed-off-by: Chris Gianelloni <wolf31o2@blinklabs.io>
---
 cmd/tx-submit-api/main.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/cmd/tx-submit-api/main.go b/cmd/tx-submit-api/main.go
index 48c30e1..7ed6068 100644
--- a/cmd/tx-submit-api/main.go
+++ b/cmd/tx-submit-api/main.go
@@ -21,7 +21,7 @@ import (
 	_ "net/http/pprof"
 	"os"
 
-	_ "go.uber.org/automaxprocs"
+	"go.uber.org/automaxprocs/maxprocs"
 
 	"github.com/blinklabs-io/tx-submit-api/internal/api"
 	"github.com/blinklabs-io/tx-submit-api/internal/config"
@@ -33,6 +33,10 @@ var cmdlineFlags struct {
 	configFile string
 }
 
+func logPrintf(format string, v ...any) {
+	logging.GetLogger().Infof(format, v...)
+}
+
 func main() {
 	flag.StringVar(
 		&cmdlineFlags.configFile,
@@ -63,6 +67,14 @@ func main() {
 
 	logger.Infof("starting tx-submit-api %s", version.GetVersionString())
 
+	// Configure max processes with our logger wrapper, toss undo func
+	_, err = maxprocs.Set(maxprocs.Logger(logPrintf))
+	if err != nil {
+		// If we hit this, something really wrong happened
+		logger.Errorf(err.Error())
+		os.Exit(1)
+	}
+
 	// Start debug listener
 	if cfg.Debug.ListenPort > 0 {
 		logger.Infof(