From f575218da9dd95f5aa91bf407469841812ba63e1 Mon Sep 17 00:00:00 2001
From: leventov <leventov@ya.ru>
Date: Mon, 23 Nov 2015 17:42:36 +0300
Subject: [PATCH] Extend persistence file on mapping

---
 .../hash/impl/VanillaChronicleHash.java         | 17 ++++++++++++++++-
 .../openhft/chronicle/map/DefaultValueTest.java |  3 ---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/main/java/net/openhft/chronicle/hash/impl/VanillaChronicleHash.java b/src/main/java/net/openhft/chronicle/hash/impl/VanillaChronicleHash.java
index f35654a99..2f5e2f02a 100644
--- a/src/main/java/net/openhft/chronicle/hash/impl/VanillaChronicleHash.java
+++ b/src/main/java/net/openhft/chronicle/hash/impl/VanillaChronicleHash.java
@@ -19,6 +19,7 @@ package net.openhft.chronicle.hash.impl;
 import net.openhft.chronicle.algo.bytes.Access;
 import net.openhft.chronicle.algo.locks.*;
 import net.openhft.chronicle.bytes.BytesStore;
+import net.openhft.chronicle.bytes.MappedBytesStoreFactory;
 import net.openhft.chronicle.bytes.NativeBytesStore;
 import net.openhft.chronicle.core.Maths;
 import net.openhft.chronicle.core.OS;
@@ -36,6 +37,7 @@ import org.slf4j.LoggerFactory;
 import java.io.*;
 import java.nio.ByteBuffer;
 import java.nio.channels.FileChannel;
+import java.nio.channels.FileLock;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
@@ -697,9 +699,22 @@ public abstract class VanillaChronicleHash<K,
         }
     }
 
+    /**
+     * @see net.openhft.chronicle.bytes.MappedFile#acquireByteStore(long, MappedBytesStoreFactory)
+     */
     private static NativeBytesStore map(
             RandomAccessFile raf, long mapSize, long mappingOffsetInFile) throws IOException {
-        long address = OS.map(raf.getChannel(), READ_WRITE, mappingOffsetInFile, mapSize);
+        long minFileSize = mappingOffsetInFile + mapSize;
+        FileChannel fileChannel = raf.getChannel();
+        if (fileChannel.size() < minFileSize) {
+            // handle a possible race condition between processes.
+            try (FileLock lock = fileChannel.lock()) {
+                if (fileChannel.size() < minFileSize) {
+                    raf.setLength(minFileSize);
+                }
+            }
+        }
+        long address = OS.map(fileChannel, READ_WRITE, mappingOffsetInFile, mapSize);
         OS.Unmapper unmapper = new OS.Unmapper(address, mapSize, DUMMY_REFERENCE_COUNTED);
         return new NativeBytesStore(address, mapSize, unmapper, false);
     }
diff --git a/src/test/java/net/openhft/chronicle/map/DefaultValueTest.java b/src/test/java/net/openhft/chronicle/map/DefaultValueTest.java
index a975d8364..74ce6db6f 100644
--- a/src/test/java/net/openhft/chronicle/map/DefaultValueTest.java
+++ b/src/test/java/net/openhft/chronicle/map/DefaultValueTest.java
@@ -16,12 +16,9 @@
 
 package net.openhft.chronicle.map;
 
-import net.openhft.chronicle.hash.*;
-import net.openhft.chronicle.hash.Data;
 import net.openhft.chronicle.hash.serialization.ListMarshaller;
 import net.openhft.chronicle.hash.serialization.impl.IntegerMarshaller;
 import net.openhft.chronicle.set.Builder;
-import org.jetbrains.annotations.NotNull;
 import org.junit.Test;
 
 import java.io.File;