fix broken context stuff after rebase

This commit is contained in:
jaffee 2016-11-15 17:04:51 -06:00
parent 86d408310c
commit 256d516726
3 changed files with 7 additions and 4 deletions

View file

@ -11,6 +11,7 @@ import (
"time"
"context"
"github.com/umbel/pilosa"
"math/rand"
)
@ -111,7 +112,7 @@ func (b *DiagonalSetBits) Run(agentNum int) map[string]interface{} {
for n := 0; n < b.Iterations; n++ {
iterID := agentizeNum(n, b.Iterations, agentNum)
query := fmt.Sprintf("SetBit(%d, 'frame.n', %d)", b.BaseBitmapID+iterID, b.BaseProfileID+iterID)
b.cli.ExecuteQuery(b.DB, query, true)
b.cli.ExecuteQuery(context.TODO(), b.DB, query, true)
}
return results
}
@ -175,7 +176,7 @@ func (b *MultiDBSetBits) Run(agentNum int) map[string]interface{} {
}
for n := 0; n < b.Iterations; n++ {
query := fmt.Sprintf("SetBit(%d, 'frame.n', %d)", b.BaseBitmapID+n, b.BaseProfileID+n)
b.cli.ExecuteQuery("multidb"+strconv.Itoa(agentNum), query, true)
b.cli.ExecuteQuery(context.TODO(), "multidb"+strconv.Itoa(agentNum), query, true)
}
return results
}
@ -266,7 +267,7 @@ func (b *RandomSetBits) Run(agentNum int) map[string]interface{} {
bitmapID := rng.Int63n(b.BitmapIDRange)
profID := rng.Int63n(b.ProfileIDRange)
query := fmt.Sprintf("SetBit(%d, 'frame.n', %d)", b.BaseBitmapID+bitmapID, b.BaseProfileID+profID)
b.cli.ExecuteQuery(b.DB, query, true)
b.cli.ExecuteQuery(context.TODO(), b.DB, query, true)
}
return results
}

View file

@ -2,8 +2,8 @@ package main_test
import (
"bytes"
"encoding/json"
"context"
"encoding/json"
"fmt"
"io"
"io/ioutil"
@ -16,6 +16,7 @@ import (
"testing"
"testing/quick"
"context"
"github.com/BurntSushi/toml"
"github.com/pilosa/pilosa"
main "github.com/pilosa/pilosa/cmd/pilosa"

View file

@ -23,6 +23,7 @@ import (
"time"
"unsafe"
"context"
"encoding/json"
"github.com/umbel/pilosa"
"github.com/umbel/pilosa/bench"