From 5aa60fa97d11c601e93f5b714644fd3285e0ba9e Mon Sep 17 00:00:00 2001
From: Alena Petraki <alena.petraki@gmail.com>
Date: Fri, 28 Apr 2023 13:05:29 +0300
Subject: [PATCH] refactor

---
 pkg/account/client.go | 4 ++--
 pkg/auth/grpc.go      | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkg/account/client.go b/pkg/account/client.go
index dcc46e28..53ae9fb8 100644
--- a/pkg/account/client.go
+++ b/pkg/account/client.go
@@ -20,7 +20,7 @@ const (
 	DefaultCacheTTL  = time.Second * 10
 )
 
-func NewClient(conn *grpc.ClientConn, opts ...Option) (*Account, error) {
+func NewClient(conn *grpc.ClientConn, opts ...Option) *Account {
 
 	client := &Account{}
 	c := &config{}
@@ -47,7 +47,7 @@ func NewClient(conn *grpc.ClientConn, opts ...Option) (*Account, error) {
 		client = WithLogging(client, c.logger, c.accessLog)
 	}
 
-	return client, nil
+	return client
 }
 
 func WithCaching(client *Account, size int, ttl time.Duration) *Account {
diff --git a/pkg/auth/grpc.go b/pkg/auth/grpc.go
index 92399d0c..e6cd0092 100644
--- a/pkg/auth/grpc.go
+++ b/pkg/auth/grpc.go
@@ -105,6 +105,11 @@ func AddAuthorizationInterceptor(auth string) grpc.UnaryClientInterceptor {
 	}
 }
 
+// AddAPIKeyInterceptor возвращает опции для создания grpc-соединения с передачей API-ключа при каждом запросе
+func AddAPIKeyInterceptor(key string) grpc.UnaryClientInterceptor {
+	return AddAuthorizationInterceptor("API-Key " + key)
+}
+
 // WithOAuth2Credentials возвращает опции для создания grpc-соединения с аутентификацией oauth2. Переданный контекст
 // будет использован для каждого запроса токена, поэтому он не может быть с таймаутом.
 func WithOAuth2Credentials(ctx context.Context, tokenURL, clientID, clientSecret, audience string) grpc.DialOption {
-- 
GitLab