diff --git a/src/main/scala/service/PullRequestService.scala b/src/main/scala/service/PullRequestService.scala index 950c4d3..5374b99 100644 --- a/src/main/scala/service/PullRequestService.scala +++ b/src/main/scala/service/PullRequestService.scala @@ -9,13 +9,10 @@ import PullRequestService._ def getPullRequest(owner: String, repository: String, issueId: Int): Option[(Issue, PullRequest)] = - defining(getIssue(owner, repository, issueId.toString)){ issue => - if(issue.isDefined){ - Query(PullRequests).filter(_.byPrimaryKey(owner, repository, issueId)).firstOption match { - case Some(pullreq) => Some((issue.get, pullreq)) - case None => None - } - } else None + getIssue(owner, repository, issueId.toString).flatMap{ issue => + Query(PullRequests).filter(_.byPrimaryKey(owner, repository, issueId)).firstOption.map{ + pullreq => (issue, pullreq) + } } def getPullRequestCountGroupByUser(closed: Boolean, owner: String, repository: Option[String]): List[PullRequestCount] = @@ -54,4 +51,4 @@ case class PullRequestCount(userName: String, count: Int) -} \ No newline at end of file +}