mirror of
https://github.com/haskell-effectful/effectful.git
synced 2024-12-18 19:41:30 +03:00
31 lines
918 B
Haskell
31 lines
918 B
Haskell
|
{-# LANGUAGE CPP #-}
|
||
|
|
{-# LANGUAGE MagicHash #-}
|
||
|
|
{-# LANGUAGE UnliftedFFITypes #-}
|
||
|
|
module Effectful.Internal.Utils
|
||
|
|
( weakThreadId
|
||
|
|
, eqThreadId
|
||
|
|
) where
|
||
|
|
|
||
|
|
import Foreign.C.Types
|
||
|
|
import GHC.Conc.Sync (ThreadId(..))
|
||
|
|
import GHC.Exts (ThreadId#)
|
||
|
|
|
||
|
|
-- | Get an id of a thread that doesn't prevent its garbage collection.
|
||
|
|
weakThreadId :: ThreadId -> Int
|
||
|
|
weakThreadId (ThreadId t#) = fromIntegral $ rts_getThreadId t#
|
||
|
|
|
||
|
|
foreign import ccall unsafe "rts_getThreadId"
|
||
|
|
#if __GLASGOW_HASKELL__ >= 900
|
||
|
|
rts_getThreadId :: ThreadId# -> CLong
|
||
|
|
#else
|
||
|
|
rts_getThreadId :: ThreadId# -> CInt
|
||
|
|
#endif
|
||
|
|
|
||
|
|
-- | 'Eq' instance for 'ThreadId' is broken in GHC < 9, see
|
||
|
|
-- https://gitlab.haskell.org/ghc/ghc/-/issues/16761 for more info.
|
||
|
|
eqThreadId :: ThreadId -> ThreadId -> Bool
|
||
|
|
eqThreadId (ThreadId t1#) (ThreadId t2#) = eq_thread t1# t2# == 1
|
||
|
|
|
||
|
|
foreign import ccall unsafe "effectful_eq_thread"
|
||
|
|
eq_thread :: ThreadId# -> ThreadId# -> CLong
|