From cb02fd9f01b03603a1712bc9a55224301e4a8b96 Mon Sep 17 00:00:00 2001 From: Jonathan Shook Date: Wed, 11 Sep 2024 15:34:01 -0500 Subject: [PATCH] wrap checked exception --- .../opsdispensers/WeaviateBaseOpDispenser.java | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/nb-adapters/adapter-weaviate/src/main/java/io/nosqlbench/adapter/weaviate/opsdispensers/WeaviateBaseOpDispenser.java b/nb-adapters/adapter-weaviate/src/main/java/io/nosqlbench/adapter/weaviate/opsdispensers/WeaviateBaseOpDispenser.java index 250a0b8de..69f75e1cc 100644 --- a/nb-adapters/adapter-weaviate/src/main/java/io/nosqlbench/adapter/weaviate/opsdispensers/WeaviateBaseOpDispenser.java +++ b/nb-adapters/adapter-weaviate/src/main/java/io/nosqlbench/adapter/weaviate/opsdispensers/WeaviateBaseOpDispenser.java @@ -2,13 +2,13 @@ package io.nosqlbench.adapter.weaviate.opsdispensers; /* * Copyright (c) 2022 nosqlbench - * + * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at - * + * * http://www.apache.org/licenses/LICENSE-2.0 - * + * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY @@ -27,6 +27,7 @@ import io.nosqlbench.adapters.api.activityimpl.BaseOpDispenser; import io.nosqlbench.adapters.api.activityimpl.uniform.DriverAdapter; import io.nosqlbench.adapters.api.templating.ParsedOp; import io.weaviate.client.WeaviateClient; +import io.weaviate.client.v1.auth.exception.AuthException; public abstract class WeaviateBaseOpDispenser extends BaseOpDispenser, WeaviateSpace> { @@ -38,7 +39,13 @@ public abstract class WeaviateBaseOpDispenser extends BaseOpDispenser targetF) { super((DriverAdapter) adapter, op); this.weaviateSpaceFunction = adapter.getSpaceFunc(op); - this.clientFunction = (long l) -> this.weaviateSpaceFunction.apply(l).getClient(); + this.clientFunction = (long l) -> { + try { + return this.weaviateSpaceFunction.apply(l).getClient(); + } catch (AuthException e) { + throw new RuntimeException(e); + } + }; this.paramF = getParamFunc(this.clientFunction, op, targetF); this.opF = createOpFunc(paramF, this.clientFunction, op, targetF); }