chore: set timeout at client instantiation

momentohq · Dec 21, 2023 · 1f9cb8e · 1f9cb8e
1 parent 84abbd6
commit 1f9cb8e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 14 deletions.
diff --git a/lib/src/cache_client.dart b/lib/src/cache_client.dart
@@ -11,7 +11,7 @@ import 'messages/values.dart';
 abstract class ICacheClient {
   Future<GetResponse> get(String cacheName, Value value);
 
-  Future<SetResponse> subscribe(String cacheName, Value key, Value value,
+  Future<SetResponse> set(String cacheName, Value key, Value value,
       {int? ttlSeconds});
 }
 
@@ -33,7 +33,7 @@ class CacheClient implements ICacheClient {
   }
 
   @override
-  Future<SetResponse> subscribe(String cacheName, Value key, Value value,
+  Future<SetResponse> set(String cacheName, Value key, Value value,
       {int? ttlSeconds}) {
     return _dataClient.set(cacheName, key, value, ttlSeconds: ttlSeconds);
   }

diff --git a/lib/src/internal/data_client.dart b/lib/src/internal/data_client.dart
@@ -28,8 +28,8 @@ class DataClient implements AbstractDataClient {
     _client = ScsClient(_channel,
         options: CallOptions(metadata: {
           'authorization': credentialProvider.apiKey,
-          'agent': 'dart:0.1.0'
-        }));
+          'agent': 'dart:0.1.0',
+        }, timeout: _configuration.transportStrategy.grpcConfig.deadline));
     _configuration = configuration;
     _defaultTtl = defaultTtl;
   }
@@ -40,11 +40,9 @@ class DataClient implements AbstractDataClient {
     request.cacheKey = key.toBinary();
     try {
       var resp = await _client.get(request,
-          options: CallOptions(
-              timeout: _configuration.transportStrategy.grpcConfig.deadline,
-              metadata: {
-                'cacheName': cacheName,
-              }));
+          options: CallOptions(metadata: {
+            'cacheName': cacheName,
+          }));
 
       switch (resp.result) {
         case ECacheResult.Miss:
@@ -75,11 +73,9 @@ class DataClient implements AbstractDataClient {
         : _defaultTtl.inMilliseconds) as Int64;
     try {
       await _client.set(request,
-          options: CallOptions(
-              timeout: _configuration.transportStrategy.grpcConfig.deadline,
-              metadata: {
-                'cacheName': cacheName,
-              }));
+          options: CallOptions(metadata: {
+            'cacheName': cacheName,
+          }));
       return SetSuccess();
     } catch (e) {
       if (e is SdkException) {