提交 dba222b5 编写于 作者: B Brian Brazil

Make rollback actually rollback.

Also add tests for basic commit/rollback
functionality.

Fixes #39
上级 25d45465
......@@ -654,7 +654,7 @@ func (a *dbAppender) Rollback() error {
var g errgroup.Group
for _, h := range a.heads {
g.Go(h.app.Commit)
g.Go(h.app.Rollback)
}
return g.Wait()
......
......@@ -12,3 +12,78 @@
// limitations under the License.
package tsdb
import (
"io/ioutil"
"os"
"testing"
"github.com/prometheus/tsdb/labels"
)
func TestDataAvailableOnlyAfterCommit(t *testing.T) {
tmpdir, _ := ioutil.TempDir("", "test")
defer os.RemoveAll(tmpdir)
db, err := Open(tmpdir, nil, nil, nil)
if err != nil {
t.Fatalf("Error opening database: %q", err)
}
defer db.Close()
app := db.Appender()
_, err = app.Add(labels.FromStrings("foo", "bar"), 0, 0)
if err != nil {
t.Fatalf("Error adding sample: %q", err)
}
querier := db.Querier(0, 1)
defer querier.Close()
seriesSet := querier.Select(labels.NewEqualMatcher("foo", "bar"))
if seriesSet.Next() {
t.Fatalf("Error, was expecting no matching series.")
}
err = app.Commit()
if err != nil {
t.Fatalf("Error committing: %q", err)
}
querier = db.Querier(0, 1)
defer querier.Close()
seriesSet = querier.Select(labels.NewEqualMatcher("foo", "bar"))
if !seriesSet.Next() {
t.Fatalf("Error, was expecting a matching series.")
}
}
func TestDataNotAvailableAfterRollback(t *testing.T) {
tmpdir, _ := ioutil.TempDir("", "test")
defer os.RemoveAll(tmpdir)
db, err := Open(tmpdir, nil, nil, nil)
if err != nil {
t.Fatalf("Error opening database: %q", err)
}
defer db.Close()
app := db.Appender()
_, err = app.Add(labels.FromStrings("foo", "bar"), 0, 0)
if err != nil {
t.Fatalf("Error adding sample: %q", err)
}
err = app.Rollback()
if err != nil {
t.Fatalf("Error rolling back: %q", err)
}
querier := db.Querier(0, 1)
defer querier.Close()
seriesSet := querier.Select(labels.NewEqualMatcher("foo", "bar"))
if seriesSet.Next() {
t.Fatalf("Error, was expecting no matching series.")
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册