[screener] enable otel (#2809)

* enable otel * fix sig
synapsecns · Jun 28, 2024 · 02767e6 · 02767e6
1 parent 17ef0b1
commit 02767e6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/contrib/screener-api/chainalysis/chainalysisapi.go b/contrib/screener-api/chainalysis/chainalysisapi.go
@@ -4,14 +4,17 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"github.com/TwiN/gocache/v2"
-	"github.com/valyala/fastjson"
 	"net/http"
 	"slices"
 	"strings"
 	"time"
 
+	"github.com/TwiN/gocache/v2"
+	"github.com/dubonzi/otelresty"
+	"github.com/valyala/fastjson"
+
 	"github.com/go-resty/resty/v2"
+	"github.com/synapsecns/sanguine/core/metrics"
 	"github.com/synapsecns/sanguine/core/retry"
 )
 
@@ -41,7 +44,7 @@ const (
 )
 
 // NewClient creates a new Chainalysis API client.
-func NewClient(riskLevels []string, apiKey, url string) Client {
+func NewClient(metricHandler metrics.Handler, riskLevels []string, apiKey, url string) Client {
 	client := resty.New().
 		SetBaseURL(url).
 		SetHeader("Content-Type", "application/json").
@@ -52,6 +55,8 @@ func NewClient(riskLevels []string, apiKey, url string) Client {
 	// TODO: make this configurable.
 	registrationCache := gocache.NewCache().WithEvictionPolicy(gocache.LeastRecentlyUsed).WithMaxMemoryUsage(maxCacheSizeGB * bytesInGB)
 
+	otelresty.TraceClient(client, otelresty.WithTracerProvider(metricHandler.GetTracerProvider()))
+
 	return &clientImpl{
 		client:            client,
 		apiKey:            apiKey,

diff --git a/contrib/screener-api/screener/screener.go b/contrib/screener-api/screener/screener.go
@@ -67,8 +67,7 @@ func NewScreener(ctx context.Context, cfg config.Config, metricHandler metrics.H
 	docs.SwaggerInfo.Title = "Screener API"
 	docs.SwaggerInfo.Host = fmt.Sprintf("localhost:%d", cfg.Port)
 
-	screener.client = chainalysis.NewClient(
-		cfg.RiskLevels, cfg.ChainalysisKey, core.GetEnv("CHAINALYSIS_URL", cfg.ChainalysisURL))
+	screener.client = chainalysis.NewClient(metricHandler, cfg.RiskLevels, cfg.ChainalysisKey, core.GetEnv("CHAINALYSIS_URL", cfg.ChainalysisURL))
 
 	screener.blacklistCache = make(map[string]bool)
 	screener.whitelist = make(map[string]bool)

diff --git a/contrib/screener-api/screener/suite_test.go b/contrib/screener-api/screener/suite_test.go
@@ -181,7 +181,7 @@ func (s *ScreenerSuite) TestScreener() {
 	Equal(s.T(), len(statuses), 1)
 	NotNil(s.T(), err)
 
-	c := chainalysis.NewClient([]string{"Severe", "High"}, "key", "url")
+	c := chainalysis.NewClient(s.metrics, []string{"Severe", "High"}, "key", "url")
 	NotNil(s.T(), c)
 
 	ot, err := c.ScreenAddress(s.GetTestContext(), "0x123")