Skip to content

Commit b6bc80b

Browse files
committed
Rework comments
1 parent d6d4175 commit b6bc80b

File tree

1 file changed

+10
-16
lines changed

1 file changed

+10
-16
lines changed

core/src/main/scala/org/apache/spark/network/nio/Connection.scala

Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ import java.util.LinkedList
2424

2525
import org.apache.spark._
2626

27-
import scala.annotation.tailrec
2827
import scala.collection.mutable.{ArrayBuffer, HashMap}
2928

3029
private[nio]
@@ -217,23 +216,18 @@ class SendingConnection(val address: InetSocketAddress, selector_ : Selector,
217216
/* val message = messages(nextMessageToBeUsed) */
218217

219218
val message = if (securityMgr.isAuthenticationEnabled() && !isSaslComplete()) {
220-
// only send security messages until sasl is complete
221-
@tailrec
222-
def getFirstSecurityMessage(pos: Int): Option[Message] = {
223-
if (pos >= messages.size()) {
224-
None
225-
} else {
226-
if (messages.get(pos).isSecurityNeg) {
227-
Some(messages.remove(pos))
228-
} else {
229-
getFirstSecurityMessage(pos + 1)
230-
}
219+
// only allow sending of security messages until sasl is complete
220+
var pos = 0
221+
var securityMsg = null
222+
while (pos < messages.size() && securityMsg == null) {
223+
if (messages.get(pos).isSecurityNeg) {
224+
securityMsg = messages.remove(pos)
231225
}
226+
pos = pos + 1
232227
}
233-
getFirstSecurityMessage(0) match {
234-
case Some(msg) => msg
235-
case None => return None
236-
}
228+
// didn't find any security messages and auth isn't completed so return
229+
if (securityMsg == null) return None
230+
securityMsg
237231
} else {
238232
messages.removeFirst()
239233
}

0 commit comments

Comments
 (0)