diff options
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; } |