From b1bd827574a0345ed65c518ad959eb6b15c7cdee Mon Sep 17 00:00:00 2001
From: Kirillkriv <kirill.krivoy@icloud.com>
Date: Fri, 9 Oct 2026 11:25:05 +0300
Subject: [PATCH] chore: long poller configuration options

---
 v1/core/dispatcher.go            |  3 ++-
 v1/core/{ => pollers}/pollers.go | 35 +++++++++++++++++++++++++++++---
 2 files changed, 34 insertions(+), 4 deletions(-)
 rename v1/core/{ => pollers}/pollers.go (88%)

diff --git a/v1/core/dispatcher.go b/v1/core/dispatcher.go
index 6ddf25a..57828e9 100644
--- a/v1/core/dispatcher.go
+++ b/v1/core/dispatcher.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	mcontext "github.com/abelith/gomax/v1/context"
+	"github.com/abelith/gomax/v1/core/pollers"
 	"github.com/abelith/gomax/v1/fsm/memory"
 	"github.com/abelith/gomax/v1/fsm/state"
 	"github.com/abelith/gomax/v1/isolation"
@@ -258,7 +259,7 @@ func WithZerolog(logger zerolog.Logger) DispatcherOption {
 
 func newDispatcherOptions(api *maxbot.Api) *dispatcherOptions {
 	return &dispatcherOptions{
-		poller:         NewLongPoller(api),
+		poller:         pollers.NewLongPoller(api),
 		fsmStorage:     memory.NewFSMStorage(),
 		eventIsolation: memory.NewEventIsolation(defaultTTL),
 		logger:         zerolog.Nop(),
diff --git a/v1/core/pollers.go b/v1/core/pollers/pollers.go
similarity index 88%
rename from v1/core/pollers.go
rename to v1/core/pollers/pollers.go
index 670af9e..53ba46d 100644
--- a/v1/core/pollers.go
+++ b/v1/core/pollers/pollers.go
@@ -1,4 +1,4 @@
-package core
+package pollers
 
 import (
 	"context"
@@ -6,6 +6,7 @@ import (
 	"errors"
 	maxbot "github.com/max-messenger/max-bot-api-client-go/v2"
 	"github.com/max-messenger/max-bot-api-client-go/v2/model"
+	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
 	"net"
 	"net/http"
@@ -35,6 +36,7 @@ type Webhook struct {
 type LongPoller struct {
 	api          *maxbot.Api
 	pollInterval time.Duration
+	logger       zerolog.Logger
 	wg           sync.WaitGroup
 }
 
@@ -57,6 +59,11 @@ type webhookOptions struct {
 
 type longPollerOptions struct {
 	pollInterval time.Duration
+	logger       zerolog.Logger
+}
+
+type longPollerOptionWrapper struct {
+	fn func(opts *longPollerOptions)
 }
 
 type webhookOptionWrapper struct {
@@ -143,6 +150,7 @@ func NewLongPoller(api *maxbot.Api, options ...LongPollerOption) *LongPoller {
 	}
 
 	lp.pollInterval = opts.pollInterval
+	lp.logger = opts.logger
 
 	return lp
 }
@@ -201,6 +209,18 @@ func (tp *TestPoller) Stop(_ context.Context) error {
 	return nil
 }
 
+func WithPollInterval(interval time.Duration) LongPollerOption {
+	return newLpOption(func(opts *longPollerOptions) {
+		opts.pollInterval = interval
+	})
+}
+
+func WithZerolog(logger zerolog.Logger) LongPollerOption {
+	return newLpOption(func(opts *longPollerOptions) {
+		opts.logger = logger
+	})
+}
+
 func WithAddress(addr string) WebhookOption {
 	return newWhOption(func(opts *webhookOptions) {
 		opts.addr = addr
@@ -271,9 +291,9 @@ func (lp *LongPoller) poll(ctx context.Context, ch chan model.Update) {
 			updates, marker, err = lp.api.Subscriptions.GetUpdates(ctx, marker)
 			latency := time.Since(start)
 			if err != nil && !errors.Is(err, context.Canceled) {
-				log.Error().Err(err).Send()
+				lp.logger.Error().Err(err).Send()
 			}
-			log.Ctx(ctx).Debug().
+			lp.logger.Debug().
 				Int("received", len(updates)).
 				Dur("latency", latency).
 				Msg("received updates")
@@ -290,6 +310,7 @@ func (lp *LongPoller) poll(ctx context.Context, ch chan model.Update) {
 func newLongPollerOptions() *longPollerOptions {
 	return &longPollerOptions{
 		pollInterval: defaultInterval,
+		logger:       zerolog.Nop(),
 	}
 }
 
@@ -299,10 +320,18 @@ func newWebhookOptions() *webhookOptions {
 	}
 }
 
+func newLpOption(fn func(opts *longPollerOptions)) LongPollerOption {
+	return longPollerOptionWrapper{fn: fn}
+}
+
 func newWhOption(fn func(options *webhookOptions)) WebhookOption {
 	return webhookOptionWrapper{fn: fn}
 }
 
+func (low longPollerOptionWrapper) apply(opts *longPollerOptions) {
+	low.fn(opts)
+}
+
 func (wow webhookOptionWrapper) apply(opts *webhookOptions) {
 	wow.fn(opts)
 }