82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
ats_tracer "gitea.timerzz.com/kedaya_haitao/coach-spider/ats-tracer"
|
|
"gitea.timerzz.com/kedaya_haitao/common/pkg/web"
|
|
v2 "gitea.timerzz.com/kedaya_haitao/common/structs/v2"
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
type AtsTracerSvc struct {
|
|
ctl *ats_tracer.Controller
|
|
providerId v2.ProviderId
|
|
}
|
|
|
|
func NewAtsTracerSvc(ctl *ats_tracer.Controller, providerId v2.ProviderId) *AtsTracerSvc {
|
|
return &AtsTracerSvc{
|
|
ctl: ctl,
|
|
providerId: providerId,
|
|
}
|
|
}
|
|
|
|
func (s *AtsTracerSvc) Registry(r fiber.Router) {
|
|
api := r.Group(fmt.Sprintf("/api/v2/provider/%s/ats-trace", s.providerId))
|
|
api.Post(":skuID", s.AddAtsTracer)
|
|
api.Delete(":skuID", s.DelAtsTracer)
|
|
api.Post("stop/:skuID", s.StopAtsTracer)
|
|
}
|
|
|
|
func (s *AtsTracerSvc) AddAtsTracer(ctx fiber.Ctx) (err error) {
|
|
skuID := ctx.Params("skuID")
|
|
if skuID == "" {
|
|
return fmt.Errorf("skuID is empty")
|
|
}
|
|
if skuID, err = url.QueryUnescape(skuID); err != nil {
|
|
return err
|
|
}
|
|
skuID = strings.ReplaceAll(skuID, " ", "-")
|
|
skuID = strings.ReplaceAll(skuID, "+", "-")
|
|
if err = s.ctl.Add(skuID); err != nil {
|
|
return
|
|
}
|
|
return ctx.JSON(web.NewResponse("ok"))
|
|
}
|
|
|
|
func (s *AtsTracerSvc) DelAtsTracer(ctx fiber.Ctx) (err error) {
|
|
skuID := ctx.Params("skuID")
|
|
if skuID == "" {
|
|
return fmt.Errorf("skuID is empty")
|
|
}
|
|
if skuID, err = url.QueryUnescape(skuID); err != nil {
|
|
return err
|
|
}
|
|
skuID = strings.ReplaceAll(skuID, " ", "-")
|
|
skuID = strings.ReplaceAll(skuID, "+", "-")
|
|
if err = s.ctl.Delete(skuID); err != nil {
|
|
return
|
|
}
|
|
|
|
return ctx.JSON(web.NewResponse("ok"))
|
|
}
|
|
|
|
func (s *AtsTracerSvc) StopAtsTracer(ctx fiber.Ctx) (err error) {
|
|
skuID := ctx.Params("skuID")
|
|
if skuID == "" {
|
|
return fmt.Errorf("skuID is empty")
|
|
}
|
|
if skuID, err = url.QueryUnescape(skuID); err != nil {
|
|
return err
|
|
}
|
|
skuID = strings.ReplaceAll(skuID, " ", "-")
|
|
skuID = strings.ReplaceAll(skuID, "+", "-")
|
|
if err = s.ctl.Stop(skuID); err != nil {
|
|
return
|
|
}
|
|
|
|
return ctx.JSON(web.NewResponse("ok"))
|
|
}
|