Compare commits

...

7 commits
v0.2.3 ... main

Author SHA1 Message Date
nquidox
9361319f0a declare queue with opts quick fix
All checks were successful
/ Make image (push) Successful in 44s
2026-04-27 00:18:39 +03:00
nquidox
c8a384efcb update
All checks were successful
/ Make image (push) Successful in 39s
2026-04-08 12:28:56 +03:00
nquidox
98bb6a5bc0 update
Some checks failed
/ Make image (push) Failing after 1m53s
2026-04-08 12:16:08 +03:00
nquidox
b29e60b712 log enabled dep 2026-04-08 12:15:24 +03:00
nquidox
c5013c03ff links validator
All checks were successful
/ Make image (push) Successful in 37s
2026-04-06 17:05:00 +03:00
nquidox
af1710c3fe env name fix
All checks were successful
/ Make image (push) Successful in 37s
2026-04-06 12:15:32 +03:00
nquidox
047b61c9a4 update
All checks were successful
/ Make image (push) Successful in 44s
2026-04-06 12:03:03 +03:00
10 changed files with 125 additions and 58 deletions

View file

@ -1,20 +1,12 @@
FROM golang:1.26.1-alpine3.23 AS builder FROM golang:1.26.1-alpine3.23 AS builder
RUN apk add --no-cache git
ENV GOPRIVATE=repo.nqws.ru/*
ENV GONOSUMDB=repo.nqws.ru/*
WORKDIR /app WORKDIR /app
COPY go.mod go.sum ./ COPY go.mod go.sum ./
RUN go mod download
RUN --mount=type=secret,id=git_creds,dst=/root/.git-credentials,mode=0600 \
git config --global credential.helper store && \
go mod download
COPY . . COPY . .
RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -trimpath -ldflags="-s -w" -o main "./cmd" RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -trimpath -ldflags="-s -w" -o main "./cmd"
FROM alpine:3.23 FROM alpine:3.23
RUN apk add --no-cache \ RUN apk add --no-cache \
@ -26,4 +18,4 @@ COPY --from=builder /app/main /usr/local/bin/app
RUN chmod +x /usr/local/bin/app RUN chmod +x /usr/local/bin/app
ENTRYPOINT ["app"] ENTRYPOINT ["app"]

View file

@ -18,4 +18,5 @@ RABBIT_HOST=
RABBIT_PORT= RABBIT_PORT=
RABBIT_USER= RABBIT_USER=
RABBIT_PASS= RABBIT_PASS=
RABBIT_VHOST= RABBIT_VHOST=
RABBIT_LOGGING_ENABLED=

View file

@ -28,11 +28,12 @@ type TasksSource struct {
} }
type Rabbit struct { type Rabbit struct {
Host string Host string
Port uint16 Port uint16
User string User string
Pass string Pass string
Vhost string Vhost string
LoggingEnabled bool
} }
func NewConfig() Config { func NewConfig() Config {
@ -50,17 +51,20 @@ func NewConfig() Config {
}, },
TasksSource: TasksSource{ TasksSource: TasksSource{
Host: getEnv("TASK_API_HOST", "127.0.0.1"), //Host: getEnv("TASK_SOURCE_HOST", "127.0.0.1"),
Port: getEnv("TASK_API_PORT", "61000"), //Port: getEnv("TASK_SOURCE_PORT", "61000"),
Host: getEnv("TASK_SOURCE_HOST", "10.0.0.1"),
Port: getEnv("TASK_SOURCE_PORT", "9099"),
Timeout: getEnvSeconds("TASK_SOURCE_TIMEOUT_SECONDS", 60), Timeout: getEnvSeconds("TASK_SOURCE_TIMEOUT_SECONDS", 60),
}, },
Rabbit: Rabbit{ Rabbit: Rabbit{
Host: getEnv("RABBIT_HOST", "10.0.0.4"), Host: getEnv("RABBIT_HOST", "10.0.0.4"),
Port: getEnvPort("RABBIT_PORT", 5672), Port: getEnvPort("RABBIT_PORT", 5672),
User: getEnv("RABBIT_USER", "taskProcessorDev"), User: getEnv("RABBIT_USER", "taskProcessorDev"),
Pass: getEnv("RABBIT_PASS", "pass1234"), Pass: getEnv("RABBIT_PASS", "pass1234"),
Vhost: getEnv("RABBIT_VHOST", "taskProcessorDevHost"), Vhost: getEnv("RABBIT_VHOST", "taskProcessorDevHost"),
LoggingEnabled: getEnvBool("RABBIT_LOGGING_ENABLED", true),
}, },
} }
} }

View file

@ -50,3 +50,14 @@ func getEnvPort(key string, fallback uint16) uint16 {
} }
return fallback return fallback
} }
func getEnvBool(key string, fallback bool) bool {
if value, ok := os.LookupEnv(key); ok {
val, err := strconv.ParseBool(value)
if err != nil {
return fallback
}
return val
}
return fallback
}

4
go.mod
View file

@ -6,7 +6,7 @@ require (
github.com/sirupsen/logrus v1.9.4 github.com/sirupsen/logrus v1.9.4
google.golang.org/grpc v1.80.0 google.golang.org/grpc v1.80.0
google.golang.org/protobuf v1.36.11 google.golang.org/protobuf v1.36.11
repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.16 repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.19
) )
require ( require (
@ -15,5 +15,5 @@ require (
golang.org/x/sys v0.42.0 // indirect golang.org/x/sys v0.42.0 // indirect
golang.org/x/text v0.35.0 // indirect golang.org/x/text v0.35.0 // indirect
golang.org/x/time v0.15.0 // indirect golang.org/x/time v0.15.0 // indirect
google.golang.org/genproto/googleapis/rpc v0.0.0-20260401024825-9d38bb4040a9 // indirect google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d // indirect
) )

8
go.sum
View file

@ -44,13 +44,13 @@ golang.org/x/time v0.15.0 h1:bbrp8t3bGUeFOx08pvsMYRTCVSMk89u4tKbNOZbp88U=
golang.org/x/time v0.15.0/go.mod h1:Y4YMaQmXwGQZoFaVFk4YpCt4FLQMYKZe9oeV/f4MSno= golang.org/x/time v0.15.0/go.mod h1:Y4YMaQmXwGQZoFaVFk4YpCt4FLQMYKZe9oeV/f4MSno=
gonum.org/v1/gonum v0.17.0 h1:VbpOemQlsSMrYmn7T2OUvQ4dqxQXU+ouZFQsZOx50z4= gonum.org/v1/gonum v0.17.0 h1:VbpOemQlsSMrYmn7T2OUvQ4dqxQXU+ouZFQsZOx50z4=
gonum.org/v1/gonum v0.17.0/go.mod h1:El3tOrEuMpv2UdMrbNlKEh9vd86bmQ6vqIcDwxEOc1E= gonum.org/v1/gonum v0.17.0/go.mod h1:El3tOrEuMpv2UdMrbNlKEh9vd86bmQ6vqIcDwxEOc1E=
google.golang.org/genproto/googleapis/rpc v0.0.0-20260401024825-9d38bb4040a9 h1:m8qni9SQFH0tJc1X0vmnpw/0t+AImlSvp30sEupozUg= google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d h1:wT2n40TBqFY6wiwazVK9/iTWbsQrgk5ZfCSVFLO9LQA=
google.golang.org/genproto/googleapis/rpc v0.0.0-20260401024825-9d38bb4040a9/go.mod h1:4Hqkh8ycfw05ld/3BWL7rJOSfebL2Q+DVDeRgYgxUU8= google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d/go.mod h1:4Hqkh8ycfw05ld/3BWL7rJOSfebL2Q+DVDeRgYgxUU8=
google.golang.org/grpc v1.80.0 h1:Xr6m2WmWZLETvUNvIUmeD5OAagMw3FiKmMlTdViWsHM= google.golang.org/grpc v1.80.0 h1:Xr6m2WmWZLETvUNvIUmeD5OAagMw3FiKmMlTdViWsHM=
google.golang.org/grpc v1.80.0/go.mod h1:ho/dLnxwi3EDJA4Zghp7k2Ec1+c2jqup0bFkw07bwF4= google.golang.org/grpc v1.80.0/go.mod h1:ho/dLnxwi3EDJA4Zghp7k2Ec1+c2jqup0bFkw07bwF4=
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE= google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco= google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.16 h1:bqoQZr4kblRyGQFjqBItakGm/p2PP0+68U12lLnuIYM= repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.19 h1:IKxMEU0Awn8cnC+x9Ptw7t2lsK8WLr4MKSKq2RN/hJY=
repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.16/go.mod h1:8PREIIYfA3UPigQNF+Hx+778/twVxJzI8bI8fcpVXEo= repo.nqws.ru/merch-tracker-v2/mt-rabbit v0.1.19/go.mod h1:8PREIIYfA3UPigQNF+Hx+778/twVxJzI8bI8fcpVXEo=

View file

@ -33,8 +33,8 @@ func NewApp(ctx context.Context, cfg config.Config) *App {
Pass: cfg.Rabbit.Pass, Pass: cfg.Rabbit.Pass,
Vhost: cfg.Rabbit.Vhost, Vhost: cfg.Rabbit.Vhost,
}, },
ChanLen: cfg.App.ProcChanLen, LoggingEnabled: cfg.Rabbit.LoggingEnabled,
LoggerEnabled: false, ChanLen: cfg.App.ProcChanLen,
}) })
return &App{ return &App{
@ -53,11 +53,13 @@ func (app *App) Run(ctx context.Context) error {
defer mainLoop.Stop() defer mainLoop.Stop()
go func() { go func() {
log.Info("Process tasks after start")
if err := app.processor.ProcessTasks(ctx); err != nil { if err := app.processor.ProcessTasks(ctx); err != nil {
errChan <- err errChan <- err
} }
for range mainLoop.C { for range mainLoop.C {
log.WithField("period", app.config.App.CheckPeriod).Info("Repeated process tasks")
if err := app.processor.ProcessTasks(ctx); err != nil { if err := app.processor.ProcessTasks(ctx); err != nil {
errChan <- err errChan <- err
} }

View file

@ -20,11 +20,11 @@ type Addr struct {
} }
type Deps struct { type Deps struct {
Ctx context.Context Ctx context.Context
TA taskAgent.TaskAgent TA taskAgent.TaskAgent
Addr Addr Addr Addr
ChanLen uint ChanLen uint
LoggerEnabled bool LoggingEnabled bool
} }
func NewHandler(deps Deps) Processor { func NewHandler(deps Deps) Processor {

View file

@ -0,0 +1,10 @@
package processor
import "strings"
func linkIsValid(link string) bool {
if strings.HasPrefix(link, "http://") || strings.HasPrefix(link, "https://") {
return true
}
return false
}

View file

@ -3,7 +3,6 @@ package processor
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"fmt"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
rabbit "repo.nqws.ru/merch-tracker-v2/mt-rabbit" rabbit "repo.nqws.ru/merch-tracker-v2/mt-rabbit"
"task-processor/internal/structs" "task-processor/internal/structs"
@ -30,7 +29,7 @@ func newService(deps Deps) *service {
s := &service{ s := &service{
taskAgent: deps.TA, taskAgent: deps.TA,
brokerAddr: ba, brokerAddr: ba,
rabbitLoggingEnabled: deps.LoggerEnabled, rabbitLoggingEnabled: deps.LoggingEnabled,
} }
s.makeTaskPublishers(deps.Ctx, ba, deps.ChanLen) s.makeTaskPublishers(deps.Ctx, ba, deps.ChanLen)
@ -62,8 +61,16 @@ func (s *service) SendResults(ctx context.Context, chanLen uint) error {
log.Debugf("%v Results sender start", pkgLogHeader) log.Debugf("%v Results sender start", pkgLogHeader)
runCtx, cancel := context.WithCancel(ctx) runCtx, cancel := context.WithCancel(ctx)
qn := "tasks-results" taskResults := rabbit.QueueOpts{
consumerClient, err := rabbit.NewClient(s.brokerAddr, qn, rabbit.WithLogging(s.rabbitLoggingEnabled)) QueueName: "tasks-results",
Durable: false,
AutoDelete: false,
Exclusive: false,
NoWait: false,
Args: nil,
}
consumerClient, err := rabbit.NewClient(s.brokerAddr, taskResults, rabbit.WithLogging(s.rabbitLoggingEnabled))
if err != nil { if err != nil {
cancel() cancel()
return err return err
@ -71,7 +78,7 @@ func (s *service) SendResults(ctx context.Context, chanLen uint) error {
resultsConsumer := rabbit.NewConsumer(consumerClient) resultsConsumer := rabbit.NewConsumer(consumerClient)
resultChan := resultsConsumer.Start(runCtx, chanLen) resultChan := resultsConsumer.Start(runCtx, chanLen)
log.Debugf("%v Results consumer started: %v", pkgLogHeader, qn) log.Debugf("%v Results consumer started: %v", pkgLogHeader, taskResults.QueueName)
go func() { go func() {
defer cancel() defer cancel()
@ -112,6 +119,10 @@ func (s *service) SendResults(ctx context.Context, chanLen uint) error {
func (s *service) sendTasks(tasks []structs.Task) error { func (s *service) sendTasks(tasks []structs.Task) error {
for _, tsk := range tasks { for _, tsk := range tasks {
for origin, link := range tsk.Origins { for origin, link := range tsk.Origins {
if !linkIsValid(link) {
continue
}
if origin == "surugaya" { if origin == "surugaya" {
pushTask(s.taskPublishers["surugaya"], tsk.MerchUuid, link) pushTask(s.taskPublishers["surugaya"], tsk.MerchUuid, link)
} }
@ -141,26 +152,62 @@ func (s *service) convertResult(b []byte) *structs.Result {
// TODO refactor this later: get origins from merch api and remove ctx pass via deps // TODO refactor this later: get origins from merch api and remove ctx pass via deps
func (s *service) makeTaskPublishers(ctx context.Context, addr rabbit.Address, chanLen uint) { func (s *service) makeTaskPublishers(ctx context.Context, addr rabbit.Address, chanLen uint) {
origins := [...]string{
"surugaya",
"mandarake",
"amiami",
}
publishers := make(map[string]chan<- []byte) publishers := make(map[string]chan<- []byte)
for _, origin := range origins { // surugaya
qn := fmt.Sprintf("task-publisher-%s", origin) surugayaOpts := rabbit.QueueOpts{
pubClient, err := rabbit.NewClient(addr, qn, rabbit.WithLogging(s.rabbitLoggingEnabled)) QueueName: "task-publisher-surugaya",
if err != nil { Durable: false,
log.WithError(err).Error("Failed to create publisher") AutoDelete: false,
continue Exclusive: false,
} NoWait: false,
Args: nil,
publishers[origin] = rabbit.NewPublisher(pubClient).Start(ctx, chanLen)
log.Debugf("%v Publisher queue created: %v", pkgLogHeader, qn)
} }
surugayaClient, err := rabbit.NewClient(addr, surugayaOpts, rabbit.WithLogging(s.rabbitLoggingEnabled))
if err != nil {
log.WithError(err).Error("Failed to create publisher")
}
publishers["surugaya"] = rabbit.NewPublisher(surugayaClient).Start(ctx, chanLen)
log.Debugf("%v Publisher queue created: %v", pkgLogHeader, surugayaOpts.QueueName)
//mandarake
mandarakeOpts := rabbit.QueueOpts{
QueueName: "task-publisher-mandarake",
Durable: false,
AutoDelete: false,
Exclusive: false,
NoWait: false,
Args: nil,
}
mandarakeClient, err := rabbit.NewClient(addr, mandarakeOpts, rabbit.WithLogging(s.rabbitLoggingEnabled))
if err != nil {
log.WithError(err).Error("Failed to create publisher")
}
publishers["mandarake"] = rabbit.NewPublisher(mandarakeClient).Start(ctx, chanLen)
log.Debugf("%v Publisher queue created: %v", pkgLogHeader, mandarakeOpts.QueueName)
//amiami
amiamiOpts := rabbit.QueueOpts{
QueueName: "task-publisher-amiami",
Durable: true,
AutoDelete: false,
Exclusive: false,
NoWait: false,
Args: nil,
}
amiamiClient, err := rabbit.NewClient(addr, amiamiOpts, rabbit.WithLogging(s.rabbitLoggingEnabled))
if err != nil {
log.WithError(err).Error("Failed to create publisher")
}
publishers["amiami"] = rabbit.NewPublisher(amiamiClient).Start(ctx, chanLen)
log.Debugf("%v Publisher queue created: %v", pkgLogHeader, amiamiOpts.QueueName)
s.taskPublishers = publishers s.taskPublishers = publishers
} }