IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-04 03:40:40 -06:00
Code Issues Packages Projects Releases Wiki Activity

phase 1 of openapi testing endpoints

Browse Source
This commit is contained in:
Jonathan Shook 2020-09-11 15:44:43 -05:00
parent 0af89f1679
commit 13240cd828
3 changed files with 1983 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
engine-rest/pom.xml
View File

@ -1,4 +1,6 @@
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
@ -8,30 +10,54 @@
<relativePath>../mvn-defaults</relativePath>
</parent>
<artifactId>engine-rest</artifactId>
<packaging>jar</packaging>
<name>${project.artifactId}</name>
<description>REST services for nosqlbench</description>
<artifactId>engine-rest</artifactId>
<packaging>jar</packaging>
<name>${project.artifactId}</name>
<description>REST services for nosqlbench</description>
<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<javadoc.name>nosqlbench REST Services</javadoc.name>
</properties>
<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<javadoc.name>nosqlbench REST Services</javadoc.name>
</properties>
<dependencies>
<dependencies>
<dependency>
<groupId>io.nosqlbench</groupId>
<artifactId>engine-cli</artifactId>
<version>3.12.145-SNAPSHOT</version>
</dependency>
<dependency>
<groupId>io.swagger</groupId>
<artifactId>swagger-parser</artifactId>
<version>2.0.0-rc1</version>
</dependency>
<dependency>
<groupId>ch.qos.logback</groupId>
<artifactId>logback-classic</artifactId>
<version>1.2.3</version>
</dependency>
<dependency>
<groupId>io.swagger</groupId>
<artifactId>swagger-models</artifactId>
<version>2.0.0-rc2</version>
</dependency>
</dependencies>
<dependency>
<groupId>io.swagger</groupId>
<artifactId>swagger-core</artifactId>
<version>2.0.0-rc2</version>
</dependency>
<dependency>
<groupId>io.swagger</groupId>
<artifactId>swagger-parser-v3</artifactId>
<version>2.0.0-rc1</version>
</dependency>
<dependency>
<groupId>io.nosqlbench</groupId>
<artifactId>engine-cli</artifactId>
<version>3.12.145-SNAPSHOT</version>
</dependency>
<dependency>
<groupId>ch.qos.logback</groupId>
<artifactId>logback-classic</artifactId>
<version>1.2.3</version>
</dependency>
</dependencies>
</project>

147
engine-rest/src/main/java/io/nosqlbench/engine/rest/resources/OpenApiEndpoint.java Normal file
View File

@ -0,0 +1,147 @@
package io.nosqlbench.engine.rest.resources;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import io.nosqlbench.docsys.api.WebServiceObject;
import io.nosqlbench.nb.annotations.Service;
import io.nosqlbench.nb.api.content.Content;
import io.nosqlbench.nb.api.content.NBIO;
import io.swagger.oas.models.OpenAPI;
import io.swagger.oas.models.Operation;
import io.swagger.oas.models.PathItem;
import io.swagger.oas.models.Paths;
import io.swagger.parser.OpenAPIParser;
import io.swagger.parser.models.ParseOptions;
import io.swagger.parser.models.SwaggerParseResult;
import io.swagger.parser.v2.SwaggerConverter;
import io.swagger.util.Json;
import javax.inject.Singleton;
import javax.ws.rs.*;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
@Service(WebServiceObject.class)
@Singleton
@Path("/openapi")
public class OpenApiEndpoint implements WebServiceObject {
private final OpenAPIParser parser = new OpenAPIParser();
private final SwaggerConverter converter = new SwaggerConverter();
private final Gson gson = new GsonBuilder().setPrettyPrinting().create();
private final Json sjson = new Json();
@POST
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
@Path("generate")
public Response putWorkload(String input) {
try {
return Response
.ok("received " + input.length() + " length request. Phase 2 implemention on the way...")
.build();
} catch (Exception e) {
return Response
.serverError()
.entity(e.getMessage())
.build();
}
}
@GET
@Path("paths")
@Produces(MediaType.APPLICATION_JSON)
public Response listPaths(@QueryParam("filepath") String filepath) {
try {
if (filepath == null) {
filepath = "stargate.yaml";
} else if (!filepath.endsWith(".yaml")) {
throw new RuntimeException("Only .yaml filepaths are supported for now.");
}
Content<?> one = NBIO.all().name(filepath).one();
String content = one.asString();
SwaggerParseResult parsed = parser.readContents(content, List.of(), new ParseOptions());
List<String> messages = parsed.getMessages();
if (messages.size() > 0) {
return Response.serverError().entity(String.join("\n", messages.toArray(new String[0]))).build();
}
OpenAPI openAPI = parsed.getOpenAPI();
Paths paths = openAPI.getPaths();
LinkedHashMap<String, Object> map = new LinkedHashMap<>();
for (String pathName : paths.keySet()) {
PathItem pathItem = paths.get(pathName);
if (pathItem.getGet() != null) {
Operation op = pathItem.getGet();
map.put("GET " + pathName, map("GET", pathName, op));
}
if (pathItem.getPost() != null) {
Operation op = pathItem.getPost();
map.put("POST " + pathName, map("POST", pathName, op));
}
if (pathItem.getPut() != null) {
Operation op = pathItem.getPut();
map.put("PUT " + pathName, map("PUT", pathName, op));
}
if (pathItem.getPatch() != null) {
Operation op = pathItem.getPatch();
map.put("PATCH " + pathName, map("PATCH", pathName, op));
}
if (pathItem.getDelete() != null) {
Operation op = pathItem.getDelete();
map.put("DELETE " + pathName, map("DELETE", pathName, op));
}
if (pathItem.getHead() != null) {
Operation op = pathItem.getHead();
map.put("HEAD " + pathName, map("HEAD", pathName, op));
}
if (pathItem.getOptions() != null) {
Operation op = pathItem.getOptions();
map.put("OPTIONS " + pathName, map("OPTIONS", pathName, op));
}
}
return Response.ok(Json.pretty(map)).build();
} catch (Exception e) {
return Response.serverError().entity(e.getMessage()).build();
}
}
private Map<String, Object> map(String method, String pathName, Operation op) {
LinkedHashMap<String, Object> map = new LinkedHashMap<>();
map.put("method", method);
map.put("path", pathName);
map.put("api", op);
map.put("summary", op.getSummary() != null ? op.getSummary() : "");
map.put("description", op.getDescription() != null ? op.getDescription() : "");
return map;
}
// private String toJson(String method, String path, Operation op) {
// try {
// LinkedHashMap<String,Object> map = new LinkedHashMap<>();
// map.put("id", method.toUpperCase() + " " + path);
// map.put("path", path);
// map.put("method", method);
// map.put("description", op.getDescription());
// map.put("summary",op.getSummary());
// map.put("body",op.getRequestBody());
// map.put("parameters",op.getParameters());
// map.put("operation_id",op.getOperationId());
// map.put("external_docs",op.getExternalDocs());
//// return gson.toJson(map);
//
// } catch (Exception e) {
// throw new RuntimeException(e);
// }
// }
}

1789
engine-rest/src/main/resources/stargate.yaml Normal file
View File

File diff suppressed because it is too large Load Diff