merch-tracker-v2/task-processor
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

parsers registration change
All checks were successful
/ Make image (push) Successful in 39s
Details

Browse source
This commit is contained in:
nquidox 2025-12-26 18:49:22 +03:00
parent 4aa4daa295
commit fedd9f1bbf
2 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
internal/app/app.go
View file

@ -71,14 +71,18 @@ func (app *App) Run() {
surugayaScrapper := newSurugayaScrapperClient(app.config.GrpcCfg.SurugayaScrapperHost + ":" + app.config.GrpcCfg.SurugayaScrapperPort)
//task processor
handlers := map[string]parsers.TaskHandler{
shared.OriginSurugaya: parsers.NewSurugayaParser(ctx, surugayaScrapper),
handlers := make(map[string]parsers.TaskHandler)
shared.OriginMandarake: mandarake.NewParser(mandarake.ParserDeps{
if app.config.OriginEnabled.Surugaya {
handlers[shared.OriginSurugaya] = parsers.NewSurugayaParser(ctx, surugayaScrapper)
}
if app.config.OriginEnabled.Mandarake {
handlers[shared.OriginMandarake] = mandarake.NewParser(mandarake.ParserDeps{
Enabled: app.config.OriginEnabled.Mandarake,
ExternalBrowser: app.config.ExternalBrowser,
GoroutinesNumber: app.numCPUs,
}),
})
}
taskProcessor := processor.New(processor.Deps{

2
internal/processor/service.go
View file

@ -26,10 +26,8 @@ func (p *Processor) StartWork(receivedTasks []shared.TaskResponse) {
wg.Add(1)
go func(origin string, tasks []shared.Task) {
defer wg.Done()
if p.handlers[origin] != nil {
log.Info("Running task handler for origin: ", origin)
p.handlers[origin].HandleTasks(tasks, p.out, p.state)
}
}(origin, tasks)
}
wg.Wait()