address the comments

CodingCat · CodingCat · commit 465ccc68368d · 2017-01-12T20:17:35.000-08:00
diff --git a/streaming/src/main/scala/org/apache/spark/streaming/scheduler/JobScheduler.scala b/streaming/src/main/scala/org/apache/spark/streaming/scheduler/JobScheduler.scala
@@ -200,6 +200,9 @@ class JobScheduler(val ssc: StreamingContext) extends Logging {
     job.setEndTime(completedTime)
     listenerBus.post(StreamingListenerOutputOperationCompleted(job.toOutputOperationInfo))
     logInfo("Finished job " + job.id + " from job set of time " + jobSet.time)
+    if (jobSet.hasCompleted) {
+      listenerBus.post(StreamingListenerBatchCompleted(jobSet.toBatchInfo))
+    }
     job.result match {
       case Failure(e) =>
         reportError("Error running job " + job, e)
@@ -211,7 +214,6 @@ class JobScheduler(val ssc: StreamingContext) extends Logging {
             jobSet.totalDelay / 1000.0, jobSet.time.toString,
             jobSet.processingDelay / 1000.0
           ))
-          listenerBus.post(StreamingListenerBatchCompleted(jobSet.toBatchInfo))
         }
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -200,6 +200,9 @@ class JobScheduler(val ssc: StreamingContext) extends Logging {`
`200`	`200`	`job.setEndTime(completedTime)`
`201`	`201`	`listenerBus.post(StreamingListenerOutputOperationCompleted(job.toOutputOperationInfo))`
`202`	`202`	`logInfo("Finished job " + job.id + " from job set of time " + jobSet.time)`
	`203`	`+ if (jobSet.hasCompleted) {`
	`204`	`+ listenerBus.post(StreamingListenerBatchCompleted(jobSet.toBatchInfo))`
	`205`	`+ }`
`203`	`206`	`job.result match {`
`204`	`207`	`case Failure(e) =>`
`205`	`208`	`reportError("Error running job " + job, e)`
`@@ -211,7 +214,6 @@ class JobScheduler(val ssc: StreamingContext) extends Logging {`
`211`	`214`	`jobSet.totalDelay / 1000.0, jobSet.time.toString,`
`212`	`215`	`jobSet.processingDelay / 1000.0`
`213`	`216`	`))`
`214`		`- listenerBus.post(StreamingListenerBatchCompleted(jobSet.toBatchInfo))`
`215`	`217`	`}`
`216`	`218`	`}`
`217`	`219`	`}`