example: fix storag example

1 files changed, 21 insertions, 0 deletions

diff --git a/examples/object_storage/storage.go b/examples/object_storage/storage.go
index 5868591..bb9ff29 100644
--- a/examples/object_storage/storage.go
+++ b/examples/object_storage/storage.go
@@ -137,6 +137,27 @@ func (i *AerospikeObjectIter) Next() (core.Object, error) {
 	return o, nil
 }
 
+func (i *AerospikeObjectIter) ForEach(cb func(obj core.Object) error) error {
+	for {
+		obj, err := i.Next()
+		if err != nil {
+			if err == io.EOF {
+				return nil
+			}
+
+			return err
+		}
+
+		if err := cb(obj); err != nil {
+			if err == core.ErrStop {
+				return nil
+			}
+
+			return err
+		}
+	}
+}
+
 func (i *AerospikeObjectIter) Close() {}
 
 type AerospikeReferenceStorage struct {