Added BitIterator interface

FeatureBaseDB · Jun 23, 2017 · cee1d49 · cee1d49
1 parent 9470942
commit cee1d49
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/client.go b/client.go
@@ -184,7 +184,7 @@ func (c *Client) Schema() (*Schema, error) {
 }
 
 // ImportFrame imports bits from the given CSV iterator
-func (c *Client) ImportFrame(frame *Frame, bitIterator *CSVBitIterator, batchSize uint) error {
+func (c *Client) ImportFrame(frame *Frame, bitIterator BitIterator, batchSize uint) error {
 	const sliceWidth = 1048576
 	linesLeft := true
 	bitGroup := map[uint64][]Bit{}

diff --git a/imports.go b/imports.go
@@ -47,9 +47,10 @@ type Bit struct {
 	Timestamp int64
 }
 
-// BitIteratorCallback gets called for each bit in a stream.
-// return false to pause iteration.
-type BitIteratorCallback func(bit Bit) bool
+// BitIterator structs return bits one by one.
+type BitIterator interface {
+	NextBit() (Bit, error)
+}
 
 // CSVBitIterator reads bits from a Reader.
 // Each line should contain a single bit in the following form: