Improve ActiveStream

DeclarativeHub · Nov 15, 2015 · 9743ada · 9743ada
1 parent c86dce9
commit 9743ada
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ReactiveKit/Streams/ActiveStream.swift b/ReactiveKit/Streams/ActiveStream.swift
@@ -58,6 +58,13 @@ public class ActiveStream<Event>: ActiveStreamType {
     self.selfReference = tmpSelfReference
   }
 
+  public init(limit: Int = 0) {
+    self.buffer = StreamBuffer(limit: limit)
+    let tmpSelfReference = Reference(self)
+    tmpSelfReference.release()
+    self.selfReference = tmpSelfReference
+  }
+
   public func observe(sink: Sink) -> DisposableType {
     return observe(on: ImmediateExecutionContext, sink: sink)
   }
@@ -89,7 +96,7 @@ public class ActiveStream<Event>: ActiveStreamType {
     return try buffer.last()
   }
 
-  internal func next(event: Event) {
+  public func next(event: Event) {
     buffer.next(event)
     dispatchNext(event)
   }