25ba8156bc
* benchmark: Changed --databases to take a parameter specifiying the size of the database to benchmark. * benchmark --databases: Display size of the populated database. * benchmark --databases: Improve the "addAssociatedFile to (new)" benchmark to really add new values, not overwriting old values.
58 lines
1.4 KiB
Haskell
58 lines
1.4 KiB
Haskell
{- git-annex benchmark
|
|
-
|
|
- Copyright 2016-2019 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
{-# LANGUAGE CPP #-}
|
|
|
|
module Command.Benchmark where
|
|
|
|
import Command
|
|
import Types.Benchmark
|
|
#ifdef WITH_BENCHMARK
|
|
import Database.Benchmark
|
|
|
|
import Criterion.Main
|
|
import Criterion.Main.Options (parseWith)
|
|
#endif
|
|
|
|
cmd :: BenchmarkGenerator -> Command
|
|
cmd generator = command "benchmark" SectionTesting
|
|
"benchmark git-annex commands"
|
|
paramNothing
|
|
(seek generator <$$> optParser)
|
|
|
|
data BenchmarkOptions
|
|
= BenchmarkOptions CmdParams CriterionMode
|
|
| BenchmarkDatabases CriterionMode Integer
|
|
|
|
optParser :: CmdParamsDesc -> Parser BenchmarkOptions
|
|
optParser desc = benchmarkoptions <|> benchmarkdatabases
|
|
where
|
|
benchmarkoptions = BenchmarkOptions
|
|
<$> cmdParams desc
|
|
<*> criterionopts
|
|
benchmarkdatabases = BenchmarkDatabases
|
|
<$> criterionopts
|
|
<*> option auto
|
|
( long "databases"
|
|
<> metavar paramNumber
|
|
<> help "benchmark sqlite databases"
|
|
)
|
|
#ifdef WITH_BENCHMARK
|
|
criterionopts = parseWith defaultConfig
|
|
#else
|
|
criterionopts = pure ()
|
|
#endif
|
|
|
|
seek :: BenchmarkGenerator -> BenchmarkOptions -> CommandSeek
|
|
#ifdef WITH_BENCHMARK
|
|
seek generator (BenchmarkOptions ps mode) = do
|
|
runner <- generator ps
|
|
liftIO $ runMode mode [ bench (unwords ps) $ nfIO runner ]
|
|
seek _ (BenchmarkDatabases mode n) = benchmarkDbs mode n
|
|
#else
|
|
seek _ _ = giveup "git-annex is not built with benchmarking support"
|
|
#endif
|