fixed

polis-vk · Dec 24, 2023 · 4aee498 · 4aee498
1 parent 32e11d7
commit 4aee498
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java b/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java
@@ -88,7 +88,6 @@ public void upsert(Entry<MemorySegment> entry) {
             long valueSize = entry.value() == null ? 0 : entry.value().byteSize();
             if (size.addAndGet(keySize + valueSize) >= maxSize) {
                 flush();
-                size.set(0);
             }
         } catch (IOException e) {
             throw new UncheckedIOException(e);
@@ -177,6 +176,7 @@ public void flush() throws IOException {
             } finally {
                 upsertLock.writeLock().unlock();
             }
+            size.set(0);
         });
     }