Add benchmarks

.github/workflows/hs-bench.yml

@@ -0,0 +1,58 @@
+name: Haskell benchmarks
+
+on:
+  workflow_dispatch:
+
+jobs:
+  get-inputs:
+    uses: ephemient/aoc2023/.github/workflows/get-inputs.yml@main
+    secrets:
+      SESSION: ${{ secrets.SESSION }}
+
+  build:
+    needs: [ get-inputs ]
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/checkout@v4
+        with:
+          ref: gh-docs
+          path: gh-docs
+      - uses: actions/download-artifact@v3
+        with:
+          name: inputs
+          path: inputs
+      - uses: haskell-actions/setup@v2
+        id: setup
+        with:
+          ghc-version: 9.4.7
+      - run: |
+          cabal configure --enable-tests --enable-benchmarks
+          cabal build all --dry-run
+        working-directory: hs
+      - uses: actions/cache/restore@v3
+        id: cache
+        env:
+          key: ${{ runner.os }}-ghc-${{ steps.setup.outputs.ghc-version }}-cabal-${{ steps.setup.outputs.cabal-version }}
+        with:
+          path: ${{ steps.setup.outputs.cabal-store }}
+          key: ${{ env.key }}-plan-${{ hashFiles('hs/dist-newstyle/cache/plan.json') }}
+          restore-keys: ${{ env.key }}-
+      - run: cabal build all --only-dependencies
+        if: steps.cache.outputs.cache-hit != 'true'
+        working-directory: hs
+      - uses: actions/cache/save@v3
+        if: steps.cache.outputs.cache-hit != 'true'
+        with:
+          path: ${{ steps.setup.outputs.cabal-store }}
+          key: ${{ steps.cache.outputs.cache-primary-key }}
+      - run: cabal bench bench:aoc2023-bench --benchmark-options='-o ${{ github.workspace }}/gh-docs/aoc2023-bench.html'
+        env:
+          AOC2023_DATADIR: ${{ github.workspace }}/inputs
+        working-directory: hs
+      - uses: EndBug/add-and-commit@v9
+        with:
+          cwd: gh-docs
+          add: aoc2023-bench.html
+          message: 'Haskell Criterion ${{ github.sha }}'

.github/workflows/hs.yml

@@ -26,7 +26,7 @@ jobs:
         with:
           ghc-version: 9.4.7
       - run: |
-          cabal configure --enable-tests
+          cabal configure --enable-tests --enable-benchmarks
           cabal build all --dry-run
         working-directory: hs
       - uses: actions/cache/restore@v3

hs/aoc2023.cabal

@@ -69,3 +69,18 @@ test-suite aoc2023-test
     build-tool-depends:
         hspec-discover:hspec-discover ^>=2.11.7
     ghc-options:      -threaded -rtsopts "-with-rtsopts=-N"
+
+benchmark aoc2023-bench
+    default-language: GHC2021
+    type:             exitcode-stdio-1.0
+
+    -- Directories containing source files.
+    hs-source-dirs:   bench
+    main-is:          Main.hs
+    build-depends:
+        aoc2023,
+        base ^>=4.17.2.0,
+        criterion ^>= 1.6.3.0,
+        filepath ^>=1.4.2.2,
+        text ^>=2.0.1
+    ghc-options:      -threaded -rtsopts "-with-rtsopts=-N"

hs/bench/Main.hs

@@ -0,0 +1,23 @@
+module Main (main) where
+
+import Criterion.Main (bench, bgroup, defaultMain, env, nf)
+import Data.Foldable (find)
+import Data.Maybe (fromMaybe)
+import Data.Text (Text)
+import qualified Data.Text.IO as TIO (readFile)
+import qualified Day1 (part1, part2)
+import System.Environment (lookupEnv)
+import System.FilePath (combine)
+
+getDayInput :: Int -> IO Text
+getDayInput i = do
+    dataDir <- fromMaybe "." . find (not . null) <$> lookupEnv "AOC2023_DATADIR"
+    TIO.readFile . combine dataDir $ "day" ++ show i ++ ".txt"
+
+main :: IO ()
+main = defaultMain
+  [ env (getDayInput 1) $ \input -> bgroup "Day 1"
+      [ bench "part 1" $ nf Day1.part1 input
+      , bench "part 2" $ nf Day1.part2 input
+      ]
+  ]