mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package bleve
|
|
|
|
import (
|
|
"github.com/blevesearch/bleve/v2"
|
|
)
|
|
|
|
// FlushingBatch is a batch of operations that automatically flushes to the
|
|
// underlying index once it reaches a certain size.
|
|
type FlushingBatch struct {
|
|
maxBatchSize int
|
|
batch *bleve.Batch
|
|
index bleve.Index
|
|
}
|
|
|
|
// NewFlushingBatch creates a new flushing batch for the specified index. Once
|
|
// the number of operations in the batch reaches the specified limit, the batch
|
|
// automatically flushes its operations to the index.
|
|
func NewFlushingBatch(index bleve.Index, maxBatchSize int) *FlushingBatch {
|
|
return &FlushingBatch{
|
|
maxBatchSize: maxBatchSize,
|
|
batch: index.NewBatch(),
|
|
index: index,
|
|
}
|
|
}
|
|
|
|
// Index add a new index to batch
|
|
func (b *FlushingBatch) Index(id string, data any) error {
|
|
if err := b.batch.Index(id, data); err != nil {
|
|
return err
|
|
}
|
|
return b.flushIfFull()
|
|
}
|
|
|
|
// Delete add a delete index to batch
|
|
func (b *FlushingBatch) Delete(id string) error {
|
|
b.batch.Delete(id)
|
|
return b.flushIfFull()
|
|
}
|
|
|
|
func (b *FlushingBatch) flushIfFull() error {
|
|
if b.batch.Size() < b.maxBatchSize {
|
|
return nil
|
|
}
|
|
return b.Flush()
|
|
}
|
|
|
|
// Flush submit the batch and create a new one
|
|
func (b *FlushingBatch) Flush() error {
|
|
err := b.index.Batch(b.batch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b.batch = b.index.NewBatch()
|
|
return nil
|
|
}
|