Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-12-22 02:01:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Avoid raw get

Browse Source
This commit is contained in:
Erik Innocent 2023-04-07 15:27:46 -07:00
parent 138bb51997
commit 6ee0d2b7ac
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
graph-feature-service/src/main/scala/com/twitter/graph_feature_service/worker/handlers/WorkerGetIntersectionHandler.scala
View File

@ -75,21 +75,23 @@ class WorkerGetIntersectionHandler @Inject() (
val rightNeighborsOpt =
rightEdgeMap.get(featureType.rightEdgeType).flatMap(_.get(candidateId))
if (leftNeighborsOpt.isEmpty && rightNeighborsOpt.isEmpty) {
EmptyWorkerIntersectionValue
} else if (rightNeighborsOpt.isEmpty) {
EmptyWorkerIntersectionValue.copy(
leftNodeDegree = computeArraySize(leftNeighborsOpt.get)
)
} else if (leftNeighborsOpt.isEmpty) {
EmptyWorkerIntersectionValue.copy(
rightNodeDegree = computeArraySize(rightNeighborsOpt.get)
)
} else {
IntersectionValueCalculator(
leftNeighborsOpt.get,
rightNeighborsOpt.get,
request.intersectionIdLimit)
(leftNeighborsOpt, rightNeighborsOpt) match {
case (Some(leftNeighbors), None) =>
EmptyWorkerIntersectionValue.copy(
leftNodeDegree = computeArraySize(leftNeighbors)
)
case (None, Some(rightNeighbors)) =>
EmptyWorkerIntersectionValue.copy(
rightNodeDegree = computeArraySize(rightNeighbors)
)
case (Some(leftNeighbors), Some(rightNeighbors)) =>
IntersectionValueCalculator(
leftNeighbors,
rightNeighbors,
request.intersectionIdLimit
)
case _ =>
EmptyWorkerIntersectionValue
}
}
}