70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package data_updater
|
|
|
|
import (
|
|
"amocrm/internal/repository"
|
|
"amocrm/pkg/amoClient"
|
|
"context"
|
|
"go.uber.org/zap"
|
|
"time"
|
|
)
|
|
|
|
type Deps struct {
|
|
AmoClient *amoClient.Amo
|
|
Repo *repository.Repository
|
|
Logger *zap.Logger
|
|
}
|
|
|
|
type DataUpdater struct {
|
|
amoClient *amoClient.Amo
|
|
repo *repository.Repository
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewDataUpdaterWC(deps Deps) *DataUpdater {
|
|
return &DataUpdater{
|
|
amoClient: deps.AmoClient,
|
|
repo: deps.Repo,
|
|
logger: deps.Logger,
|
|
}
|
|
}
|
|
|
|
func (wc *DataUpdater) Start(ctx context.Context) {
|
|
nextStart := calculateTime()
|
|
ticker := time.NewTicker(time.Second * time.Duration(nextStart))
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
wc.processTasks(ctx)
|
|
nextStart = calculateTime()
|
|
ticker.Reset(time.Second * time.Duration(nextStart))
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (wc *DataUpdater) processTasks(ctx context.Context) {
|
|
// сначала получаем список токенов
|
|
allTokens, err := wc.repo.GetAllTokens(ctx)
|
|
if err != nil {
|
|
wc.logger.Error("error fetch all tokens from mongo:", zap.Error(err))
|
|
return
|
|
}
|
|
for _, token := range allTokens {
|
|
pipelines, err := wc.amoClient.GetListPipelines(token.AccessToken)
|
|
if err != nil {
|
|
wc.logger.Error("error getting list pipelines:", zap.Error(err))
|
|
}
|
|
err = wc.repo.CheckPipelines(ctx, token.AccountID, pipelines.Embedded.Pipelines)
|
|
if err != nil {
|
|
wc.logger.Error("error update pipelines in mongo:", zap.Error(err))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (wc *DataUpdater) Stop(ctx context.Context) error {
|
|
return nil
|
|
}
|