mirror of
https://github.com/anoma/juvix.git
synced 2024-12-14 17:32:00 +03:00
2f4a3f809b
## Overview
This PR makes the compiler pipeline thread-safe so that the test suite
can be run in parallel.
This is achieved by:
* Removing use of `{get, set, with}CurrentDir` functions.
* Adding locking around shared file resources like the the
global-project and internal build directory.
NB: **Locking is disabled for the main compiler target**, as it is
single threaded they are not required.
## Run test suite in parallel
To run the test suite in parallel you must add `--ta '+RTS -N -RTS'` to
your stack test arguments. For example:
```
stack test --fast --ta '+RTS -N -RTS'
```
The `-N` instructs the Haskell runtime to choose the number of threads
to use based on how many processors there are on your machine. You can
use `-Nn` to see the number of threads to `n`.
These flags are already [set in the
Makefile](e6dca22cfd/Makefile (L26))
when you or CI uses `stack test`.
## Locking
The Haskell package
[filelock](https://hackage.haskell.org/package/filelock) is used for
locking. File locks are used instead of MVars because Juvix code does
not control when new threads are created, they are created by the test
suite. This means that MVars created by Juvix code will have no effect,
because they are created independently on each test-suite thread.
Additionally the resources we're locking live on the filesystem and so
can be conveniently tagged by path.
### FileLock
The filelock library is wrapped in a FileLock effect:
e6dca22cfd/src/Juvix/Data/Effect/FileLock/Base.hs (L6-L8)
There is an [IO
interpreter](e6dca22cfd/src/Juvix/Data/Effect/FileLock/IO.hs (L8))
that uses filelock and an [no-op
interpreter](e6dca22cfd/src/Juvix/Data/Effect/FileLock/Permissive.hs (L7))
that just runs actions unconditionally.
### TaggedLock
To make the file locks simpler to use a TaggedLock effect is introduced:
e6dca22cfd/src/Juvix/Data/Effect/TaggedLock/Base.hs (L5-L11)
And convenience function:
e6dca22cfd/src/Juvix/Data/Effect/TaggedLock.hs (L28)
This allows an action to be locked, tagged by a directory that may or
may not exist. For example in the following code, an action is performed
on a directory `root` that may delete the directory before repopulating
the files. So the lockfile cannot be stored in the `root` itself.
e6dca22cfd/src/Juvix/Extra/Files.hs (L55-L60)
## Pipeline
As noted above, we only use locking in the test suite. The main app
target pipeline is single threaded and so locking is unnecessary. So the
interpretation of locks is parameterised so that locking can be disabled
e6dca22cfd/src/Juvix/Compiler/Pipeline/Run.hs (L64)
50 lines
1.8 KiB
Haskell
50 lines
1.8 KiB
Haskell
module Compilation.Base where
|
|
|
|
import Base
|
|
import Core.Compile.Base
|
|
import Core.Eval.Base
|
|
import Juvix.Compiler.Core qualified as Core
|
|
import Juvix.Data.Effect.TaggedLock
|
|
import Juvix.Data.PPOutput
|
|
|
|
data CompileAssertionMode
|
|
= EvalOnly
|
|
| -- | Specify text to be sent to stdin of the process under test
|
|
CompileOnly Text
|
|
| EvalAndCompile
|
|
|
|
compileAssertion ::
|
|
Path Abs Dir ->
|
|
Int ->
|
|
CompileAssertionMode ->
|
|
Path Abs File ->
|
|
Path Abs File ->
|
|
(String -> IO ()) ->
|
|
Assertion
|
|
compileAssertion root' optLevel mode mainFile expectedFile step = do
|
|
step "Translate to JuvixCore"
|
|
entryPoint <- defaultEntryPointIO' LockModeExclusive root' mainFile
|
|
tab <- (^. Core.coreResultTable) . snd <$> runIOExclusive entryPoint upToCore
|
|
case run $ runReader Core.defaultCoreOptions $ runError $ Core.toEval' tab of
|
|
Left err -> assertFailure (show (pretty (fromJuvixError @GenericError err)))
|
|
Right tab' -> do
|
|
let evalAssertion = coreEvalAssertion' EvalModePlain tab' mainFile expectedFile step
|
|
compileAssertion' stdinText = coreCompileAssertion' optLevel tab' mainFile expectedFile stdinText step
|
|
case mode of
|
|
EvalOnly -> evalAssertion
|
|
CompileOnly stdinText -> compileAssertion' stdinText
|
|
EvalAndCompile -> evalAssertion >> compileAssertion' ""
|
|
|
|
compileErrorAssertion ::
|
|
Path Abs Dir ->
|
|
Path Abs File ->
|
|
(String -> IO ()) ->
|
|
Assertion
|
|
compileErrorAssertion root' mainFile step = do
|
|
step "Translate to JuvixCore"
|
|
entryPoint <- defaultEntryPointIO' LockModeExclusive root' mainFile
|
|
tab <- (^. Core.coreResultTable) . snd <$> runIOExclusive entryPoint upToCore
|
|
case run $ runReader Core.defaultCoreOptions $ runError @JuvixError $ Core.toStripped' tab of
|
|
Left _ -> assertBool "" True
|
|
Right _ -> assertFailure "no error"
|