mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-09 06:31:02 +00:00
enable TopK on mutex fields
I think it was just an oversight that it wasn't, because this seems to work
This commit is contained in:
parent
d47cf8d3de
commit
58b4f40cdc
2 changed files with 31 additions and 1 deletions
|
|
@ -2098,7 +2098,7 @@ func (e *executor) executeTopKShard(ctx context.Context, qcx *Qcx, index string,
|
|||
return e.executeTopKShardTime(ctx, tx, filterBitmap, index, fieldName, shard, fromTime, toTime)
|
||||
}
|
||||
fallthrough
|
||||
case FieldTypeSet:
|
||||
case FieldTypeSet, FieldTypeMutex:
|
||||
return e.executeTopKShardSet(ctx, tx, filterBitmap, index, fieldName, shard)
|
||||
default:
|
||||
return nil, errors.Errorf("field type %q is not yet supported by TopK", ftype)
|
||||
|
|
|
|||
|
|
@ -1719,6 +1719,36 @@ func TestExecutor_Execute_TopK_Set(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestExecutor_Execute_TopK_Mutex(t *testing.T) {
|
||||
c := test.MustRunCluster(t, 3)
|
||||
defer c.Close()
|
||||
|
||||
// Load some test data into a mutex field.
|
||||
c.CreateField(t, "i", pilosa.IndexOptions{TrackExistence: true}, "f", pilosa.OptFieldTypeMutex(pilosa.CacheTypeRanked, 10))
|
||||
c.ImportBits(t, "i", "f", [][2]uint64{
|
||||
{0, 0},
|
||||
{0, ShardWidth + 2},
|
||||
{10, 2},
|
||||
{10, ShardWidth},
|
||||
{10, 2 * ShardWidth},
|
||||
{10, ShardWidth + 1},
|
||||
{20, ShardWidth},
|
||||
})
|
||||
|
||||
// Execute query.
|
||||
if result, err := c.GetNode(0).API.Query(context.Background(), &pilosa.QueryRequest{Index: "i", Query: `TopK(f, k=2)`}); err != nil {
|
||||
t.Fatal(err)
|
||||
} else if !reflect.DeepEqual(result.Results, []interface{}{&pilosa.PairsField{
|
||||
Pairs: []pilosa.Pair{
|
||||
{ID: 10, Count: 3},
|
||||
{ID: 0, Count: 2},
|
||||
},
|
||||
Field: "f",
|
||||
}}) {
|
||||
t.Fatalf("unexpected result: %s", spew.Sdump(result))
|
||||
}
|
||||
}
|
||||
|
||||
func TestExecutor_Execute_TopK_Time(t *testing.T) {
|
||||
c := test.MustRunCluster(t, 3)
|
||||
defer c.Close()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue