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

merge fixups, NBIO refactorings, UI stubs

Browse Source
This commit is contained in:
Jonathan Shook
2020-04-09 12:44:31 -05:00
parent 7fa5ec9ebd 510a9dd354
commit 5e8c5bdfca
83 changed files with 1269 additions and 817 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
engine-core/src/main/java/io/nosqlbench/engine/core/services/WorkloadFinderEndpoint.java Normal file
View File

@@ -0,0 +1,58 @@
package io.nosqlbench.engine.core.services;
import io.nosqlbench.docsys.api.WebServiceObject;
import io.nosqlbench.engine.api.scenarios.NBCLIScenarioParser;
import io.nosqlbench.engine.api.scenarios.WorkloadDesc;
import io.nosqlbench.nb.api.annotations.Service;
import io.nosqlbench.virtdata.userlibs.apps.docsapp.AutoDocsWebService;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import javax.inject.Singleton;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.MediaType;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Service(WebServiceObject.class)
@Singleton
@Path("/services/nb/")
public class WorkloadFinderEndpoint implements WebServiceObject {
private final static Logger logger = LogManager.getLogger(AutoDocsWebService.class);
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("enabled")
public boolean isEnabled() {
return true;
}
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("workloads")
public List<String> getWorkloadNames() {
List<WorkloadDesc> workloads = NBCLIScenarioParser.getWorkloadsWithScenarioScripts();
return workloads.stream().map(x -> x.getWorkloadName()).collect(Collectors.toList());
}
@GET
@Produces(MediaType.APPLICATION_JSON)
@Path("parameters")
public Map<String, String> getParametersByWorkload(@QueryParam("workloadName") String workloadName) {
List<WorkloadDesc> workloads = NBCLIScenarioParser.getWorkloadsWithScenarioScripts();
Map<String, String> templates = null;
templates = workloads.stream()
.filter(workload -> workload.getWorkloadName().equals(workloadName))
.map(workload -> workload.getTemplates())
.collect(Collectors.toSet()).iterator().next();
return templates;
}
}