buth · May 2, 2024 05:38 · Feb 22, 2016 · Feb 22, 2016 · Feb 22, 2016
diff --git a/mongodb.go b/mongodb.go
@@ -0,0 +1,29 @@
+package mongodb
+
+type iter struct {
+	done, closed bool
+	iter         *mgo.Iter
+	result       func() interface{}
+}
+
+func (i *iter) Next() (interface{}, bool, error) {
+
+	if !i.done {
+
+		// Get a new instance.
+		r := i.result()
+
+		// Get the next value and return it.
+		i.done = !(i.iter.Next(r))
+		return r, i.done, nil
+	}
+
+	if !i.closed {
+		i.closed = true
+		if err := i.iter.Close(); err != nil {
+			return nil, true, err
+		}
+	}
+
+	return nil, true, nil
+}
diff --git a/iterator.go b/iterator.go
@@ -0,0 +1,5 @@
+package iterator
+
+type Iterator interface {
+	Next() (value interface{}, done bool, err error)
+}