IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

inject adapter context into spaces

Browse Source
This commit is contained in:
Jonathan Shook
2024-10-30 13:09:08 -05:00
parent b969f6927c
commit 919a741fa8
37 changed files with 73 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nb-adapters/adapter-azure-aisearch/src/main/java/io/nosqlbench/adapter/azureaisearch/AzureAISearchDriverAdapter.java
View File

@@ -45,7 +45,7 @@ public class AzureAISearchDriverAdapter extends BaseDriverAdapter<AzureAISearchB
@Override
public LongFunction<AzureAISearchSpace> getSpaceInitializer(NBConfiguration cfg) {
return (s) -> new AzureAISearchSpace(s, cfg);
return (s) -> new AzureAISearchSpace(this,s, cfg);
}
@Override

6
nb-adapters/adapter-azure-aisearch/src/main/java/io/nosqlbench/adapter/azureaisearch/AzureAISearchSpace.java
View File

@@ -51,7 +51,7 @@ import io.nosqlbench.nb.api.config.standard.Param;
* "https://github.com/Azure/azure-sdk-for-java/tree/main/sdk/search/azure-search-documents/">Azure
* AI Search Java searchIndexClient</a>
*/
public class AzureAISearchSpace extends BaseSpace {
public class AzureAISearchSpace extends BaseSpace<AzureAISearchSpace> {
private final static Logger logger = LogManager.getLogger(AzureAISearchSpace.class);
private final NBConfiguration cfg;
@@ -65,8 +65,8 @@ public class AzureAISearchSpace extends BaseSpace {
* @param idx The name of this space
* @param cfg The configuration ({@link NBConfiguration}) for this nb run
*/
public AzureAISearchSpace(long idx, NBConfiguration cfg) {
super(idx);
public AzureAISearchSpace(AzureAISearchDriverAdapter adapter, long idx, NBConfiguration cfg) {
super(adapter,idx);
this.cfg = cfg;
}