mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-13 14:48:54 +02:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
82 lines
3.5 KiB
Java
82 lines
3.5 KiB
Java
package com.twitter.search.earlybird_root.filters;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
import com.twitter.finagle.Service;
|
|
import com.twitter.finagle.SimpleFilter;
|
|
import com.twitter.search.common.schema.earlybird.EarlybirdCluster;
|
|
import com.twitter.search.common.util.earlybird.EarlybirdResponseMergeUtil;
|
|
import com.twitter.search.earlybird.thrift.EarlybirdResponse;
|
|
import com.twitter.search.earlybird.thrift.EarlybirdResponseCode;
|
|
import com.twitter.search.earlybird_root.common.EarlybirdRequestContext;
|
|
import com.twitter.search.earlybird_root.common.EarlybirdRequestType;
|
|
import com.twitter.search.earlybird_root.validators.FacetsResponseValidator;
|
|
import com.twitter.search.earlybird_root.validators.PassThroughResponseValidator;
|
|
import com.twitter.search.earlybird_root.validators.ServiceResponseValidator;
|
|
import com.twitter.search.earlybird_root.validators.TermStatsResultsValidator;
|
|
import com.twitter.search.earlybird_root.validators.TopTweetsResultsValidator;
|
|
import com.twitter.util.Function;
|
|
import com.twitter.util.Future;
|
|
|
|
/**
|
|
* Filter responsible for handling invalid response returned by downstream services, and
|
|
* translating them into EarlybirdResponseExceptions.
|
|
*/
|
|
public class ServiceResponseValidationFilter
|
|
extends SimpleFilter<EarlybirdRequestContext, EarlybirdResponse> {
|
|
|
|
private final Map<EarlybirdRequestType, ServiceResponseValidator<EarlybirdResponse>>
|
|
requestTypeToResponseValidators = new HashMap<>();
|
|
private final EarlybirdCluster cluster;
|
|
|
|
/**
|
|
* Creates a new filter for handling invalid response
|
|
*/
|
|
public ServiceResponseValidationFilter(EarlybirdCluster cluster) {
|
|
this.cluster = cluster;
|
|
|
|
ServiceResponseValidator<EarlybirdResponse> passThroughValidator =
|
|
new PassThroughResponseValidator();
|
|
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.FACETS, new FacetsResponseValidator(cluster));
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.RECENCY, passThroughValidator);
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.RELEVANCE, passThroughValidator);
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.STRICT_RECENCY, passThroughValidator);
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.TERM_STATS, new TermStatsResultsValidator(cluster));
|
|
requestTypeToResponseValidators
|
|
.put(EarlybirdRequestType.TOP_TWEETS, new TopTweetsResultsValidator(cluster));
|
|
}
|
|
|
|
@Override
|
|
public Future<EarlybirdResponse> apply(
|
|
final EarlybirdRequestContext requestContext,
|
|
Service<EarlybirdRequestContext, EarlybirdResponse> service) {
|
|
return service.apply(requestContext).flatMap(
|
|
new Function<EarlybirdResponse, Future<EarlybirdResponse>>() {
|
|
@Override
|
|
public Future<EarlybirdResponse> apply(EarlybirdResponse response) {
|
|
if (response == null) {
|
|
return Future.exception(new IllegalStateException(
|
|
cluster + " returned null response"));
|
|
}
|
|
|
|
if (response.getResponseCode() == EarlybirdResponseCode.SUCCESS) {
|
|
return requestTypeToResponseValidators
|
|
.get(requestContext.getEarlybirdRequestType())
|
|
.validate(response);
|
|
}
|
|
|
|
return Future.value(EarlybirdResponseMergeUtil.transformInvalidResponse(
|
|
response,
|
|
String.format("Failure from %s (%s)", cluster, response.getResponseCode())));
|
|
}
|
|
});
|
|
}
|
|
}
|