mjsax commented on code in PR #14570: URL: https://github.com/apache/kafka/pull/14570#discussion_r1382305305
########## streams/src/main/java/org/apache/kafka/streams/state/internals/MeteredTimestampedKeyValueStore.java: ########## @@ -102,4 +139,232 @@ static class RawAndDeserializedValue<ValueType> { this.value = value; } } + + @SuppressWarnings("unchecked") + @Override + public <R> QueryResult<R> query(final Query<R> query, + final PositionBound positionBound, + final QueryConfig config) { + + final long start = time.nanoseconds(); + final QueryResult<R> result; + + final StoreQueryUtils.QueryHandler handler = queryHandlers.get(query.getClass()); + if (handler == null) { + result = wrapped().query(query, positionBound, config); + if (config.isCollectExecutionInfo()) { + result.addExecutionInfo( + "Handled in " + getClass() + " in " + (time.nanoseconds() - start) + "ns"); + } + } else { + result = (QueryResult<R>) handler.apply( + query, + positionBound, + config, + this + ); + if (config.isCollectExecutionInfo()) { + result.addExecutionInfo( + "Handled in " + getClass() + " with serdes " + + serdes + " in " + (time.nanoseconds() - start) + "ns"); + } + } + return result; + } + + + + @SuppressWarnings("unchecked") + protected <R> QueryResult<R> runTimestampedKeyQuery(final Query<R> query, Review Comment: Why is it `protected` in `MeteredKeyValueStore` -- I believe it should be `private` there, too? Can you try it out? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org