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.
51 lines
1.6 KiB
Java
51 lines
1.6 KiB
Java
package com.twitter.search.earlybird_root.filters;
|
|
|
|
import java.util.Map;
|
|
|
|
import com.google.common.collect.Maps;
|
|
|
|
import com.twitter.finagle.Service;
|
|
import com.twitter.finagle.SimpleFilter;
|
|
import com.twitter.search.common.metrics.SearchCounter;
|
|
import com.twitter.search.earlybird.thrift.EarlybirdRequest;
|
|
import com.twitter.search.earlybird.thrift.EarlybirdResponse;
|
|
import com.twitter.search.earlybird.thrift.EarlybirdResponseCode;
|
|
import com.twitter.util.Future;
|
|
import com.twitter.util.FutureEventListener;
|
|
|
|
public class ResponseCodeStatFilter
|
|
extends SimpleFilter<EarlybirdRequest, EarlybirdResponse> {
|
|
|
|
private final Map<EarlybirdResponseCode, SearchCounter> responseCodeCounters;
|
|
|
|
/**
|
|
* Create ResponseCodeStatFilter
|
|
*/
|
|
public ResponseCodeStatFilter() {
|
|
responseCodeCounters = Maps.newEnumMap(EarlybirdResponseCode.class);
|
|
for (EarlybirdResponseCode code : EarlybirdResponseCode.values()) {
|
|
SearchCounter stat = SearchCounter.export("response_code_" + code.name().toLowerCase());
|
|
responseCodeCounters.put(code, stat);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Future<EarlybirdResponse> apply(
|
|
final EarlybirdRequest request,
|
|
final Service<EarlybirdRequest, EarlybirdResponse> service) {
|
|
|
|
return service.apply(request).addEventListener(
|
|
new FutureEventListener<EarlybirdResponse>() {
|
|
|
|
@Override
|
|
public void onSuccess(final EarlybirdResponse response) {
|
|
responseCodeCounters.get(response.getResponseCode()).increment();
|
|
}
|
|
|
|
@Override
|
|
public void onFailure(final Throwable cause) { }
|
|
});
|
|
|
|
}
|
|
}
|