Compare commits

..

No commits in common. "8395cf71b4a7d751987569c1e78c1eee97b1561c" and "13ebb273359f09639e8d73fab8f006fe7dd082b5" have entirely different histories.

6 changed files with 9 additions and 29 deletions

View file

@ -2,9 +2,6 @@ package main
import ( import (
"context" "context"
log "github.com/sirupsen/logrus"
"net/http"
_ "net/http/pprof"
"os" "os"
"os/signal" "os/signal"
"syscall" "syscall"
@ -21,12 +18,6 @@ func main() {
logging.LogSetup(c.LogLevel) logging.LogSetup(c.LogLevel)
if c.Metrics.GinMode != "release" {
go func() {
log.Println(http.ListenAndServe("localhost:6060", nil))
}()
}
appl := app.New(c) appl := app.New(c)
appl.Run(ctx) appl.Run(ctx)

View file

@ -14,7 +14,6 @@ GRPC_SURUGAYA_SCRAPPER_PORT=9070
TASK_RETRY_COUNT=3 TASK_RETRY_COUNT=3
TASK_RETRY_MINUTES=5 TASK_RETRY_MINUTES=5
TASK_TIMEOUT_MINUTES=5
ORIGIN_SURUGAYA_ENABLED=false ORIGIN_SURUGAYA_ENABLED=false
ORIGIN_MANDARAKE_ENABLED=false ORIGIN_MANDARAKE_ENABLED=false

View file

@ -29,7 +29,6 @@ type GrpcConfig struct {
type TasksConfig struct { type TasksConfig struct {
RetryCount int RetryCount int
RetryMinutes int RetryMinutes int
TaskTimeout int
} }
type OriginEnabled struct { type OriginEnabled struct {
@ -63,7 +62,6 @@ func NewConfig() *Config {
TasksCfg: TasksConfig{ TasksCfg: TasksConfig{
RetryCount: getEnvInt("TASK_RETRY_COUNT", 3), RetryCount: getEnvInt("TASK_RETRY_COUNT", 3),
RetryMinutes: getEnvInt("TASK_RETRY_MINUTES", 5), RetryMinutes: getEnvInt("TASK_RETRY_MINUTES", 5),
TaskTimeout: getEnvInt("TASK_TIMEOUT_MINUTES", 5),
}, },
OriginEnabled: OriginEnabled{ OriginEnabled: OriginEnabled{

View file

@ -75,6 +75,9 @@ func (app *App) Run(ctx context.Context) {
//main //main
apiClient := newApiClient(app.config.GrpcCfg.ApiClientHost + ":" + app.config.GrpcCfg.ApiClientPort) apiClient := newApiClient(app.config.GrpcCfg.ApiClientHost + ":" + app.config.GrpcCfg.ApiClientPort)
period := time.NewTicker(app.checkPeriod * time.Hour)
defer period.Stop()
sender := make(chan shared.TaskResult, app.numCPUs*10) sender := make(chan shared.TaskResult, app.numCPUs*10)
defer close(sender) defer close(sender)
@ -93,7 +96,6 @@ func (app *App) Run(ctx context.Context) {
Enabled: app.config.OriginEnabled.Mandarake, Enabled: app.config.OriginEnabled.Mandarake,
ExternalBrowser: app.config.ExternalBrowser, ExternalBrowser: app.config.ExternalBrowser,
GoroutinesNumber: app.numCPUs, GoroutinesNumber: app.numCPUs,
TaskTimeout: app.config.TasksCfg.TaskTimeout,
}) })
} }
@ -115,9 +117,6 @@ func (app *App) Run(ctx context.Context) {
taskProcessor.StartWork(ctx, receivedTasks) taskProcessor.StartWork(ctx, receivedTasks)
} }
period := time.NewTicker(app.checkPeriod * time.Hour)
defer period.Stop()
go func() { go func() {
process() //immediate start process() //immediate start
for range period.C { for range period.C {

View file

@ -16,6 +16,9 @@ func (s *Parser) HandleTasks(ctx context.Context, tasks []shared.Task, sender ch
allocCtx, allocCancel := chromedp.NewRemoteAllocator(ctx, s.externalBrowser) allocCtx, allocCancel := chromedp.NewRemoteAllocator(ctx, s.externalBrowser)
defer allocCancel() defer allocCancel()
sessionCtx, sessionCancel := chromedp.NewContext(allocCtx /* chromedp.WithLogf(log.Printf) */, chromedp.WithLogf(func(string, ...any) {}))
defer sessionCancel()
receiver := make(chan shared.Task, len(tasks)) receiver := make(chan shared.Task, len(tasks))
for _, task := range tasks { for _, task := range tasks {
receiver <- task receiver <- task
@ -28,7 +31,7 @@ func (s *Parser) HandleTasks(ctx context.Context, tasks []shared.Task, sender ch
wg.Add(1) wg.Add(1)
go func() { go func() {
defer wg.Done() defer wg.Done()
s.worker(allocCtx, receiver, sender) s.worker(sessionCtx, receiver, sender)
}() }()
} }
wg.Wait() wg.Wait()
@ -39,20 +42,14 @@ func (s *Parser) HandleTasks(ctx context.Context, tasks []shared.Task, sender ch
func (s *Parser) worker(ctx context.Context, receiver chan shared.Task, sender chan shared.TaskResult) { func (s *Parser) worker(ctx context.Context, receiver chan shared.Task, sender chan shared.TaskResult) {
for task := range receiver { for task := range receiver {
taskCtx, taskCancel := chromedp.NewContext(ctx /* chromedp.WithLogf(log.Printf) */, chromedp.WithLogf(func(string, ...any) {}))
timeoutCtx, timeoutCancel := context.WithTimeout(taskCtx, s.taskTimeout)
log.WithField("task_uuid", task.MerchUuid).Infof("%v %v processing task", logHeader, logWorker) log.WithField("task_uuid", task.MerchUuid).Infof("%v %v processing task", logHeader, logWorker)
//price will be zeroPrice value in case of any error or if price not found //price will be zeroPrice value in case of any error or if price not found
price := s.getMinimalPrice(timeoutCtx, task) price := s.getMinimalPrice(ctx, task)
sender <- shared.TaskResult{ sender <- shared.TaskResult{
MerchUuid: task.MerchUuid, MerchUuid: task.MerchUuid,
Origin: task.Origin, Origin: task.Origin,
Price: price, Price: price,
} }
timeoutCancel()
taskCancel()
} }
} }

View file

@ -2,7 +2,6 @@ package mandarake
import ( import (
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"time"
) )
const ( const (
@ -17,25 +16,22 @@ const (
type Parser struct { type Parser struct {
externalBrowser string externalBrowser string
goroutinesNumber int goroutinesNumber int
taskTimeout time.Duration
} }
type Deps struct { type Deps struct {
Enabled bool Enabled bool
ExternalBrowser string ExternalBrowser string
GoroutinesNumber int GoroutinesNumber int
TaskTimeout int
} }
func NewParser(deps Deps) *Parser { func NewParser(deps Deps) *Parser {
if !deps.Enabled { if !deps.Enabled {
log.Infof("%v disabled", logHeader) log.Info(logHeader + "disabled")
return nil return nil
} }
return &Parser{ return &Parser{
externalBrowser: deps.ExternalBrowser, externalBrowser: deps.ExternalBrowser,
goroutinesNumber: deps.GoroutinesNumber, goroutinesNumber: deps.GoroutinesNumber,
taskTimeout: time.Minute * time.Duration(deps.TaskTimeout),
} }
} }