Skip to content

Commit 1ef1b71

Browse files
Chen Zhilingfeast-ci-bot
authored andcommitted
Stop using prom timer since its in seconds (feast-dev#338)
1 parent 8c30640 commit 1ef1b71

File tree

2 files changed

+8
-10
lines changed

2 files changed

+8
-10
lines changed

serving/src/main/java/feast/serving/service/BigQueryServingService.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,6 @@
5555
import feast.serving.ServingAPIProto.JobType;
5656
import feast.serving.util.BigQueryUtil;
5757
import io.grpc.Status;
58-
import io.prometheus.client.Histogram.Timer;
5958
import java.io.IOException;
6059
import java.util.ArrayList;
6160
import java.util.List;
@@ -113,7 +112,7 @@ public GetOnlineFeaturesResponse getOnlineFeatures(GetOnlineFeaturesRequest getF
113112
/** {@inheritDoc} */
114113
@Override
115114
public GetBatchFeaturesResponse getBatchFeatures(GetBatchFeaturesRequest getFeaturesRequest) {
116-
Timer getBatchFeaturesTimer = requestLatency.labels("getBatchFeatures").startTimer();
115+
long startTime = System.currentTimeMillis();
117116
List<FeatureSetSpec> featureSetSpecs =
118117
getFeaturesRequest.getFeatureSetsList().stream()
119118
.map(
@@ -249,7 +248,7 @@ public GetBatchFeaturesResponse getBatchFeatures(GetBatchFeaturesRequest getFeat
249248
})
250249
.start();
251250

252-
getBatchFeaturesTimer.observeDuration();
251+
requestLatency.labels("getBatchFeatures").observe(System.currentTimeMillis() - startTime);
253252
return GetBatchFeaturesResponse.newBuilder().setJob(feastJob).build();
254253
}
255254

serving/src/main/java/feast/serving/service/RedisServingService.java

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@
4646
import io.grpc.Status;
4747
import io.opentracing.Scope;
4848
import io.opentracing.Tracer;
49-
import io.prometheus.client.Histogram.Timer;
5049
import java.util.List;
5150
import java.util.Map;
5251
import java.util.stream.Collectors;
@@ -80,7 +79,7 @@ public GetFeastServingInfoResponse getFeastServingInfo(
8079
@Override
8180
public GetOnlineFeaturesResponse getOnlineFeatures(GetOnlineFeaturesRequest request) {
8281
try (Scope scope = tracer.buildSpan("Redis-getOnlineFeatures").startActive(true)) {
83-
Timer getOnlineFeaturesTimer = requestLatency.labels("getOnlineFeatures").startTimer();
82+
long startTime = System.currentTimeMillis();
8483
GetOnlineFeaturesResponse.Builder getOnlineFeaturesResponseBuilder =
8584
GetOnlineFeaturesResponse.newBuilder();
8685

@@ -121,7 +120,7 @@ public GetOnlineFeaturesResponse getOnlineFeatures(GetOnlineFeaturesRequest requ
121120
featureValuesMap.values().stream()
122121
.map(m -> FieldValues.newBuilder().putAllFields(m).build())
123122
.collect(Collectors.toList());
124-
getOnlineFeaturesTimer.observeDuration();
123+
requestLatency.labels("getOnlineFeatures").observe(System.currentTimeMillis() - startTime);
125124
return getOnlineFeaturesResponseBuilder.addAllFieldValues(fieldValues).build();
126125
}
127126
}
@@ -197,7 +196,7 @@ private void sendAndProcessMultiGet(
197196
throws InvalidProtocolBufferException {
198197

199198
List<byte[]> jedisResps = sendMultiGet(redisKeys);
200-
Timer processResponseTimer = requestLatency.labels("processResponse").startTimer();
199+
long startTime = System.currentTimeMillis();
201200
try (Scope scope = tracer.buildSpan("Redis-processResponse").startActive(true)) {
202201
String featureSetId =
203202
String.format("%s:%d", featureSetRequest.getName(), featureSetRequest.getVersion());
@@ -234,7 +233,7 @@ private void sendAndProcessMultiGet(
234233
.forEach(f -> featureValues.put(featureSetId + ":" + f.getName(), f.getValue()));
235234
}
236235
} finally {
237-
processResponseTimer.observeDuration();
236+
requestLatency.labels("processResponse").observe(System.currentTimeMillis() - startTime);
238237
}
239238
}
240239

@@ -259,7 +258,7 @@ private boolean isStale(
259258
*/
260259
private List<byte[]> sendMultiGet(List<RedisKey> keys) {
261260
try (Scope scope = tracer.buildSpan("Redis-sendMultiGet").startActive(true)) {
262-
Timer sendMultiGetTimer = requestLatency.labels("sendMultiGet").startTimer();
261+
long startTime = System.currentTimeMillis();
263262
try (Jedis jedis = jedisPool.getResource()) {
264263
byte[][] binaryKeys =
265264
keys.stream()
@@ -273,7 +272,7 @@ private List<byte[]> sendMultiGet(List<RedisKey> keys) {
273272
.withCause(e)
274273
.asRuntimeException();
275274
} finally {
276-
sendMultiGetTimer.observeDuration();
275+
requestLatency.labels("sendMultiGet").observe(System.currentTimeMillis() - startTime);
277276
}
278277
}
279278
}

0 commit comments

Comments
 (0)