com.redhat.lightblue.Response Maven / Gradle / Ivy
/*
Copyright 2013 Red Hat, Inc. and/or its affiliates.
This file is part of lightblue.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see resultMetadata;
private String hostname;
private final List dataErrors = new ArrayList<>();
private final List errors = new ArrayList<>();
private final JsonNodeFactory jsonNodeFactory;
private static final String HOSTNAME;
static {
String hostName = "unknown";
try {
InetAddress localHost = InetAddress.getLocalHost();
if (localHost != null) {
hostName = localHost.getHostName();
}
} catch (UnknownHostException e) {
}
HOSTNAME = hostName;
}
/**
* @deprecated use Response(JsonNodeFactory)
*/
@Deprecated
public Response() {
jsonNodeFactory = JsonNodeFactory.withExactBigDecimals(true);
}
public Response(JsonNodeFactory jsonNodeFactory) {
this.jsonNodeFactory = jsonNodeFactory;
this.hostname = HOSTNAME;
}
public EntityVersion getEntity() {
return entity;
}
public void setEntity(EntityVersion e) {
entity=e;
}
public void setEntity(String entityName,String version) {
this.entity=new EntityVersion(entityName,version);
}
/**
* Status of the completed operation
*/
public OperationStatus getStatus() {
return status;
}
/**
* Status of the completed operation
*/
public void setStatus(OperationStatus s) {
status = s;
}
public String getHostname() {
return hostname;
}
public void setHostname(String hostname) {
this.hostname = hostname;
}
/**
* Number of documents inserted/updated/deleted
*/
public long getModifiedCount() {
return modifiedCount;
}
/**
* Number of documents inserted/updated/deleted
*/
public void setModifiedCount(long l) {
modifiedCount = l;
}
/**
* Number of documents that matched the search cResponseriteria
*/
public long getMatchCount() {
return matchCount;
}
/**
* Number of documents that matched the search criteria
*/
public void setMatchCount(long l) {
matchCount = l;
}
/**
* If the operation continues asynchronously, the task handle can be used to
* retrieve status information, and the result of the call once the
* operation is complete
*/
public String getTaskHandle() {
return taskHandle;
}
/**
* If the operation continues asynchronously, the task handle can be used to
* retrieve status information, and the result of the call once the
* operation is complete
*/
public void setTaskHandle(String t) {
taskHandle = t;
}
/**
* If the operation starts a session or uses an existing session, the
* session information
*/
public SessionInfo getSessionInfo() {
return session;
}
/**
* If the operation starts a session or uses an existing session, the
* session information
*/
public void setSessionInfo(SessionInfo s) {
session = s;
}
/**
* Returns the entity data resulting from the call.
*/
public JsonNode getEntityData() {
return entityData;
}
/**
* Returns the entity data resulting from the call.
*/
public void setEntityData(JsonNode node) {
// if the node is not an array then wrap it in an array
if (node != null && !node.isArray()) {
ArrayNode arrayNode = new ArrayNode(jsonNodeFactory);
arrayNode.add(node);
entityData = arrayNode;
} else {
entityData = node;
}
}
/**
* Adds a document, or array of documents to the entity data
*/
public void addEntityData(JsonNode doc) {
if(doc!=null) {
if(entityData==null) {
entityData=JsonNodeFactory.instance.arrayNode();
}
if(doc instanceof ArrayNode) {
for(Iterator itr=doc.elements();itr.hasNext();) {
((ArrayNode)entityData).add(itr.next());
}
} else {
((ArrayNode)entityData).add(doc);
}
}
}
/**
* Metadata list for documents in entityData. If there are more
* than one documents, the entitydata and metadata indexes match.
*/
public List getResultMetadata() {
return resultMetadata;
}
public void setResultMetadata(List l) {
resultMetadata=l;
}
/**
* Errors related to each document
*/
public List getDataErrors() {
return dataErrors;
}
/**
* Errors related to the operation
*/
public List getErrors() {
return errors;
}
/**
* Returns JSON representation of this
*/
@Override
public JsonNode toJson() {
JsonNodeBuilder builder = new JsonNodeBuilder();
if(entity!=null) {
builder.add(PROPERTY_ENTITY,entity.getEntity());
builder.add(PROPERTY_VERSION,entity.getVersion());
}
builder.add(PROPERTY_STATUS, status);
builder.add(PROPERTY_MOD_COUNT, modifiedCount);
builder.add(PROPERTY_MATCH_COUNT, matchCount);
builder.add(PROPERTY_TASK_HANDLE, taskHandle);
builder.add(PROPERTY_SESSION, session);
builder.add(PROPERTY_PROCESSED, entityData);
builder.add(PROPERTY_HOSTNAME, HOSTNAME);
builder.addJsonObjectsList(PROPERTY_DATA_ERRORS, dataErrors);
builder.addErrorsList(PROPERTY_ERRORS, errors);
if(resultMetadata!=null)
builder.addJsonObjectsList(PROPERTY_RESULT_METADATA,resultMetadata);
return builder.build();
}
// This class is not used
@Deprecated
public static class ResponseBuilder {
private OperationStatus status;
private long modifiedCount;
private long matchCount;
private String taskHandle;
private SessionInfo session;
private JsonNode entityData;
private String hostname;
private List dataErrors = new ArrayList<>();
private List errors = new ArrayList<>();
private final JsonNodeFactory jsonNodeFactory;
public ResponseBuilder(JsonNodeFactory jsonNodeFactory) {
this.jsonNodeFactory = jsonNodeFactory;
}
public ResponseBuilder(Response response) {
status = response.getStatus();
modifiedCount = response.getModifiedCount();
matchCount = response.getMatchCount();
hostname = response.getHostname();
taskHandle = response.getTaskHandle();
session = response.getSessionInfo();
entityData = response.getEntityData();
dataErrors = response.getDataErrors();
errors = response.getErrors();
jsonNodeFactory = response.jsonNodeFactory;
}
public ResponseBuilder withHostname(JsonNode node) {
if (node != null) {
hostname = node.asText();
}
return this;
}
public ResponseBuilder withStatus(JsonNode node) {
if (node != null) {
try {
status = OperationStatus.valueOf(node.asText().toUpperCase());
} catch (IllegalArgumentException e) {
status = OperationStatus.ERROR;
}
}
return this;
}
public ResponseBuilder withModifiedCount(JsonNode node) {
if (node != null) {
modifiedCount = node.asLong();
}
return this;
}
public ResponseBuilder withMatchCount(JsonNode node) {
if (node != null) {
matchCount = node.asLong();
}
return this;
}
public ResponseBuilder withTaskHandle(JsonNode node) {
if (node != null) {
taskHandle = node.asText();
}
return this;
}
public ResponseBuilder withSession(JsonNode node) {
//TODO
return this;
}
public ResponseBuilder withEntityData(JsonNode node) {
if (node != null) {
entityData = node;
}
return this;
}
public ResponseBuilder withDataErrors(JsonNode node) {
if (node instanceof ArrayNode) {
for (Iterator itr = ((ArrayNode) node).elements();
itr.hasNext();) {
dataErrors.add(DataError.fromJson((ObjectNode) itr.next()));
}
}
return this;
}
public ResponseBuilder withErrors(JsonNode node) {
if (node instanceof ArrayNode) {
for (Iterator itr = ((ArrayNode) node).elements();
itr.hasNext();) {
errors.add(Error.fromJson(itr.next()));
}
}
return this;
}
public Response buildResponse() {
Response response = new Response(jsonNodeFactory);
response.setStatus(status);
response.setModifiedCount(modifiedCount);
response.setMatchCount(matchCount);
response.setTaskHandle(taskHandle);
response.setSessionInfo(session);
response.setEntityData(entityData);
response.getDataErrors().addAll(dataErrors);
response.getErrors().addAll(errors);
return response;
}
}
}