Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.weaverplatform.sdk.Weaver Maven / Gradle / Ivy
package com.weaverplatform.sdk;
import com.google.gson.Gson;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.reflect.TypeToken;
import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.WebResource;
import com.sun.jersey.api.client.config.ClientConfig;
import com.sun.jersey.api.client.config.DefaultClientConfig;
import com.sun.jersey.core.header.FormDataContentDisposition;
import com.sun.jersey.multipart.FormDataMultiPart;
import com.sun.jersey.multipart.MultiPart;
import com.sun.jersey.multipart.file.StreamDataBodyPart;
import com.sun.jersey.multipart.impl.MultiPartWriter;
import com.weaverplatform.CliOptions;
import javax.ws.rs.core.MediaType;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.List;
/**
* @author bastbijl, Sysunite 2017
*/
public class Weaver {
public static int MAX_OPERATIONS = 5000;
private static enum OP { get, post };
private CliOptions options;
public Weaver() {
this.options = new CliOptions(new String[0]);
}
public Weaver(CliOptions options) {
this.options = options;
}
public String getUri() {
return uri;
}
public void setUri(String uri) {
this.uri = uri;
}
public String getUsername() {
return username;
}
public void setUsername(String username) {
this.username = username;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public String getProject() {
return project;
}
public void setProject(String project) {
this.project = project;
}
private String uri = "http://localhost:9487";
private String username = "admin";
private String password = "admin";
private String project = null;
private String authToken = null;
private Gson converter = new Gson();
private Client client;
private Client clientMultiPart;
private Client getClient() {
if(client == null) {
client = Client.create();
}
return client;
}
private Client getClientMultiPart() {
if(clientMultiPart == null) {
ClientConfig cc = new DefaultClientConfig();
cc.getClasses().add(MultiPartWriter.class);
clientMultiPart = Client.create(cc);
}
return clientMultiPart;
}
public String login() {
if(authToken != null) {
return authToken;
} else {
// Sign in
JsonObject signIn = new JsonObject();
signIn.addProperty("username", username);
signIn.addProperty("password", password);
ClientResponse response = send(signIn, "/user/signInUsername", OP.post);
authToken = response.getEntity(String.class);
return authToken;
}
}
public void setAuthToken(String authToken) {
this.authToken = authToken;
}
public ClientResponse sendProject(JsonObject payload, String target, OP op) {
payload.addProperty("target", project);
return sendAuthenticated(payload, target, op);
}
public ClientResponse sendAuthenticated(JsonObject payload, String target, OP op) {
String authToken = login();
payload.addProperty("authToken", authToken);
return send(payload, target, op);
}
public ClientResponse send(JsonObject payload, String target, OP op) {
Client client = getClient();
WebResource webResource = client.resource(uri + target);
ClientResponse soapResponse;
switch (op) {
case get:
try {
String json = URLEncoder.encode(converter.toJson(payload), "UTF-8");
webResource = webResource.queryParam("payload", json);
} catch (UnsupportedEncodingException e) {
throw new RuntimeException(e);
}
soapResponse = webResource.type("application/json; charset=utf-8").get(ClientResponse.class);
break;
case post:
default:
soapResponse = webResource.type("application/json; charset=utf-8").post(ClientResponse.class, converter.toJson(payload));
}
// Send response
if (soapResponse.getStatus() != 200) {
String response = soapResponse.getEntity(String.class);
throw new RuntimeException("Message delivery failed on " + target + " response: " + response);
}
return soapResponse;
}
public ClientResponse sendMultipart(InputStream fileStream, String fileName, String target) {
Client client = getClientMultiPart();
String authToken = login();
WebResource webResource = client.resource(uri + target);
StreamDataBodyPart fileDataBodyPart = new StreamDataBodyPart("file", fileStream, fileName, MediaType.APPLICATION_OCTET_STREAM_TYPE);
fileDataBodyPart.setContentDisposition(FormDataContentDisposition.name("file").fileName(fileName).build());
final MultiPart multiPart = new FormDataMultiPart()
.field("authToken", authToken, MediaType.TEXT_PLAIN_TYPE)
.field("target", project, MediaType.TEXT_PLAIN_TYPE)
.field("filename", fileName, MediaType.TEXT_PLAIN_TYPE)
.bodyPart(fileDataBodyPart);
multiPart.setMediaType(MediaType.MULTIPART_FORM_DATA_TYPE);
return webResource.type("multipart/form-data").post(ClientResponse.class, multiPart);
}
public void createProject(String id) {
JsonObject payload = new JsonObject();
payload.addProperty("id", id);
payload.addProperty("name", id);
sendAuthenticated(payload, "/project/create", OP.post);
}
public String projectReady(String id) {
JsonObject payload = new JsonObject();
payload.addProperty("id", id);
return sendAuthenticated(payload, "/project/ready", OP.post).getEntity(String.class);
}
public String getVersion() {
JsonObject payload = new JsonObject();
return sendProject(payload, "/application/version", OP.get).getEntity(String.class);
}
public void wipe() {
JsonObject payload = new JsonObject();
sendProject(payload, "/project/wipe", OP.post);
}
public void sendCreate(JsonArray operations, boolean quick) {
JsonArray array = new JsonArray();
boolean stop = false;
while(array.size() < MAX_OPERATIONS && operations.size() > 0 && !stop) {
JsonObject peek = operations.get(0).getAsJsonObject();
if(peek.get("action").getAsString().equals("increment-attribute")) {
stop = true;
if(array.size() == 0) {
array.add(operations.remove(0));
}
} else {
array.add(operations.remove(0));
}
}
if(operations.size() > 0) {
System.out.println("will send " + array.size() + " (" + operations.size() + " to go)");
} else {
System.out.println("will send " + array.size());
}
reallySendCreate(array, quick);
if(operations.size() > 0) {
sendCreate(operations, quick);
}
}
public void reallySendCreate(JsonArray operations, boolean quick) {
JsonObject payload = new JsonObject();
payload.add("operations", operations);
final String target = "/write";
sendProject(payload, target, OP.post);
}
public JsonObject createNode(String id) {
JsonObject payload = new JsonObject();
payload.addProperty("timestamp", System.currentTimeMillis());
payload.addProperty("action", "create-node");
payload.addProperty("id", id);
return payload;
}
public JsonObject createAttribute(String id, String key, String value) {
JsonObject payload = new JsonObject();
payload.addProperty("timestamp", System.currentTimeMillis());
payload.addProperty("action", "create-attribute");
payload.addProperty("id", id);
payload.addProperty("key", key);
payload.addProperty("value", value);
return payload;
}
public JsonObject createRelation(String from, String key, String to, String id) {
JsonObject payload = new JsonObject();
payload.addProperty("timestamp", System.currentTimeMillis());
payload.addProperty("action", "create-relation");
payload.addProperty("from", from);
payload.addProperty("key", key);
payload.addProperty("to", to);
payload.addProperty("id", id);
return payload;
}
public Project[] getProjects() {
ClientResponse res = sendAuthenticated(new JsonObject(), "/project", OP.get);
String s = res.getEntity(String.class);
return converter.fromJson(s, Project[].class);
}
public WeaverFile snapshotToStore(boolean zipped) {
JsonObject payload = new JsonObject();
payload.addProperty("json", false);
payload.addProperty("zipped", zipped);
payload.addProperty("stored", true);
ClientResponse soapResponse = sendProject(payload, "/snapshot", OP.get);
return converter.fromJson(soapResponse.getEntity(String.class), WeaverFile.class);
}
public InputStream getSnapshot(boolean zipped) {
JsonObject payload = new JsonObject();
payload.addProperty("json", false);
payload.addProperty("zipped", zipped);
payload.addProperty("stored", false);
ClientResponse soapResponse = sendProject(payload, "/snapshot", OP.get);
return soapResponse.getEntityInputStream();
}
public WeaverFile snapshotGraphToStore(List graphs, List toGraphs, List fromGraphs, boolean zipped) {
JsonObject payload = new JsonObject();
if(graphs != null && !graphs.isEmpty()) {
JsonElement array = converter.toJsonTree(graphs, new TypeToken>() {}.getType());
payload.add("graphs", array);
}
if(toGraphs != null && !toGraphs.isEmpty()) {
JsonElement array = converter.toJsonTree(toGraphs, new TypeToken>() {}.getType());
payload.add("toGraphs", array);
}
if(fromGraphs != null && !fromGraphs.isEmpty()) {
JsonElement array = converter.toJsonTree(fromGraphs, new TypeToken>() {}.getType());
payload.add("fromGraphs", array);
}
payload.addProperty("json", false);
payload.addProperty("zipped", zipped);
payload.addProperty("stored", true);
ClientResponse soapResponse = sendProject(payload, "/snapshotGraph", OP.get);
return converter.fromJson(soapResponse.getEntity(String.class), WeaverFile.class);
}
public InputStream getSnapshotGraph(List graphs, List toGraphs, List fromGraphs, boolean zipped) {
JsonObject payload = new JsonObject();
if(graphs != null && !graphs.isEmpty()) {
JsonElement array = converter.toJsonTree(graphs, new TypeToken>() {}.getType());
payload.add("graphs", array);
}
if(toGraphs != null && !toGraphs.isEmpty()) {
JsonElement array = converter.toJsonTree(toGraphs, new TypeToken>() {}.getType());
payload.add("toGraphs", array);
}
if(fromGraphs != null && !fromGraphs.isEmpty()) {
JsonElement array = converter.toJsonTree(fromGraphs, new TypeToken>() {}.getType());
payload.add("fromGraphs", array);
}
payload.addProperty("json", false);
payload.addProperty("zipped", zipped);
payload.addProperty("stored", false);
ClientResponse soapResponse = sendProject(payload, "/snapshotGraph", OP.get);
return soapResponse.getEntityInputStream();
}
public String getHistory() {
ClientResponse soapResponse = sendProject(new JsonObject(), "/history", OP.get);
return soapResponse.getEntity(String.class);
}
public void restore(InputStream operationsStream) {
JsonArray ops = converter.fromJson(new InputStreamReader(operationsStream), JsonArray.class);
System.out.println("Going to write " + ops.size() + " operations.");
sendCreate(ops, false);
}
public MinioFile[] listFiles() {
JsonObject payload = new JsonObject();
ClientResponse soapResponse = sendProject(payload, "/file/list", OP.get);
String s = soapResponse.getEntity(String.class);
return converter.fromJson(s, MinioFile[].class);
}
public WeaverFile uploadFile(InputStream fileStream, String fileName) {
ClientResponse soapResponse = sendMultipart(fileStream, fileName, "/file/upload");
return converter.fromJson(soapResponse.getEntity(String.class), WeaverFile.class);
}
public InputStream downloadFile(String id) throws IOException {
JsonObject payload = new JsonObject();
payload.addProperty("fileId", id);
ClientResponse soapResponse = sendProject(payload, "/file/download", OP.get);
return soapResponse.getEntityInputStream();
}
public String deleteFile(String id) {
JsonObject payload = new JsonObject();
payload.addProperty("fileId", id);
ClientResponse soapResponse = sendProject(payload, "/file/delete", OP.post);
return soapResponse.getEntity(String.class);
}
}