From 2f06dce6c7f5a1b5afac6f209370ecbdd3a50a34 Mon Sep 17 00:00:00 2001 From: Alexander Zobnin Date: Tue, 3 Sep 2024 17:42:17 +0200 Subject: [PATCH] instrument zanzana client --- pkg/services/authz/zanzana/client/client.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pkg/services/authz/zanzana/client/client.go b/pkg/services/authz/zanzana/client/client.go index 6bd77f8afe8..49101a7632d 100644 --- a/pkg/services/authz/zanzana/client/client.go +++ b/pkg/services/authz/zanzana/client/client.go @@ -7,6 +7,7 @@ import ( openfgav1 "github.com/openfga/api/proto/openfga/v1" "github.com/openfga/language/pkg/go/transformer" + "go.opentelemetry.io/otel" "google.golang.org/grpc" "google.golang.org/protobuf/types/known/wrapperspb" @@ -14,6 +15,8 @@ import ( "github.com/grafana/grafana/pkg/services/authz/zanzana/schema" ) +var tracer = otel.Tracer("github.com/grafana/grafana/pkg/services/authz/zanzana/client") + type ClientOption func(c *Client) func WithTenantID(tenantID string) ClientOption { @@ -82,12 +85,18 @@ func New(ctx context.Context, cc grpc.ClientConnInterface, opts ...ClientOption) } func (c *Client) Check(ctx context.Context, in *openfgav1.CheckRequest) (*openfgav1.CheckResponse, error) { + ctx, span := tracer.Start(ctx, "authz.zanzana.client.Check") + defer span.End() + in.StoreId = c.storeID in.AuthorizationModelId = c.modelID return c.client.Check(ctx, in) } func (c *Client) ListObjects(ctx context.Context, in *openfgav1.ListObjectsRequest) (*openfgav1.ListObjectsResponse, error) { + ctx, span := tracer.Start(ctx, "authz.zanzana.client.ListObjects") + defer span.End() + in.StoreId = c.storeID in.AuthorizationModelId = c.modelID return c.client.ListObjects(ctx, in)