[BuilderTransform] Handle ReturnStmt via CONTROL_FLOW_STMT macro

Author: xedin

lib/Sema/BuilderTransform.cpp

@@ -337,12 +337,6 @@ class BuilderClosureVisitor
     return captureExpr(call, /*oneWay=*/true, braceStmt);
   }
 
-  VarDecl *visitReturnStmt(ReturnStmt *stmt) {
-    if (!unhandledNode)
-      unhandledNode = stmt;
-    return nullptr;
-  }
-
   VarDecl *visitDoStmt(DoStmt *doStmt) {
     auto childVar = visitBraceStmt(doStmt->getBody());
     if (!childVar)
@@ -904,6 +898,7 @@ class BuilderClosureVisitor
   CONTROL_FLOW_STMT(Fallthrough)
   CONTROL_FLOW_STMT(Fail)
   CONTROL_FLOW_STMT(PoundAssert)
+  CONTROL_FLOW_STMT(Return)
 
 #undef CONTROL_FLOW_STMT
 };