mirror of
https://github.com/makayabou/asg-server.git
synced 2026-05-02 17:43:36 +02:00
110 lines
2.3 KiB
Go
110 lines
2.3 KiB
Go
package push
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/android-sms-gateway/server/pkg/types/cache"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
"go.uber.org/fx"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type Config struct {
|
|
Mode Mode
|
|
|
|
ClientOptions map[string]string
|
|
|
|
Debounce time.Duration
|
|
Timeout time.Duration
|
|
}
|
|
|
|
type Params struct {
|
|
fx.In
|
|
|
|
Config Config
|
|
|
|
Client client
|
|
|
|
Logger *zap.Logger
|
|
}
|
|
|
|
type Service struct {
|
|
config Config
|
|
|
|
client client
|
|
|
|
cache *cache.Cache[Event]
|
|
|
|
enqueuedCounter *prometheus.CounterVec
|
|
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func New(params Params) *Service {
|
|
if params.Config.Timeout == 0 {
|
|
params.Config.Timeout = time.Second
|
|
}
|
|
if params.Config.Debounce < 5*time.Second {
|
|
params.Config.Debounce = 5 * time.Second
|
|
}
|
|
|
|
enqueuedCounter := promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: "sms",
|
|
Subsystem: "push",
|
|
Name: "enqueued_total",
|
|
Help: "Total number of messages enqueued",
|
|
}, []string{"event"})
|
|
|
|
return &Service{
|
|
config: params.Config,
|
|
client: params.Client,
|
|
cache: cache.New[Event](cache.Config{}),
|
|
enqueuedCounter: enqueuedCounter,
|
|
logger: params.Logger,
|
|
}
|
|
}
|
|
|
|
// Run runs the service with the provided context if a debounce is set.
|
|
func (s *Service) Run(ctx context.Context) {
|
|
ticker := time.NewTicker(s.config.Debounce)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
s.sendAll(ctx)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Enqueue adds the data to the cache and immediately sends all messages if the debounce is 0.
|
|
func (s *Service) Enqueue(token string, event *Event) error {
|
|
if err := s.cache.Set(token, *event); err != nil {
|
|
return fmt.Errorf("can't add message to cache: %w", err)
|
|
}
|
|
|
|
s.enqueuedCounter.WithLabelValues(string(event.Event)).Inc()
|
|
|
|
return nil
|
|
}
|
|
|
|
// sendAll sends messages to all targets from the cache after initializing the service.
|
|
func (s *Service) sendAll(ctx context.Context) {
|
|
targets := s.cache.Drain()
|
|
if len(targets) == 0 {
|
|
return
|
|
}
|
|
|
|
s.logger.Info("Sending messages", zap.Int("count", len(targets)))
|
|
ctx, cancel := context.WithTimeout(ctx, s.config.Timeout)
|
|
if err := s.client.Send(ctx, targets); err != nil {
|
|
s.logger.Error("Can't send messages", zap.Error(err))
|
|
}
|
|
cancel()
|
|
}
|
