4646import io .grpc .Status ;
4747import io .opentracing .Scope ;
4848import io .opentracing .Tracer ;
49- import io .prometheus .client .Histogram .Timer ;
5049import java .util .List ;
5150import java .util .Map ;
5251import 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