Skip to content

Commit

Permalink
Condense into single benchmark class
Browse files Browse the repository at this point in the history
  • Loading branch information
clairemcginty committed Sep 18, 2024
1 parent 6a50da6 commit 7bbba13
Showing 1 changed file with 10 additions and 17 deletions.
27 changes: 10 additions & 17 deletions jmh/src/test/scala/magnolify/jmh/MagnolifyBench.scala
Original file line number Diff line number Diff line change
Expand Up @@ -167,29 +167,22 @@ class ExampleBench {
@BenchmarkMode(Array(Mode.AverageTime))
@OutputTimeUnit(TimeUnit.NANOSECONDS)
@State(Scope.Thread)
class ParquetMagnolifyBench {
import MagnolifyBench._

@Benchmark def parquetWrite(state: ParquetStates.ParquetCaseClassWriteState): Unit =
state.writer.write(nested)
@Benchmark def parquetRead(state: ParquetStates.ParquetCaseClassReadState): Nested =
state.reader.read()
}

@BenchmarkMode(Array(Mode.AverageTime))
@OutputTimeUnit(TimeUnit.NANOSECONDS)
@State(Scope.Thread)
class ParquetAvroBench {
class ParquetBench {
import MagnolifyBench._
import ParquetStates._
import magnolify.avro._
import org.apache.avro.generic.GenericRecord

private val record = AvroType[Nested].to(nested)
private val genericRecord = AvroType[Nested].to(nested)

@Benchmark def parquetWriteMagnolify(state: ParquetCaseClassWriteState): Unit =
state.writer.write(nested)
@Benchmark def parquetWriteAvro(state: ParquetAvroWriteState): Unit =
state.writer.write(genericRecord)

@Benchmark def parquetWrite(state: ParquetStates.ParquetAvroWriteState): Unit =
state.writer.write(record)
@Benchmark def parquetRead(state: ParquetStates.ParquetAvroReadState): GenericRecord =
@Benchmark def parquetReadMagnolify(state: ParquetCaseClassReadState): Nested =
state.reader.read()
@Benchmark def parquetReadAvro(state: ParquetAvroReadState): GenericRecord = state.reader.read()
}

object ParquetStates {
Expand Down

0 comments on commit 7bbba13

Please sign in to comment.