mirror of
https://github.com/luxfi/zapdb.git
synced 2026-10-30 15:12:20 +00:00
Manish R JainandGitHub
f0958c6bf2
This API would allow Badger to completely get rid of the given prefix, P. It does this by: 1. Flushing memtables to disk (skipping over the keys with P). 2. Doing normal compaction from L0 -> L1 (skipping keys with P). 3. Doing same level compaction on all the rest of the levels, picking tables which might have the prefix and rewriting them to new tables (again, skipping keys with P). Before starting, all the writes are paused, the usual bits which apply to `DropAll` apply here as well. Note: This PR has a breaking change for the Logger interface. It adds a Debugf API. Changes: * Working drop prefix * Make the drop prefix more robust. Add a new Debugf API for logging. * Add parallel tests for DropPrefix * Avoid a test failure due to removing p dir. * Only test DropPrefix in DropPrefix test. * Address Martin's comments
72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
/*
|
|
* Copyright 2017 Dgraph Labs, Inc. and Contributors
|
|
*
|
|
* 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 badger
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type mockLogger struct {
|
|
output string
|
|
}
|
|
|
|
func (l *mockLogger) Errorf(f string, v ...interface{}) {
|
|
l.output = fmt.Sprintf("ERROR: "+f, v...)
|
|
}
|
|
|
|
func (l *mockLogger) Infof(f string, v ...interface{}) {
|
|
l.output = fmt.Sprintf("INFO: "+f, v...)
|
|
}
|
|
|
|
func (l *mockLogger) Warningf(f string, v ...interface{}) {
|
|
l.output = fmt.Sprintf("WARNING: "+f, v...)
|
|
}
|
|
|
|
func (l *mockLogger) Debugf(f string, v ...interface{}) {
|
|
l.output = fmt.Sprintf("DEBUG: "+f, v...)
|
|
}
|
|
|
|
// Test that the DB-specific log is used instead of the global log.
|
|
func TestDbLog(t *testing.T) {
|
|
l := &mockLogger{}
|
|
opt := Options{Logger: l}
|
|
|
|
opt.Errorf("test")
|
|
require.Equal(t, "ERROR: test", l.output)
|
|
opt.Infof("test")
|
|
require.Equal(t, "INFO: test", l.output)
|
|
opt.Warningf("test")
|
|
require.Equal(t, "WARNING: test", l.output)
|
|
}
|
|
|
|
// Test that the global logger is used when no logger is specified in Options.
|
|
func TestNoDbLog(t *testing.T) {
|
|
l := &mockLogger{}
|
|
opt := Options{}
|
|
opt.Logger = l
|
|
|
|
opt.Errorf("test")
|
|
require.Equal(t, "ERROR: test", l.output)
|
|
opt.Infof("test")
|
|
require.Equal(t, "INFO: test", l.output)
|
|
opt.Warningf("test")
|
|
require.Equal(t, "WARNING: test", l.output)
|
|
}
|