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

Compare commits

base: Mirrors:cb8902669e8475bd781ef47219281116ff97a4e3
Branches Tags
Mirrors:main
...
compare: Mirrors:c632d708f2f7e5b34e6cb54fde21a7c149ee2e49
Branches Tags
Mirrors:main

3 Commits
cb8902669e ... c632d708f2

Author SHA1 Message Date
William Woods
c632d708f2
Merge 0917e50b70 into fb54d8b549 2023-05-22 17:37:28 -05:00
William Woods
0917e50b70 Refactored code for performance and readability 2023-04-03 07:11:21 +00:00
William Woods
ef04e156fd Refactored code for performance and readability 2023-04-03 04:26:18 +00:00
1 changed files with 23 additions and 39 deletions
Show Stats Expand all files Collapse all files

60
follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/clients/geoduck/UserLocationFetcher.scala
View File

@ -1,11 +1,9 @@
package com.twitter.follow_recommendations.common.clients.geoduck
import com.twitter.finagle.stats.StatsReceiver
import com.twitter.follow_recommendations.common.models.GeohashAndCountryCode
import com.twitter.stitch.Stitch
import javax.inject.Inject
import javax.inject.Singleton
import javax.inject.{Inject, Singleton}
@Singleton
class UserLocationFetcher @Inject() (
@ -13,47 +11,33 @@ class UserLocationFetcher @Inject() (
reverseGeocodeClient: ReverseGeocodeClient,
statsReceiver: StatsReceiver) {
private val stats: StatsReceiver = statsReceiver.scope("user_location_fetcher")
private val totalRequestsCounter = stats.counter("requests")
private val emptyResponsesCounter = stats.counter("empty")
private val locationServiceExceptionCounter = stats.counter("location_service_exception")
private val reverseGeocodeExceptionCounter = stats.counter("reverse_geocode_exception")
private val stats = statsReceiver.scope("user_location_fetcher")
def getGeohashAndCountryCode(
userId: Option[Long],
ipAddress: Option[String]
): Stitch[Option[GeohashAndCountryCode]] = {
totalRequestsCounter.incr()
val lscLocationStitch = Stitch
.collect {
userId.map(locationServiceClient.getGeohashAndCountryCode)
}.rescue {
case _: Exception =>
locationServiceExceptionCounter.incr()
Stitch.None
}
def getGeohashAndCountryCode(userId: Option[Long], ipAddress: Option[String]): Stitch[Option[GeohashAndCountryCode]] = {
val totalRequestsCounter = stats.counter("requests").incr()
val ipLocationStitch = Stitch
.collect {
ipAddress.map(reverseGeocodeClient.getGeohashAndCountryCode)
}.rescue {
case _: Exception =>
reverseGeocodeExceptionCounter.incr()
Stitch.None
}
val lscLocationStitch = Stitch.collect(userId.map(locationServiceClient.getGeohashAndCountryCode)).rescue {
case _: Exception =>
stats.counter("location_service_exception").incr()
Stitch.None
}
val ipLocationStitch = Stitch.collect(ipAddress.map(reverseGeocodeClient.getGeohashAndCountryCode)).rescue {
case _: Exception =>
stats.counter("reverse_geocode_exception").incr()
Stitch.None
}
Stitch.join(lscLocationStitch, ipLocationStitch).map {
case (lscLocation, ipLocation) => {
val geohash = lscLocation.flatMap(_.geohash).orElse(ipLocation.flatMap(_.geohash))
val countryCode =
lscLocation.flatMap(_.countryCode).orElse(ipLocation.flatMap(_.countryCode))
(geohash, countryCode) match {
case (None, None) =>
emptyResponsesCounter.incr()
case (lscLocation, ipLocation) =>
(lscLocation.flatMap(_.geohash).orElse(ipLocation.flatMap(_.geohash)),
lscLocation.flatMap(_.countryCode).orElse(ipLocation.flatMap(_.countryCode))) match {
case (Some(geohash), Some(countryCode)) =>
Some(GeohashAndCountryCode(geohash, countryCode))
case _ =>
stats.counter("empty").incr()
None
case _ => Some(GeohashAndCountryCode(geohash, countryCode))
}
}
}
}
}