Allow CacheType to be set for a RangeEnabled frame (to apply to the standard frame)

This commit is contained in:
Travis Turner 2017-11-06 14:32:20 -06:00
parent 0d5a2e7bfd
commit 7e797efdc2
No known key found for this signature in database
GPG key ID: 7F08008DFD9314C9
6 changed files with 72 additions and 58 deletions

View file

@ -500,15 +500,17 @@ func TestCluster_ResizeStates(t *testing.T) {
}
// Add Bit Data to node0.
tc.CreateFrame("i", "f", pilosa.FrameOptions{})
if err := tc.CreateFrame("i", "f", pilosa.FrameOptions{}); err != nil {
t.Fatal(err)
}
tc.SetBit("i", "f", "standard", 1, 101, nil)
tc.SetBit("i", "f", "standard", 1, 1300000, nil)
// Add Field Data to node0.
tc.CreateFrame("i", "fields", pilosa.FrameOptions{
if err := tc.CreateFrame("i", "fields", pilosa.FrameOptions{
InverseEnabled: false,
RangeEnabled: true,
CacheType: pilosa.CacheTypeNone,
//CacheType: pilosa.CacheTypeNone,
Fields: []*pilosa.Field{
{
Name: "fld0",
@ -517,14 +519,18 @@ func TestCluster_ResizeStates(t *testing.T) {
Max: 100,
},
},
})
}); err != nil {
t.Fatal(err)
}
tc.SetFieldValue("i", "fields", 1, "fld0", -10)
tc.SetFieldValue("i", "fields", 1, "fld0", 10)
tc.SetFieldValue("i", "fields", 1300000, "fld0", -99)
tc.SetFieldValue("i", "fields", 1300000, "fld0", 99)
// AddNode needs to block until the resize process has completed.
tc.AddNode(false)
if err := tc.AddNode(false); err != nil {
t.Fatal(err)
}
node0 := tc.Clusters[0]
node1 := tc.Clusters[1]

View file

@ -254,6 +254,7 @@ func (f *Fragment) openCache() error {
f.cache = NewLRUCache(f.CacheSize)
case CacheTypeNone:
f.cache = NewNopCache()
return nil
default:
return ErrInvalidCacheType
}
@ -1451,6 +1452,10 @@ func (f *Fragment) flushCache() error {
return nil
}
if f.CacheType == CacheTypeNone {
return nil
}
// Retrieve a list of row ids from the cache.
ids := f.cache.IDs()

View file

@ -22,6 +22,7 @@ import (
"os"
"path/filepath"
"sort"
"strings"
"sync"
"time"
@ -584,6 +585,12 @@ func (f *Frame) CreateViewIfNotExists(name string) (*View, error) {
}
view := f.newView(f.ViewPath(name), name)
// Never keep a cache for field views.
if strings.HasPrefix(name, ViewFieldPrefix) {
view.cacheType = CacheTypeNone
}
if err := view.Open(); err != nil {
return nil, err
}

View file

@ -454,8 +454,6 @@ func (i *Index) createFrame(name string, opt FrameOptions) (*Frame, error) {
if opt.RangeEnabled {
if opt.InverseEnabled {
return nil, ErrInverseRangeNotAllowed
} else if opt.CacheType != "" && opt.CacheType != CacheTypeNone {
return nil, ErrRangeCacheNotAllowed
}
} else {
if len(opt.Fields) > 0 {

View file

@ -136,18 +136,6 @@ func TestIndex_CreateFrame(t *testing.T) {
}
})
t.Run("ErrRangeCacheNotAllowed", func(t *testing.T) {
index := test.MustOpenIndex()
defer index.Close()
if _, err := index.CreateFrame("f", pilosa.FrameOptions{
RangeEnabled: true,
CacheType: pilosa.CacheTypeRanked,
}); err != pilosa.ErrRangeCacheNotAllowed {
t.Fatal(err)
}
})
t.Run("RangeEnabledWithCacheTypeNone", func(t *testing.T) {
index := test.MustOpenIndex()
defer index.Close()

View file

@ -179,7 +179,6 @@ func (t *TestCluster) AddNode(saveTopology bool) error {
URI: c.URI,
}
//go coord.ReceiveEvent(ev)
if err := coord.ReceiveEvent(ev); err != nil {
return err
}
@ -320,7 +319,10 @@ func (t *TestCluster) SendAsync(pb proto.Message) error {
func (t *TestCluster) SendTo(to *pilosa.Node, pb proto.Message) error {
switch obj := pb.(type) {
case *internal.ResizeInstruction:
t.FollowResizeInstruction(obj)
err := t.FollowResizeInstruction(obj)
if err != nil {
return err
}
case *internal.ResizeInstructionComplete:
coord := t.clusterByURI(to.URI)
go coord.MarkResizeInstructionComplete(obj)
@ -338,50 +340,58 @@ func (t *TestCluster) FollowResizeInstruction(instr *internal.ResizeInstruction)
Error: "",
}
// figure out which node it was meant for, then call the operation on that cluster
// basically need to mimic this: client.RetrieveSliceFromURI(context.Background(), src.Index, src.Frame, src.View, src.Slice, srcURI)
instrURI := pilosa.DecodeURI(instr.URI)
destCluster := t.clusterByURI(instrURI)
// Stop processing on any error.
if err := func() error {
// Sync the schema received in the resize instruction.
if err := destCluster.Holder.ApplySchema(instr.Schema); err != nil {
return err
}
// figure out which node it was meant for, then call the operation on that cluster
// basically need to mimic this: client.RetrieveSliceFromURI(context.Background(), src.Index, src.Frame, src.View, src.Slice, srcURI)
instrURI := pilosa.DecodeURI(instr.URI)
destCluster := t.clusterByURI(instrURI)
for _, src := range instr.Sources {
srcURI := pilosa.DecodeURI(src.URI)
srcCluster := t.clusterByURI(srcURI)
// Sync the schema received in the resize instruction.
if err := destCluster.Holder.ApplySchema(instr.Schema); err != nil {
return err
}
srcFragment := srcCluster.Holder.Fragment(src.Index, src.Frame, src.View, src.Slice)
destFragment := destCluster.Holder.Fragment(src.Index, src.Frame, src.View, src.Slice)
if destFragment == nil {
// Create fragment on destination if it doesn't exist.
f := destCluster.Holder.Frame(src.Index, src.Frame)
v := f.View(src.View)
var err error
destFragment, err = v.CreateFragmentIfNotExists(src.Slice)
if err != nil {
for _, src := range instr.Sources {
srcURI := pilosa.DecodeURI(src.URI)
srcCluster := t.clusterByURI(srcURI)
srcFragment := srcCluster.Holder.Fragment(src.Index, src.Frame, src.View, src.Slice)
destFragment := destCluster.Holder.Fragment(src.Index, src.Frame, src.View, src.Slice)
if destFragment == nil {
// Create fragment on destination if it doesn't exist.
f := destCluster.Holder.Frame(src.Index, src.Frame)
v := f.View(src.View)
var err error
destFragment, err = v.CreateFragmentIfNotExists(src.Slice)
if err != nil {
return err
}
}
buf := bytes.NewBuffer(nil)
bw := bufio.NewWriter(buf)
br := bufio.NewReader(buf)
// Get the fragment from source.
if _, err := srcFragment.WriteTo(bw); err != nil {
return err
}
// Flush the bufio.buf to the io.Writer (buf).
bw.Flush()
// Write data to destination.
if _, err := destFragment.ReadFrom(br); err != nil {
return err
}
}
buf := bytes.NewBuffer(nil)
bw := bufio.NewWriter(buf)
br := bufio.NewReader(buf)
// Get the fragment from source.
if _, err := srcFragment.WriteTo(bw); err != nil {
return err
}
// Flush the bufio.buf to the io.Writer (buf).
bw.Flush()
// Write data to destination.
if _, err := destFragment.ReadFrom(br); err != nil {
return err
}
return nil
}(); err != nil {
complete.Error = err.Error()
}
node := &pilosa.Node{