mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-12 15:51:01 +00:00
957cba1768
- rbfcfg package holds Config for --rbf- command line flags - wal.go: replace linear search with bisection for wal segment
247 lines
7 KiB
Go
247 lines
7 KiB
Go
// Copyright 2020 Pilosa Corp.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package pilosa
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/pilosa/pilosa/v2/rbf"
|
|
)
|
|
|
|
// Shard per db evaluation
|
|
func TestShardPerDB_SetBit(t *testing.T) {
|
|
f, idx, tx := mustOpenFragment(t, "i", "f", viewStandard, 0, "")
|
|
_ = idx
|
|
defer f.Clean(t)
|
|
|
|
// Set bits on the fragment.
|
|
if _, err := f.setBit(tx, 120, 1); err != nil {
|
|
t.Fatal(err)
|
|
} else if _, err := f.setBit(tx, 120, 6); err != nil {
|
|
t.Fatal(err)
|
|
} else if _, err := f.setBit(tx, 121, 0); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// should have two containers set in the fragment.
|
|
|
|
// Verify counts on rows.
|
|
if n := f.mustRow(tx, 120).Count(); n != 2 {
|
|
t.Fatalf("unexpected count: %d", n)
|
|
} else if n := f.mustRow(tx, 121).Count(); n != 1 {
|
|
t.Fatalf("unexpected count: %d", n)
|
|
}
|
|
|
|
// commit the change, and verify it is still there
|
|
panicOn(tx.Commit())
|
|
|
|
// Close and reopen the fragment & verify the data.
|
|
err := f.Reopen() // roaring data not being flushed? red on roaring
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tx = idx.holder.txf.NewTx(Txo{Write: writable, Index: idx, Fragment: f, Shard: f.shard})
|
|
defer tx.Rollback()
|
|
|
|
if n := f.mustRow(tx, 120).Count(); n != 2 {
|
|
t.Fatalf("unexpected count (reopen): %d", n)
|
|
} else if n := f.mustRow(tx, 121).Count(); n != 1 {
|
|
t.Fatalf("unexpected count (reopen): %d", n)
|
|
}
|
|
}
|
|
|
|
// test that we find all *local* shards
|
|
func Test_DBPerShard_GetShardsForIndex_LocalOnly(t *testing.T) {
|
|
tmpdir, err := ioutil.TempDir("", "Test_DBPerShard_GetShardsForIndex_LocalOnly")
|
|
panicOn(err)
|
|
|
|
orig := os.Getenv("PILOSA_TXSRC")
|
|
defer os.Setenv("PILOSA_TXSRC", orig) // must restore or will mess up other tests!
|
|
|
|
for _, src := range []string{"lmdb", "roaring", "bolt", "rbf"} {
|
|
|
|
os.Setenv("PILOSA_TXSRC", src)
|
|
|
|
// must make Holder AFTER setting src.
|
|
holder := NewHolder(tmpdir, nil)
|
|
|
|
makeSampleRoaringDir(tmpdir, src, 1, holder)
|
|
|
|
idx, err := NewIndex(holder, tmpdir, "rick")
|
|
panicOn(err)
|
|
estd := "rick/_exists/views/standard"
|
|
std := "rick/f/views/standard"
|
|
|
|
shards, err := holder.txf.GetShardsForIndex(idx, tmpdir+sep+std, false)
|
|
panicOn(err)
|
|
|
|
for _, shard := range []uint64{93, 223, 221, 215, 219, 217} {
|
|
if !shards[shard] {
|
|
panic(fmt.Sprintf("missing shard=%v from shards='%#v'", shard, shards))
|
|
}
|
|
}
|
|
if src == "roaring" {
|
|
// check estd too
|
|
shards, err = holder.txf.GetShardsForIndex(idx, tmpdir+sep+estd, false)
|
|
panicOn(err)
|
|
for _, shard := range []uint64{93, 223, 221, 215, 219, 217} {
|
|
if !shards[shard] {
|
|
panic(fmt.Sprintf("missing shard=%v from shards='%#v'", shard, shards))
|
|
}
|
|
}
|
|
}
|
|
holder.Close()
|
|
}
|
|
}
|
|
|
|
// data for Test_DBPerShard_GetShardsForIndex
|
|
//
|
|
var sampleRoaringDirList = map[string]string{"roaring": `
|
|
rick/f/views/standard/fragments/215.cache
|
|
rick/f/views/standard/fragments/221.cache
|
|
rick/f/views/standard/fragments/223.cache
|
|
rick/f/views/standard/fragments/93.cache
|
|
rick/f/views/standard/fragments/217.cache
|
|
rick/f/views/standard/fragments/219.cache
|
|
rick/f/views/standard/fragments/217
|
|
rick/f/views/standard/fragments/219
|
|
rick/f/views/standard/fragments/215
|
|
rick/f/views/standard/fragments/221
|
|
rick/f/views/standard/fragments/223
|
|
rick/f/views/standard/fragments/93
|
|
rick/_exists/views/standard/fragments/221
|
|
rick/_exists/views/standard/fragments/215
|
|
rick/_exists/views/standard/fragments/217
|
|
rick/_exists/views/standard/fragments/93
|
|
rick/_exists/views/standard/fragments/219
|
|
rick/_exists/views/standard/fragments/223
|
|
`,
|
|
"lmdb": `
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0093-lmdb@
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0215-lmdb@
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0217-lmdb@
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0219-lmdb@
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0221-lmdb@
|
|
rick.index.txstores@@@/store-lmdb@@/shard.0223-lmdb@
|
|
`,
|
|
"bolt": `
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0093-boltdb@/bolt.db
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0215-boltdb@/bolt.db
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0217-boltdb@/bolt.db
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0219-boltdb@/bolt.db
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0221-boltdb@/bolt.db
|
|
rick.index.txstores@@@/store-boltdb@@/shard.0223-boltdb@/bolt.db
|
|
`,
|
|
"rbf": `
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0093-rbfdb@
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0215-rbfdb@
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0217-rbfdb@
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0219-rbfdb@
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0221-rbfdb@
|
|
rick.index.txstores@@@/store-rbfdb@@/shard.0223-rbfdb@
|
|
`,
|
|
}
|
|
|
|
func makeSampleRoaringDir(root, txsrc string, minBytes int, h *Holder) {
|
|
|
|
index := "rick"
|
|
shards := []uint64{0, 93, 215, 217, 219, 221, 223}
|
|
fns := strings.Split(sampleRoaringDirList[txsrc], "\n")
|
|
for i, fn := range fns {
|
|
if fn == "" {
|
|
continue
|
|
}
|
|
var shard uint64
|
|
if txsrc != "roaring" {
|
|
// only have shards for the non-roaring
|
|
shard = shards[i]
|
|
}
|
|
switch txsrc {
|
|
case "lmdb":
|
|
makeLMDBtestDB(root+sep+fn, h, shard)
|
|
// also have to make the DBShard in our in-memory tree,
|
|
// or else the search won't find it because
|
|
// DBPerShard won't know anything about it.
|
|
helperCreateDBShard(h, index, shard)
|
|
continue
|
|
case "bolt":
|
|
makeBolttestDB(root+sep+fn, h, shard)
|
|
helperCreateDBShard(h, index, shard)
|
|
continue
|
|
case "rbf":
|
|
makeRBFtestDB(root+sep+fn, h, shard)
|
|
helperCreateDBShard(h, index, shard)
|
|
continue
|
|
}
|
|
|
|
path := root + sep + filepath.Dir(fn)
|
|
panicOn(os.MkdirAll(path, 0755))
|
|
fd, err := os.Create(root + sep + fn)
|
|
panicOn(err)
|
|
if minBytes > 0 {
|
|
_, err := fd.Write(make([]byte, minBytes))
|
|
panicOn(err)
|
|
}
|
|
fd.Close()
|
|
}
|
|
}
|
|
|
|
func helperCreateDBShard(h *Holder, index string, shard uint64) {
|
|
idx, err := h.CreateIndexIfNotExists(index, IndexOptions{})
|
|
panicOn(err)
|
|
dbs, err := h.txf.dbPerShard.GetDBShard(index, shard, idx)
|
|
panicOn(err)
|
|
_ = dbs
|
|
}
|
|
|
|
func makeLMDBtestDB(path string, h *Holder, shard uint64) {
|
|
i := uint64(1)
|
|
w, _ := mustOpenEmptyLMDBWrapper(path)
|
|
LMDBMustSetBitvalue(w, "index", "field", "view", shard, i)
|
|
w.Close()
|
|
|
|
}
|
|
|
|
func makeBolttestDB(path string, h *Holder, shard uint64) {
|
|
i := uint64(1)
|
|
w, _ := mustOpenEmptyBoltWrapper(path)
|
|
BoltMustSetBitvalue(w, "index", "field", "view", shard, i)
|
|
w.Close()
|
|
}
|
|
|
|
func makeRBFtestDB(path string, h *Holder, shard uint64) {
|
|
i := uint64(1)
|
|
|
|
db := rbf.NewDB(path, nil)
|
|
err := db.Open()
|
|
panicOn(err)
|
|
defer db.Close()
|
|
|
|
tx, err := db.Begin(true)
|
|
panicOn(err)
|
|
|
|
err = tx.CreateBitmap("x")
|
|
panicOn(err)
|
|
|
|
_, err = tx.Add("x", i)
|
|
panicOn(err)
|
|
|
|
err = tx.Commit()
|
|
panicOn(err)
|
|
}
|