Please see the README on Github at https://github.com/githubuser/timemap#readme.
Please see the README on Github at https://github.com/githubuser/timemap#readme
timemap
A mutable Data.HashMap
-like structure, where each entity is implicitly indexed by their last-modified time. Useful for keyed caches.
Usage
import qualified Data.TimeMap as TM
import Control.Concurrent.STM
main :: IO ()
main = do
-- create a new, empty reference
mapRef <- atomically TM.newTimeMap
-- insert a new key/value pair in the map. Note that
-- `someKey` should implement `Hashable`. This also
-- sets the creation time of the value to "now".
TM.insert someKey 0 mapRef
-- adjusts the value of `someKey` to `1`. Note that
-- also resets the creation time of the value to "now".
TM.adjust (+1) someKey mapRef
-- wait a second
threadDelay 1000000
-- delete all values older than 1 second from now.
atomically $ TM.filterFromNow 1 mapRef
-- Will return `Nothing`
atomically $ TM.lookup someKey mapRef
return ()
How it works
There are two internal maps for a TimeMap k a
:
- a "time-indexed" map reference:
TVar (Map UTCTime (HashSet k))
- a mutable map of values:
STMContainers.Map.Map k (UTCTime, a)
Insertion
Inserting a new value first performs a lookup on the hashtable to see if the key already exists:
- If it doesn't, insert the current time and the element that into the mutable map. Then, insert the key as the value, and the time as the key into the time-indexed multimap.
- If it does, remove the entry from the time-indexed multimap for the old time, before doing the same thing as the first bullet.
Lookups
This doesn't have to create new data, and thus doesn't need full IO
. All it does is lookup the key in the mutable map.
Filtering
To filter out all entries older than some time, you have to use the Data.Map.splitLookup
function to split the time-indexed multimap into the entries you want to delete from the main container, and the ones you want to keep. You simply writeTVar
the map you want to keep, but for the ones you want to delete, you need to get all the elems
of that map. Then, for every key that we need to delete, we delete it from the main container. Suprisingly, this appears to operate in constant time, regardless of the size of the map.
How to run tests
stack test
Benchmarks
stack bench --benchmark-arguments="--output profile.html"
You can also view the results on my box here.