From 1b9b37f27d551885a055e74695a5e17cf896cf68 Mon Sep 17 00:00:00 2001 From: Robert Lin Date: Thu, 18 Jul 2024 14:05:42 -0700 Subject: [PATCH] trim prefix --- cmd/enterprise-portal/internal/codyaccessservice/v1_store.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cmd/enterprise-portal/internal/codyaccessservice/v1_store.go b/cmd/enterprise-portal/internal/codyaccessservice/v1_store.go index ba9bc998721..c5440b29639 100644 --- a/cmd/enterprise-portal/internal/codyaccessservice/v1_store.go +++ b/cmd/enterprise-portal/internal/codyaccessservice/v1_store.go @@ -81,6 +81,8 @@ func (s *storeV1) GetCodyGatewayUsage(ctx context.Context, subscriptionID string return nil, errStoreUnimplemented } + subscriptionID = strings.TrimPrefix(subscriptionID, subscriptionsv1.EnterpriseSubscriptionIDPrefix) + // Collect results concurrently, cancelling on the first error result. p := pool.NewWithResults[func(*codyaccessv1.CodyGatewayUsage)](). WithContext(ctx). @@ -153,7 +155,7 @@ func (s *storeV1) GetCodyGatewayUsage(ctx context.Context, subscriptionID string func (s *storeV1) GetCodyGatewayAccessBySubscription(ctx context.Context, subscriptionID string) (*codyaccess.CodyGatewayAccessWithSubscriptionDetails, error) { return s.CodyAccess.CodyGateway().Get(ctx, codyaccess.GetCodyGatewayAccessOptions{ - SubscriptionID: subscriptionID, + SubscriptionID: strings.TrimPrefix(subscriptionID, subscriptionsv1.EnterpriseSubscriptionIDPrefix), }) }