Return self in subscribeState:usingBlock of STNetTaskGroup to support invocation chain.

kevin-lyn · kevin-lyn · commit 8d99bb29b2c5 · 2016-05-08T22:57:07.000+08:00
diff --git a/STNetTaskQueue/STNetTaskGroup.h b/STNetTaskQueue/STNetTaskGroup.h
@@ -88,7 +88,7 @@ typedef void (^STNetTaskGroupSubscriptionBlock)(STNetTaskGroup *group, NSError *
  @param state STNetTaskGroupState
  @param block STNetTaskGroupSubscriptionBlock
  */
-- (void)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block;
+- (STNetTaskGroup *)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block;
 
 /**
  Start executing tasks in this group.
diff --git a/STNetTaskQueue/STNetTaskGroup.m b/STNetTaskQueue/STNetTaskGroup.m
@@ -45,7 +45,7 @@ - (void)addTask:(STNetTask *)task
     self.tasks = [NSArray arrayWithArray:tasks];
 }
 
-- (void)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block
+- (STNetTaskGroup *)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block
 {
     if (!self.stateToBlock) {
         self.stateToBlock = [NSMutableDictionary new];
@@ -56,6 +56,7 @@ - (void)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubs
         self.stateToBlock[@(state)] = blocks;
     }
     [blocks addObject:[block copy]];
+    return self;
 }
 
 - (void)notifyState:(STNetTaskGroupState)state withError:(NSError *)error

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ - (void)addTask:(STNetTask *)task`
`45`	`45`	`self.tasks = [NSArray arrayWithArray:tasks];`
`46`	`46`	`}`
`47`	`47`
`48`		`-- (void)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block`
	`48`	`+- (STNetTaskGroup *)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubscriptionBlock)block`
`49`	`49`	`{`
`50`	`50`	`if (!self.stateToBlock) {`
`51`	`51`	`self.stateToBlock = [NSMutableDictionary new];`
`@@ -56,6 +56,7 @@ - (void)subscribeState:(STNetTaskGroupState)state usingBlock:(STNetTaskGroupSubs`
`56`	`56`	`self.stateToBlock[@(state)] = blocks;`
`57`	`57`	`}`
`58`	`58`	`[blocks addObject:[block copy]];`
	`59`	`+ return self;`
`59`	`60`	`}`
`60`	`61`
`61`	`62`	`- (void)notifyState:(STNetTaskGroupState)state withError:(NSError *)error`