ntminer 6 years ago
parent
commit
53ff76f9c0

+ 1 - 1
src/AppModels/Vms/LocalMessagesViewModel.cs

@@ -119,7 +119,7 @@
         }
 
         private void Init() {
-            var data = VirtualRoot.LocalMessages.Select(a => new LocalMessageViewModel(a));
+            var data = VirtualRoot.LocalMessages.AsEnumerable().Select(a => new LocalMessageViewModel(a));
             _localMessageVms = new ObservableCollection<LocalMessageViewModel>(data);
             foreach (var dic in _count.Values) {
                 foreach (var key in dic.Keys) {

+ 2 - 1
src/NTMinerlib/LocalMessage/ILocalMessageSet.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 
 namespace NTMiner.LocalMessage {
-    public interface ILocalMessageSet : IEnumerable<ILocalMessage> {
+    public interface ILocalMessageSet {
+        IEnumerable<ILocalMessage> AsEnumerable();
     }
 }

+ 2 - 8
src/NTMinerlib/LocalMessage/LocalMessageSet.cs

@@ -1,6 +1,5 @@
 using LiteDB;
 using NTMiner.MinerClient;
-using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 
@@ -85,14 +84,9 @@ namespace NTMiner.LocalMessage {
             }
         }
 
-        public IEnumerator<ILocalMessage> GetEnumerator() {
+        public IEnumerable<ILocalMessage> AsEnumerable() {
             InitOnece();
-            return _records.GetEnumerator();
-        }
-
-        IEnumerator IEnumerable.GetEnumerator() {
-            InitOnece();
-            return _records.GetEnumerator();
+            return _records;
         }
     }
 }

+ 1 - 1
src/UnitTests/LocalMessageTests.cs

@@ -14,7 +14,7 @@ namespace UnitTests {
             for (int i = 0; i < times; i++) {
                 VirtualRoot.ThisLocalInfo(nameof(LocalMessageTests), content);
             }
-            Assert.IsTrue(VirtualRoot.LocalMessages.Count() == NTKeyword.LocalMessageSetCapacity);
+            Assert.IsTrue(VirtualRoot.LocalMessages.AsEnumerable().Count() == NTKeyword.LocalMessageSetCapacity);
         }
     }
 }