diff --git a/cmd/test/main.go b/cmd/test/main.go index 2340057..5015160 100644 --- a/cmd/test/main.go +++ b/cmd/test/main.go @@ -1,38 +1,26 @@ package main import ( - "fmt" - "sipro-mps/internal/ym/products" + "context" + "sync" + "time" ) -// package main -// -// import ( -// -// "context" -// "fmt" -// "sipro-mps/internal/wb/products/mapping/generated" -// "strings" -// -// pb "sipro-mps/api/generated/v1/wb/products" -// "sipro-mps/pkg/api/wb/client" -// -// "github.com/deliveryhero/pipeline/v2" -// -// ) -// -// func main() { -// -// input := "1,2,3,4,5" -// -// for out := range pipeline.Process(context.Background(), apply, pipeline.Emit(input)) { -// for j := range out { -// fmt.Printf("process: %s\n", out[j]) -// } -// } -// } - func main() { - rs := products.GetCategoriesError("{\"status\":\"ERROR\",\"errors\":[{\"code\":\"BAD_REQUEST\",\"message\":\"Following categories are not leaf categories: 91735\"}]}") - fmt.Println(rs) + ctx, cacnel := context.WithTimeout(context.Background(), time.Second*5) + defer cacnel() + wg := sync.WaitGroup{} + go func() { + defer func() { + wg.Done() + println("goroutine exit") + }() + select { + case <-ctx.Done(): + println("context done") + return + } + }() + wg.Add(1) + wg.Wait() } diff --git a/internal/wb/rate_limiter.go b/internal/wb/rate_limiter.go index 51f41c6..f9366d1 100644 --- a/internal/wb/rate_limiter.go +++ b/internal/wb/rate_limiter.go @@ -5,6 +5,7 @@ import ( "encoding/json" "fmt" "net/http" + "os" "sipro-mps/internal/redis" "sipro-mps/pkg/utils" "time" @@ -69,6 +70,9 @@ type RateLimitTransport struct { func (t *RateLimitTransport) RoundTrip(req *http.Request) (*http.Response, error) { ctx := req.Context() + req.Header.Set("X-Client-Secret", os.Getenv("WB_SECRET_TOKEN")) + req.Header.Set("User-Agent", "wbas_seller.denco.store3547") + tokenString := req.Header.Get("Authorization") authData := utils.NewWbAuthData(tokenString) authDataBytes, err := json.Marshal(authData)