diff options
author | Jon Marius Venstad <venstad@gmail.com> | 2021-06-09 21:15:58 +0200 |
---|---|---|
committer | Jon Marius Venstad <venstad@gmail.com> | 2021-06-10 07:50:42 +0200 |
commit | 5be6949b581dc9c4b78c8d270a1868eae2bdda0b (patch) | |
tree | caac02229918df118b83445dab3b578c4fed7f88 /vespa-feed-client | |
parent | 64324d85537dc7714150032a64ae50a5f3e0e702 (diff) |
Use stats aggregated by client
Diffstat (limited to 'vespa-feed-client')
-rw-r--r-- | vespa-feed-client/src/main/java/ai/vespa/feed/client/JsonFeeder.java | 10 | ||||
-rw-r--r-- | vespa-feed-client/src/main/java/ai/vespa/feed/client/OperationStats.java | 4 |
2 files changed, 11 insertions, 3 deletions
diff --git a/vespa-feed-client/src/main/java/ai/vespa/feed/client/JsonFeeder.java b/vespa-feed-client/src/main/java/ai/vespa/feed/client/JsonFeeder.java index 2a6d2e15747..de32e7abdf5 100644 --- a/vespa-feed-client/src/main/java/ai/vespa/feed/client/JsonFeeder.java +++ b/vespa-feed-client/src/main/java/ai/vespa/feed/client/JsonFeeder.java @@ -58,18 +58,18 @@ public class JsonFeeder implements Closeable { * @param result Non-null if operation completed successfully * @param error Non-null if operation failed */ - void onNextResult(Result result, Throwable error); + default void onNextResult(Result result, Throwable error) { } /** * Invoked if an unrecoverable error occurred during feed processing, * after which no other {@link ResultCallback} methods are invoked. */ - void onError(Throwable error); + default void onError(Throwable error) { } /** * Invoked when all feed operations are either completed successfully or failed. */ - void onComplete(); + default void onComplete() { } } public static Builder builder(FeedClient client) { return new Builder(client); } @@ -103,6 +103,10 @@ public class JsonFeeder implements Closeable { return feedMany(jsonStream, 1 << 26, resultCallback); } + public CompletableFuture<Void> feedMany(InputStream jsonStream) { + return feedMany(jsonStream, new ResultCallback() { }); + } + CompletableFuture<Void> feedMany(InputStream jsonStream, int size, ResultCallback resultCallback) { RingBufferStream buffer = new RingBufferStream(jsonStream, size); CompletableFuture<Void> overallResult = new CompletableFuture<>(); diff --git a/vespa-feed-client/src/main/java/ai/vespa/feed/client/OperationStats.java b/vespa-feed-client/src/main/java/ai/vespa/feed/client/OperationStats.java index 9305709d873..d36475a51fb 100644 --- a/vespa-feed-client/src/main/java/ai/vespa/feed/client/OperationStats.java +++ b/vespa-feed-client/src/main/java/ai/vespa/feed/client/OperationStats.java @@ -42,6 +42,10 @@ public class OperationStats { return requests - inflight; } + public long successes() { + return responsesByCode.getOrDefault(200, 0L); + } + public Map<Integer, Long> responsesByCode() { return responsesByCode; } |