123 lines
2.5 KiB
Go
123 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.nicholasnovak.io/nnovak/spatial-db/storage"
|
|
"git.nicholasnovak.io/nnovak/spatial-db/world"
|
|
)
|
|
|
|
var (
|
|
emptyDir = "./empty"
|
|
smallSparseDir = "./small-sparse"
|
|
medSparseDir = "./med-sparse"
|
|
largeSparseDir = "./lg-sparse"
|
|
smallDenseDir = "./small-dense"
|
|
medDenseDir = "./med-dense"
|
|
largeDenseDir = "./lg-dense"
|
|
)
|
|
|
|
// Point densities
|
|
const (
|
|
sparse = 1_000
|
|
dense = 10_000
|
|
)
|
|
|
|
var pointCounts = []int{200, 1_000, 2_500}
|
|
|
|
var dirs = []string{emptyDir, smallSparseDir, medSparseDir, largeSparseDir,
|
|
smallDenseDir, medDenseDir, largeDenseDir}
|
|
|
|
func init() {
|
|
for index, dir := range dirs {
|
|
if index > 2 {
|
|
populateStorageDir(dir, sparse, pointCounts[index%3], false)
|
|
} else {
|
|
populateStorageDir(dir, dense, pointCounts[index%3], false)
|
|
}
|
|
}
|
|
}
|
|
|
|
// insertPointTemplate inserts a configurable variety of points into the server
|
|
func insertPointTemplate(testDir string, b *testing.B) {
|
|
var server storage.SimpleServer
|
|
|
|
server.SetStorageRoot(testDir)
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
pos := world.RandomBlockPosWithRange(2048)
|
|
if err := server.ChangeBlock(pos, world.Generic); err != nil {
|
|
b.Error(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func fetchChunkTemplate(testDir string, b *testing.B) {
|
|
var server storage.SimpleServer
|
|
|
|
server.SetStorageRoot(testDir)
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
pos := world.RandomBlockPosWithRange(2048).ToChunkPos()
|
|
if _, err := server.ReadChunkAt(pos); err != nil {
|
|
b.Error(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Insert blocks
|
|
|
|
func BenchmarkInsertSmallSparse(b *testing.B) {
|
|
insertPointTemplate(smallSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkInsertMedSparse(b *testing.B) {
|
|
insertPointTemplate(medSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkInsertLgSparse(b *testing.B) {
|
|
insertPointTemplate(largeSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkInsertSmallDense(b *testing.B) {
|
|
insertPointTemplate(smallDenseDir, b)
|
|
}
|
|
|
|
func BenchmarkInsertMedDense(b *testing.B) {
|
|
insertPointTemplate(medDenseDir, b)
|
|
}
|
|
|
|
func BenchmarkInsertLgDense(b *testing.B) {
|
|
insertPointTemplate(largeDenseDir, b)
|
|
}
|
|
|
|
// Fetch chunks
|
|
|
|
func BenchmarkFetchChunkSmallSparse(b *testing.B) {
|
|
fetchChunkTemplate(smallSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkFetchChunkMedSparse(b *testing.B) {
|
|
fetchChunkTemplate(medSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkFetchChunkLgSparse(b *testing.B) {
|
|
fetchChunkTemplate(largeSparseDir, b)
|
|
}
|
|
|
|
func BenchmarkFetchChunkSmallDense(b *testing.B) {
|
|
fetchChunkTemplate(smallDenseDir, b)
|
|
}
|
|
|
|
func BenchmarkFetchChunkMedDense(b *testing.B) {
|
|
fetchChunkTemplate(medDenseDir, b)
|
|
}
|
|
|
|
func BenchmarkFetchChunkLgDense(b *testing.B) {
|
|
fetchChunkTemplate(largeDenseDir, b)
|
|
}
|