Make Event Polling a little bit more customizable.

azenla · azenla · commit 71d16834b6bd · 2014-08-28T13:57:39.000-04:00
diff --git a/lib/src/common/events.dart b/lib/src/common/events.dart
@@ -12,15 +12,13 @@ class EventPoller {
 
   EventPoller(this.github, this.path);
 
-  Stream<Event> start() {
+  Stream<Event> start({bool onlyNew: false, int interval}) {
     if (_timer != null) {
       throw new Exception("Polling already started.");
     }
 
     _controller = new StreamController();
 
-    int interval;
-
     void handleEvent(http.Response response) {
       if (interval == null) {
         interval = int.parse(response.headers['x-poll-interval']);
@@ -32,16 +30,18 @@ class EventPoller {
       
       var json = JSON.decode(response.body);
       
-      for (var item in json) {
-        var event = Event.fromJSON(github, item);
-        
-        if (handledEvents.contains(event.id)) {
-          continue;
+      if (!(onlyNew && _timer == null)) {
+        for (var item in json) {
+          var event = Event.fromJSON(github, item);
+          
+          if (handledEvents.contains(event.id)) {
+            continue;
+          }
+          
+          handledEvents.add(event.id);
+          
+          _controller.add(event);
         }
-        
-        handledEvents.add(event.id);
-        
-        _controller.add(event);
       }
 
       if (_timer == null) {
diff --git a/test/polling.dart b/test/polling.dart
@@ -5,7 +5,7 @@ void main() {
   
   var github = new GitHub(auth: new Authentication.withToken("5fdec2b77527eae85f188b7b2bfeeda170f26883"));
   
-  var poller = github.pollPublicEvents();
+  EventPoller poller = github.pollPublicEvents();
   
   poller.start().listen((event) {
     print("New Event:");
