From 0998b278cf934874d11b2b0502f77809c4dc69f2 Mon Sep 17 00:00:00 2001
From: unknown <ogoun.er@yandex.ru>
Date: Fri, 10 Dec 2021 05:51:43 +0300
Subject: [PATCH] Update HNSWMap.cs

Make map serializable
---
 ZeroLevel.HNSW/HNSWMap.cs | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/ZeroLevel.HNSW/HNSWMap.cs b/ZeroLevel.HNSW/HNSWMap.cs
index 129151c..f6986d8 100644
--- a/ZeroLevel.HNSW/HNSWMap.cs
+++ b/ZeroLevel.HNSW/HNSWMap.cs
@@ -1,5 +1,6 @@
 using System.Collections.Concurrent;
 using System.Collections.Generic;
+using ZeroLevel.Services.Serialization;
 
 namespace ZeroLevel.HNSW
 {
@@ -7,9 +8,10 @@ namespace ZeroLevel.HNSW
     // HNSW vectorId + vector
     // Map object feature - vectorId
     public class HNSWMap<TFeature>
+        : IBinarySerializable
     {
-        private readonly ConcurrentDictionary<TFeature, int> _map = new ConcurrentDictionary<TFeature, int>();
-        private readonly ConcurrentDictionary<int, TFeature> _reverse_map = new ConcurrentDictionary<int, TFeature>();
+        private ConcurrentDictionary<TFeature, int> _map = new ConcurrentDictionary<TFeature, int>();
+        private ConcurrentDictionary<int, TFeature> _reverse_map = new ConcurrentDictionary<int, TFeature>();
 
         public void Append(TFeature feature, int vectorId)
         {
@@ -40,5 +42,17 @@ namespace ZeroLevel.HNSW
                 }
             }
         }
+
+        public void Deserialize(IBinaryReader reader)
+        {
+            this._map = reader.ReadDictionaryAsConcurrent<TFeature, int>();
+            this._reverse_map = reader.ReadDictionaryAsConcurrent<int, TFeature>();
+        }
+
+        public void Serialize(IBinaryWriter writer)
+        {
+            writer.WriteDictionary<TFeature, int>(this._map);
+            writer.WriteDictionary<int, TFeature>(this._reverse_map);
+        }
     }
 }