From 817dd9349abf78193c6e096417f67bf4417e5fa3 Mon Sep 17 00:00:00 2001
From: Dmitry Zuikov <dzuikov@gmail.com>
Date: Thu, 26 Jan 2023 10:00:26 +0300
Subject: [PATCH] better

---
 hbs2-core/lib/HBS2/Actors/ChunkWriter.hs | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/hbs2-core/lib/HBS2/Actors/ChunkWriter.hs b/hbs2-core/lib/HBS2/Actors/ChunkWriter.hs
index fd5a97d0..6c1f1c07 100644
--- a/hbs2-core/lib/HBS2/Actors/ChunkWriter.hs
+++ b/hbs2-core/lib/HBS2/Actors/ChunkWriter.hs
@@ -73,7 +73,7 @@ data ChunkWriter h m = forall a . ( MonadIO m
   , dir         :: FilePath
   , storage     :: a
   , perBlock    :: TVar (HashMap FilePath [IO ()])
-  , perBlockSem :: Cache FilePath TSem
+  , perBlockSem :: TVar (HashMap FilePath TSem)
   }
 
 
@@ -125,7 +125,7 @@ newChunkWriterIO s tmp = do
   let d  = fromMaybe def tmp
 
   mt <- liftIO $ newTVarIO mempty
-  mts <- liftIO $ Cache.newCache Nothing
+  mts <- liftIO $ newTVarIO mempty
 
   running <- liftIO $ newTVarIO False
 
@@ -219,14 +219,24 @@ writeChunk2  w salt h o bs = do
   where
     fn = makeFileName w salt h
 
-
 flush w fn = do
   let cache = perBlock w
+  let sems = perBlockSem w
 
   liftIO $ do
+
+    nsem <- atomically $ Sem.newTSem 1
+    sem <- atomically $ stateTVar sems $ \hm -> let found = HashMap.lookup fn hm
+                                                in case found of
+                                                     Nothing -> (nsem, HashMap.insert fn nsem hm)
+                                                     Just s  -> (s, hm)
+    atomically $ Sem.waitTSem sem
+
     actions <- atomically $ stateTVar cache (\v -> (HashMap.lookup fn v, HashMap.delete fn v))
     sequence_ (fromMaybe mempty actions)
 
+    atomically $ Sem.signalTSem sem
+
 -- Blocking!
 -- we need to write last chunk before this will happen
 -- FIXME: incremental calculation,
@@ -267,11 +277,15 @@ commitBlock2 :: forall salt h m .
 
 commitBlock2 w@(ChunkWriter {storage = stor}) salt h = do
   let cache = perBlock w
+  let se   = perBlockSem w
   flush w fn
   s <- liftIO $ B.readFile fn
   void $ putBlock stor s
   delBlock w salt h
 
+  liftIO $ atomically $ TV.modifyTVar' cache $ HashMap.delete fn
+  liftIO $ atomically $ TV.modifyTVar' se $ HashMap.delete fn
+
   where
     fn = makeFileName w salt h