feat: add retry middleware & configure retries in waitForOperation ut…

…ility function

package.json

@@ -31,6 +31,7 @@
     "luxon": "^2.2.0",
     "nice-grpc": "^1.0.6",
     "nice-grpc-client-middleware-deadline": "^1.0.6",
+    "nice-grpc-client-middleware-retry": "^1.1.2",
     "protobufjs": "^6.11.3",
     "utility-types": "^3.10.0"
   },

src/types.ts

@@ -1,21 +1,26 @@
 import {
-    ChannelCredentials, ChannelOptions, Client, ServiceDefinition,
+    ChannelCredentials,
+    ChannelOptions,
+    Client,
+    ServiceDefinition,
 } from '@grpc/grpc-js';
 import { RawClient } from 'nice-grpc';
-import { NormalizedServiceDefinition } from 'nice-grpc/lib/service-definitions';
 import { DeadlineOptions } from 'nice-grpc-client-middleware-deadline';
+import { RetryOptions } from 'nice-grpc-client-middleware-retry';
+import { NormalizedServiceDefinition } from 'nice-grpc/lib/service-definitions';
 
 export interface TokenService {
     getToken: () => Promise<string>;
 }
 
 export interface GeneratedServiceClientCtor<T extends ServiceDefinition> {
+    service: T
+
     new(
         address: string,
         credentials: ChannelCredentials,
         options?: Partial<ChannelOptions>,
     ): Client;
-    service: T
 }
 
 export interface IIAmCredentials {
@@ -59,4 +64,5 @@ export type SessionConfig =
     | ServiceAccountCredentialsConfig
     | GenericCredentialsConfig;
 
-export type WrappedServiceClientType<S extends ServiceDefinition> = RawClient<NormalizedServiceDefinition<S>, DeadlineOptions>;
+// eslint-disable-next-line max-len
+export type WrappedServiceClientType<S extends ServiceDefinition> = RawClient<NormalizedServiceDefinition<S>, DeadlineOptions & RetryOptions>;

src/utils/client-factory.ts

@@ -1,4 +1,7 @@
 import { createClientFactory } from 'nice-grpc';
 import { deadlineMiddleware } from 'nice-grpc-client-middleware-deadline';
+import { retryMiddleware } from 'nice-grpc-client-middleware-retry';
 
-export const clientFactory = createClientFactory().use(deadlineMiddleware);
+export const clientFactory = createClientFactory()
+    .use(retryMiddleware)
+    .use(deadlineMiddleware);

src/utils/operation/wait-for.ts

@@ -1,6 +1,9 @@
-import { Session } from '../../session';
+import {
+    cloudApi,
+    serviceClients,
+} from '../..';
 import { Operation } from '../../generated/yandex/cloud/operation/operation';
-import { serviceClients, cloudApi } from '../..';
+import { Session } from '../../session';
 
 const { operation: { operation_service: { GetOperationRequest } } } = cloudApi;
 
@@ -18,6 +21,10 @@ export const waitForOperation = (op: Operation, session: Session, timeoutMs: num
                 GetOperationRequest.fromPartial({
                     operationId: op.id,
                 }),
+                {
+                    retry: true,
+                    retryMaxAttempts: 3,
+                },
             );
 
             checksCount++;