Code update

masesdevelopers · Jul 2, 2024 · 31689d6 · 31689d6
1 parent eab09d4
commit 31689d6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/src/net/KEFCore/Storage/Internal/KNetStreamsRetriever.cs b/src/net/KEFCore/Storage/Internal/KNetStreamsRetriever.cs
@@ -291,7 +291,7 @@ public IEnumerator<ValueBuffer> GetEnumerator()
 #if DEBUG_PERFORMANCE
             Infrastructure.KafkaDbContext.ReportString($"Requesting KafkaEnumerator for {_entityType.Name} on {DateTime.Now:HH:mm:ss.FFFFFFF}");
 #endif
-            return new KafkaEnumerator(_kafkaCluster, _entityType, _keyValueStore?.All, _useEnumeratorWithPrefetch, false);
+            return new KafkaEnumerator(_kafkaCluster, _entityType, _keyValueStore?.All(), _useEnumeratorWithPrefetch, false);
         }
 
         System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
@@ -306,7 +306,7 @@ public IAsyncEnumerator<ValueBuffer> GetAsyncEnumerator(CancellationToken cancel
 #if DEBUG_PERFORMANCE
             Infrastructure.KafkaDbContext.ReportString($"Requesting async KafkaEnumerator for {_entityType.Name} on {DateTime.Now:HH:mm:ss.FFFFFFF}");
 #endif
-            return new KafkaEnumerator(_kafkaCluster, _entityType, _keyValueStore?.All, _useEnumeratorWithPrefetch, true);
+            return new KafkaEnumerator(_kafkaCluster, _entityType, _keyValueStore?.All(), _useEnumeratorWithPrefetch, true);
         }
 #endif
     }
@@ -382,7 +382,7 @@ public ValueTask DisposeAsync()
 #if DEBUG_PERFORMANCE
             Infrastructure.KafkaDbContext.ReportString($"KafkaEnumerator _moveNextSw: {_moveNextSw.Elapsed} _currentSw: {_currentSw.Elapsed} _valueGetSw: {_valueGetSw.Elapsed} _valueGet2Sw: {_valueGet2Sw.Elapsed} _valueBufferSw: {_valueBufferSw.Elapsed}");
 #endif
-            return _asyncEnumerator.DisposeAsync();
+            return _asyncEnumerator != null ? _asyncEnumerator.DisposeAsync() : new ValueTask();
         }
 #endif
 
@@ -396,18 +396,18 @@ public bool MoveNext()
                 {
                     _moveNextSw.Start();
 #endif
-            if (_useEnumeratorWithPrefetch ? _enumerator != null && _enumerator.MoveNext() : _keyValueIterator != null && _keyValueIterator.HasNext)
+            if (_useEnumeratorWithPrefetch ? _enumerator != null && _enumerator.MoveNext() : _keyValueIterator != null && _keyValueIterator.HasNext())
             {
 #if DEBUG_PERFORMANCE
                 _cycles++;
                 _valueGetSw.Start();
 #endif
-                KeyValue<TKey, TValue, TJVMKey, TJVMValue> kv = _useEnumeratorWithPrefetch ? _enumerator.Current : _keyValueIterator.Next;
+                KeyValue<TKey, TValue, TJVMKey, TJVMValue>? kv = _useEnumeratorWithPrefetch ? _enumerator?.Current : _keyValueIterator?.Next();
 #if DEBUG_PERFORMANCE
                 _valueGetSw.Stop();
                 _valueGet2Sw.Start();
 #endif
-                TValue value = kv.Value;
+                TValue value = kv != null ? kv.Value : default;
 #if DEBUG_PERFORMANCE
                 _valueGet2Sw.Stop();
                 _valueBufferSw.Start();

diff --git a/src/net/KEFCore/Storage/Internal/KafkaStreamsBaseRetriever.cs b/src/net/KEFCore/Storage/Internal/KafkaStreamsBaseRetriever.cs
@@ -360,7 +360,8 @@ public bool MoveNext()
                 V? data;
                 using (KeyValue<K, V> kv = _keyValueIterator.Next())
                 {
-                    data = kv.value != null ? (V)(object)kv.value! : default;
+                    var kvSupport = new MASES.KNet.Streams.KeyValueSupport<K, V>(kv.BridgeInstance);
+                    data = kvSupport.Value != null ? (V)(object)kvSupport.Value! : default;
                 }
 #if DEBUG_PERFORMANCE
                 _valueGetSw.Stop();
@@ -372,7 +373,7 @@ public bool MoveNext()
                 _valueBufferSw.Start();
 #endif
                 object[] array = null!;
-                entityTypeData.GetData(_entityType, ref array);
+                entityTypeData?.GetData(_entityType, ref array);
 #if DEBUG_PERFORMANCE
                 _valueBufferSw.Stop();
 #endif