Added ability to set initial minimum value for seqset when known.

We know the minimum value when creating a new filestore msgBlk.

Signed-off-by: Derek Collison <derek@nats.io>

server/avl/seqset.go

@@ -62,17 +62,31 @@ func (ss *SequenceSet) Exists(seq uint64) bool {
 	return false
 }
 
+// SetInitialMin should be used to set the initial minimum sequence when known.
+// This will more effectively utilize space versus self selecting.
+// The set should be empty.
+func (ss *SequenceSet) SetInitialMin(min uint64) error {
+	if !ss.IsEmpty() {
+		return ErrSetNotEmpty
+	}
+	ss.root = &node{base: min, h: 1}
+	return nil
+}
+
 // Delete will remove the sequence from the set.
-// Wil optionally remove nodes and rebalance.
-func (ss *SequenceSet) Delete(seq uint64) {
+// Will optionally remove nodes and rebalance.
+// Returns where the sequence was set.
+func (ss *SequenceSet) Delete(seq uint64) bool {
 	if ss == nil || ss.root == nil {
-		return
+		return false
 	}
 	ss.root = ss.root.delete(seq, &ss.changed, &ss.nodes)
 	if ss.changed {
 		ss.changed = false
 		ss.size--
+		return true
 	}
+	return false
 }
 
 // Size returns the number of items in the set.
@@ -228,7 +242,10 @@ func (ss SequenceSet) Encode(buf []byte) ([]byte, error) {
 }
 
 // ErrBadEncoding is returned when we can not decode properly.
-var ErrBadEncoding = errors.New("ss: bad encoding")
+var (
+	ErrBadEncoding = errors.New("ss: bad encoding")
+	ErrSetNotEmpty = errors.New("ss: set not empty")
+)
 
 func Decode(buf []byte) (*SequenceSet, error) {
 	if len(buf) < minLen || buf[0] != magic || buf[1] != version {

server/avl/seqset_test.go

@@ -218,6 +218,23 @@ func TestSeqSetUnion(t *testing.T) {
 	}
 }
 
+func TestSeqSetFirst(t *testing.T) {
+	var ss SequenceSet
+
+	seqs := []uint64{22, 222, 2222, 222_222}
+	for _, seq := range seqs {
+		// Normal case where we pick first/base.
+		ss.Insert(seq)
+		require_True(t, ss.root.base == (seq/numEntries)*numEntries)
+		ss.Empty()
+		// Where we set the minimum start value.
+		ss.SetInitialMin(seq)
+		ss.Insert(seq)
+		require_True(t, ss.root.base == seq)
+		ss.Empty()
+	}
+}
+
 func require_NoError(t *testing.T, err error) {
 	t.Helper()
 	if err != nil {

server/filestore.go

@@ -2221,6 +2221,8 @@ func (fs *fileStore) newMsgBlockForWrite() (*msgBlock, error) {
 	// Remember our last sequence number.
 	mb.first.seq = fs.state.LastSeq + 1
 	mb.last.seq = fs.state.LastSeq
+	// Mark initial base for delete set.
+	mb.dmap.SetInitialMin(mb.first.seq)
 	mb.mu.Unlock()
 
 	// If we know we will need this so go ahead and spin up.