Compare commits
No commits in common. "8395cf71b4a7d751987569c1e78c1eee97b1561c" and "13ebb273359f09639e8d73fab8f006fe7dd082b5" have entirely different histories.
8395cf71b4
...
13ebb27335
6 changed files with 9 additions and 29 deletions
|
|
@ -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)
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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{
|
||||||
|
|
|
||||||
|
|
@ -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 {
|
||||||
|
|
|
||||||
|
|
@ -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()
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -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),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue