com.couchbase.client.java.view.ViewQueryResponseMapper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-client Show documentation
Show all versions of java-client Show documentation
The official Couchbase Java SDK
/**
* Copyright (C) 2014 Couchbase, Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALING
* IN THE SOFTWARE.
*/
package com.couchbase.client.java.view;
import com.couchbase.client.core.annotations.InterfaceAudience;
import com.couchbase.client.core.annotations.InterfaceStability;
import com.couchbase.client.core.message.ResponseStatus;
import com.couchbase.client.core.message.view.ViewQueryResponse;
import com.couchbase.client.deps.io.netty.buffer.ByteBuf;
import com.couchbase.client.deps.io.netty.util.CharsetUtil;
import com.couchbase.client.java.AsyncBucket;
import com.couchbase.client.java.CouchbaseAsyncBucket;
import com.couchbase.client.java.document.Document;
import com.couchbase.client.java.document.json.JsonObject;
import com.couchbase.client.java.document.json.JsonValue;
import com.couchbase.client.java.error.TranscodingException;
import com.couchbase.client.java.error.ViewDoesNotExistException;
import com.couchbase.client.java.transcoder.JsonTranscoder;
import rx.Observable;
import rx.functions.Func1;
/**
* Encapsulates functionality required to map from a {@link ViewQueryResponse} into a {@link AsyncViewResult}.
*
* @author Michael Nitschinger
* @since 2.0.1
*/
@InterfaceStability.Committed
@InterfaceAudience.Private
public class ViewQueryResponseMapper {
/**
* The JSON transcoder used to convert to {@link JsonValue}s.
*/
private static final JsonTranscoder TRANSCODER = CouchbaseAsyncBucket.JSON_OBJECT_TRANSCODER;
/**
* Maps a raw {@link ViewQueryResponse} into a {@link AsyncViewResult}.
*
* @param bucket reference to the bucket.
* @param query the original query object.
* @param response the response from the server.
* @return a converted {@link AsyncViewResult}.
*/
public static Observable mapToViewResult(final AsyncBucket bucket, final ViewQuery query,
final ViewQueryResponse response) {
return response
.info()
.singleOrDefault(null)
.map(new ByteBufToJsonObject())
.map(new BuildViewResult(bucket, query, response));
}
/**
* Maps a raw {@link ViewQueryResponse} into a {@link AsyncSpatialViewResult}.
*
* @param bucket reference to the bucket.
* @param query the original query object.
* @param response the response from the server.
* @return a converted {@link AsyncSpatialViewResult}.
*/
public static Observable mapToSpatialViewResult(final AsyncBucket bucket,
final SpatialViewQuery query, final ViewQueryResponse response) {
return response
.info()
.singleOrDefault(null)
.map(new ByteBufToJsonObject())
.map(new BuildSpatialViewResult(bucket, query, response));
}
/**
* Function which takes a {@link ByteBuf} and converts it into a {@link JsonObject}.
*/
static class ByteBufToJsonObject implements Func1 {
@Override
public JsonObject call(final ByteBuf input) {
if (input == null || input.readableBytes() == 0) {
return JsonObject.empty();
}
try {
return TRANSCODER.byteBufToJsonObject(input);
} catch (Exception e) {
throw new TranscodingException("Could not decode View JSON: " + input.toString(CharsetUtil.UTF_8), e);
} finally {
if (input.refCnt() > 0) {
input.release();
}
}
}
}
/**
* Function which converts the {@link JsonObject} info into a {@link AsyncSpatialViewResult}.
*/
static class BuildSpatialViewResult implements Func1 {
private final AsyncBucket bucket;
private final SpatialViewQuery query;
private final ViewQueryResponse response;
BuildSpatialViewResult(AsyncBucket bucket, SpatialViewQuery query, ViewQueryResponse response) {
this.bucket = bucket;
this.query = query;
this.response = response;
}
@Override
public AsyncSpatialViewResult call(JsonObject jsonInfo) {
JsonObject debug = null;
boolean success = response.status().isSuccess();
if (success) {
debug = jsonInfo.getObject("debug_info");
} else if (response.status() == ResponseStatus.NOT_EXISTS) {
throw new ViewDoesNotExistException("View " + query.getDesign() + "/"
+ query.getView() + " does not exist.");
}
Observable rows = response
.rows()
.map(new ByteBufToJsonObject())
.flatMap(new Func1>() {
@Override
public Observable call(final JsonObject row) {
final String id = row.getString("id");
if (query.isIncludeDocs()) {
return bucket.get(id, query.includeDocsTarget()).map(new Func1, AsyncSpatialViewRow>() {
@Override
public AsyncSpatialViewRow call(Document document) {
return new DefaultAsyncSpatialViewRow(bucket, row.getString("id"), row.getArray("key"),
row.get("value"), row.getObject("geometry"), document);
}
});
} else {
return Observable.just((AsyncSpatialViewRow)
new DefaultAsyncSpatialViewRow(bucket, row.getString("id"), row.getArray("key"),
row.get("value"), row.getObject("geometry"), null)
);
}
}
});
Observable error = response
.error()
.map(new Func1() {
@Override
public JsonObject call(String input) {
try {
return TRANSCODER.stringToJsonObject(input);
} catch (Exception e) {
throw new TranscodingException("Could not decode View JSON: " + input, e);
}
}
});
return new DefaultAsyncSpatialViewResult(rows, success, error, debug);
}
}
/**
* Function which converts the {@link JsonObject} info into a {@link AsyncViewResult}.
*/
static class BuildViewResult implements Func1 {
private final AsyncBucket bucket;
private final ViewQuery query;
private final ViewQueryResponse response;
BuildViewResult(AsyncBucket bucket, ViewQuery query, ViewQueryResponse response) {
this.bucket = bucket;
this.query = query;
this.response = response;
}
@Override
public AsyncViewResult call(final JsonObject jsonInfo) {
JsonObject debug = null;
int totalRows = 0;
boolean success = response.status().isSuccess();
if (success) {
debug = jsonInfo.getObject("debug_info");
Integer trows = jsonInfo.getInt("total_rows");
if (trows != null) {
totalRows = trows;
}
} else if (response.status() == ResponseStatus.NOT_EXISTS) {
throw new ViewDoesNotExistException("View " + query.getDesign() + "/"
+ query.getView() + " does not exist.");
}
Observable rows = response
.rows()
.map(new ByteBufToJsonObject())
.compose(new Observable.Transformer() {
@Override
public Observable call(Observable observable) {
if (!query.isIncludeDocs()) {
return observable.concatMap(buildAsyncViewRow());
} else if (query.isOrderRetained()) {
return observable.concatMapEager(buildAsyncViewRow());
} else {
return observable.flatMap(buildAsyncViewRow());
}
}
});
Observable error = response
.error()
.map(new Func1() {
@Override
public JsonObject call(String input) {
try {
return TRANSCODER.stringToJsonObject(input);
} catch (Exception e) {
throw new TranscodingException("Could not decode View JSON: " + input, e);
}
}
});
return new DefaultAsyncViewResult(rows, totalRows, success, error, debug);
}
private Func1> buildAsyncViewRow() {
return new Func1>() {
@Override
public Observable call(final JsonObject row) {
final String id = row.getString("id");
if (query.isIncludeDocs()) {
return bucket.get(id, query.includeDocsTarget()).map(new Func1, AsyncViewRow>() {
@Override
public AsyncViewRow call(Document document) {
return new DefaultAsyncViewRow(bucket, id, row.get("key"), row.get("value"), document);
}
});
} else {
return Observable.just((AsyncViewRow)
new DefaultAsyncViewRow(bucket, id, row.get("key"), row.get("value"), null)
);
}
}
};
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy