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.marklogic.client.impl.QueryManagerImpl Maven / Gradle / Ivy
/*
* Copyright 2012-2015 MarkLogic Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.marklogic.client.impl;
import java.util.ArrayList;
import java.util.List;
import javax.xml.namespace.QName;
import org.w3c.dom.Document;
import org.w3c.dom.NodeList;
import com.marklogic.client.DatabaseClientFactory.HandleFactoryRegistry;
import com.marklogic.client.Transaction;
import com.marklogic.client.io.DOMHandle;
import com.marklogic.client.io.Format;
import com.marklogic.client.io.SearchHandle;
import com.marklogic.client.io.TuplesHandle;
import com.marklogic.client.io.ValuesHandle;
import com.marklogic.client.io.marker.ContentHandle;
import com.marklogic.client.io.marker.QueryOptionsListReadHandle;
import com.marklogic.client.io.marker.SearchReadHandle;
import com.marklogic.client.io.marker.StructureReadHandle;
import com.marklogic.client.io.marker.StructureWriteHandle;
import com.marklogic.client.io.marker.TuplesReadHandle;
import com.marklogic.client.io.marker.ValuesListReadHandle;
import com.marklogic.client.io.marker.ValuesReadHandle;
import com.marklogic.client.query.DeleteQueryDefinition;
import com.marklogic.client.query.ElementLocator;
import com.marklogic.client.query.KeyLocator;
import com.marklogic.client.query.KeyValueQueryDefinition;
import com.marklogic.client.query.MatchDocumentSummary;
import com.marklogic.client.query.QueryDefinition;
import com.marklogic.client.query.QueryManager;
import com.marklogic.client.query.RawCombinedQueryDefinition;
import com.marklogic.client.query.RawQueryByExampleDefinition;
import com.marklogic.client.query.RawStructuredQueryDefinition;
import com.marklogic.client.query.StringQueryDefinition;
import com.marklogic.client.query.StructuredQueryBuilder;
import com.marklogic.client.query.SuggestDefinition;
import com.marklogic.client.query.ValuesDefinition;
import com.marklogic.client.query.ValuesListDefinition;
public class QueryManagerImpl
extends AbstractLoggingManager
implements QueryManager
{
private RESTServices services;
private HandleFactoryRegistry handleRegistry;
private long pageLen = -1;
private QueryView view = QueryView.DEFAULT;
public QueryManagerImpl(RESTServices services) {
super();
this.services = services;
}
HandleFactoryRegistry getHandleRegistry() {
return handleRegistry;
}
void setHandleRegistry(HandleFactoryRegistry handleRegistry) {
this.handleRegistry = handleRegistry;
}
@Override
public long getPageLength() {
return pageLen;
}
@Override
public void setPageLength(long length) {
pageLen = length;
}
@Override
public QueryView getView() {
return view;
}
@Override
public void setView(QueryView view) {
this.view = (view == null) ? QueryView.DEFAULT : view;
}
@Override
public StringQueryDefinition newStringDefinition() {
return new StringQueryDefinitionImpl(null);
}
@Override
public StringQueryDefinition newStringDefinition(String optionsName) {
return new StringQueryDefinitionImpl(optionsName);
}
@Override
public KeyValueQueryDefinition newKeyValueDefinition() {
return new KeyValueQueryDefinitionImpl(null);
}
@Override
public KeyValueQueryDefinition newKeyValueDefinition(String optionsName) {
return new KeyValueQueryDefinitionImpl(optionsName);
}
@Override
public StructuredQueryBuilder newStructuredQueryBuilder() {
return new StructuredQueryBuilder();
}
@Override
public StructuredQueryBuilder newStructuredQueryBuilder(String optionsName) {
return new StructuredQueryBuilder(optionsName);
}
@Override
public DeleteQueryDefinition newDeleteDefinition() {
return new DeleteQueryDefinitionImpl();
}
@Override
public ElementLocator newElementLocator(QName element) {
return new ElementLocatorImpl(element);
}
@Override
public ElementLocator newElementLocator(QName element, QName attribute) {
return new ElementLocatorImpl(element, attribute);
}
@Override
public KeyLocator newKeyLocator(String key) {
return new KeyLocatorImpl(key);
}
@Override
public ValuesDefinition newValuesDefinition(String name) {
return new ValuesDefinitionImpl(name, null);
}
@Override
public ValuesDefinition newValuesDefinition(String name, String optionsName) {
return new ValuesDefinitionImpl(name, optionsName);
}
@Override
public ValuesListDefinition newValuesListDefinition() {
return new ValuesListDefinitionImpl(null);
}
@Override
public ValuesListDefinition newValuesListDefinition(String optionsName) {
return new ValuesListDefinitionImpl(optionsName);
}
@Override
public T search(QueryDefinition querydef, T searchHandle) {
return search(querydef, searchHandle, 1, null);
}
@Override
public T search(QueryDefinition querydef, T searchHandle, long start) {
return search(querydef, searchHandle, start, null);
}
@Override
public T search(QueryDefinition querydef, T searchHandle, Transaction transaction) {
return search(querydef, searchHandle, 1, transaction);
}
@Override
@SuppressWarnings("unchecked")
public T search(QueryDefinition querydef, T searchHandle, long start, Transaction transaction) {
@SuppressWarnings("rawtypes")
HandleImplementation searchBase = HandleAccessor.checkHandle(searchHandle, "search");
if (searchHandle instanceof SearchHandle) {
SearchHandle responseHandle = (SearchHandle) searchHandle;
responseHandle.setHandleRegistry(getHandleRegistry());
responseHandle.setQueryCriteria(querydef);
}
Format searchFormat = searchBase.getFormat();
switch(searchFormat) {
case UNKNOWN:
searchFormat = Format.XML;
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON search results are possible.");
}
String mimetype = searchFormat.getDefaultMimetype();
searchBase.receiveContent(services.search(requestLogger, searchBase.receiveAs(), querydef, mimetype, start, pageLen, view, transaction));
return searchHandle;
}
@Override
public void delete(DeleteQueryDefinition querydef) {
delete(querydef, null);
}
@Override
public void delete(DeleteQueryDefinition querydef, Transaction transaction) {
services.deleteSearch(requestLogger, querydef, transaction);
}
@Override
public T values(ValuesDefinition valdef, T valueHandle) {
return values(valdef, valueHandle, -1, null);
}
@Override
public T values(ValuesDefinition valdef, T valueHandle, long start) {
return values(valdef, valueHandle, start, null);
}
@Override
public T values(ValuesDefinition valdef, T valueHandle, Transaction transaction) {
return values(valdef, valueHandle, -1, transaction);
}
@Override
@SuppressWarnings("unchecked")
public T values(ValuesDefinition valdef, T valueHandle, long start, Transaction transaction) {
@SuppressWarnings("rawtypes")
HandleImplementation valuesBase = HandleAccessor.checkHandle(valueHandle, "values");
if (valueHandle instanceof ValuesHandle) {
((ValuesHandle) valueHandle).setQueryCriteria(valdef);
}
Format valuesFormat = valuesBase.getFormat();
switch(valuesFormat) {
case UNKNOWN:
valuesFormat = Format.XML;
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON values results are possible.");
}
String mimetype = valuesFormat.getDefaultMimetype();
long pageLength = (start == -1) ? -1 : getPageLength();
String tid = transaction == null ? null : transaction.getTransactionId();
valuesBase.receiveContent(
services.values(
valuesBase.receiveAs(), valdef, mimetype, start, pageLength, transaction
)
);
return valueHandle;
}
@Override
public T tuples(ValuesDefinition valdef, T tupleHandle) {
return tuples(valdef, tupleHandle, -1, null);
}
@Override
public T tuples(ValuesDefinition valdef, T tupleHandle, long start) {
return tuples(valdef, tupleHandle, start, null);
}
@Override
public T tuples(ValuesDefinition valdef, T tupleHandle, Transaction transaction) {
return tuples(valdef, tupleHandle, -1, transaction);
}
@Override
@SuppressWarnings("unchecked")
public T tuples(ValuesDefinition valdef, T tupleHandle, long start, Transaction transaction) {
@SuppressWarnings("rawtypes")
HandleImplementation valuesBase = HandleAccessor.checkHandle(tupleHandle, "values");
if (tupleHandle instanceof TuplesHandle) {
((TuplesHandle) tupleHandle).setQueryCriteria(valdef);
}
Format valuesFormat = valuesBase.getFormat();
switch(valuesFormat) {
case UNKNOWN:
valuesFormat = Format.XML;
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON values results are possible.");
}
String mimetype = valuesFormat.getDefaultMimetype();
long pageLength = (start == -1) ? -1 : getPageLength();
String tid = transaction == null ? null : transaction.getTransactionId();
valuesBase.receiveContent(
services.values(
valuesBase.receiveAs(), valdef, mimetype, start, pageLength, transaction
)
);
return tupleHandle;
}
@Override
public T valuesList(ValuesListDefinition valdef, T valueHandle) {
return valuesList(valdef, valueHandle, null);
}
@SuppressWarnings("unchecked")
@Override
public T valuesList(ValuesListDefinition valdef, T valuesHandle, Transaction transaction) {
@SuppressWarnings("rawtypes")
HandleImplementation valuesBase = HandleAccessor.checkHandle(valuesHandle, "valueslist");
Format valuesFormat = valuesBase.getFormat();
switch(valuesFormat) {
case UNKNOWN:
valuesFormat = Format.XML;
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON values list results are possible.");
}
String mimetype = valuesFormat.getDefaultMimetype();
String tid = transaction == null ? null : transaction.getTransactionId();
valuesBase.receiveContent(services.valuesList(valuesBase.receiveAs(), valdef, mimetype, transaction));
return valuesHandle;
}
@Override
public T optionsList(T optionsHandle) {
return optionsList(optionsHandle, null);
}
@SuppressWarnings("unchecked")
@Override
public T optionsList(T optionsHandle, Transaction transaction) {
@SuppressWarnings("rawtypes")
HandleImplementation optionsBase = HandleAccessor.checkHandle(optionsHandle, "optionslist");
Format optionsFormat = optionsBase.getFormat();
switch(optionsFormat) {
case UNKNOWN:
optionsFormat = Format.XML;
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON options list results are possible.");
}
String mimetype = optionsFormat.getDefaultMimetype();
String tid = transaction == null ? null : transaction.getTransactionId();
optionsBase.receiveContent(services.optionsList(optionsBase.receiveAs(), mimetype, transaction));
return optionsHandle;
}
@Override
public MatchDocumentSummary findOne(QueryDefinition querydef) {
SearchHandle results = search(querydef, new SearchHandle());
MatchDocumentSummary[] summaries = results.getMatchResults();
if (summaries.length > 0) {
return summaries[0];
} else {
return null;
}
}
@Override
public MatchDocumentSummary findOne(QueryDefinition querydef, Transaction transaction) {
SearchHandle results = search(querydef, new SearchHandle(), transaction);
MatchDocumentSummary[] summaries = results.getMatchResults();
if (summaries.length > 0) {
return summaries[0];
} else {
return null;
}
}
@SuppressWarnings("unchecked")
@Override
public String[] suggest(SuggestDefinition suggestDef) {
DOMHandle handle = new DOMHandle();
@SuppressWarnings("rawtypes")
HandleImplementation suggestBase = HandleAccessor.checkHandle(handle, "suggest");
suggestBase.receiveContent(services.suggest(suggestBase.receiveAs(), suggestDef));
Document doc = handle.get();
NodeList nodeList = doc.getDocumentElement().getChildNodes();
List suggestions = new ArrayList();
for (int i=0; i T convert(RawQueryByExampleDefinition query, T convertedHandle) {
@SuppressWarnings("rawtypes")
HandleImplementation convertedBase = HandleAccessor.checkHandle(convertedHandle, "convert");
Format convertedFormat = convertedBase.getFormat();
switch(convertedFormat) {
case UNKNOWN:
@SuppressWarnings("rawtypes")
HandleImplementation queryBase = HandleAccessor.checkHandle(query.getHandle(), "validate");
convertedFormat = queryBase.getFormat();
if (convertedFormat == Format.UNKNOWN) {
convertedFormat = Format.XML;
}
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON conversions are possible.");
}
String mimetype = convertedFormat.getDefaultMimetype();
convertedBase.receiveContent(
services.search(requestLogger, convertedBase.receiveAs(), query, mimetype, "structured")
);
return convertedHandle;
}
@SuppressWarnings("unchecked")
@Override
public T validate(RawQueryByExampleDefinition query, T reportHandle) {
@SuppressWarnings("rawtypes")
HandleImplementation reportBase = HandleAccessor.checkHandle(reportHandle, "validate");
Format reportFormat = reportBase.getFormat();
switch(reportFormat) {
case UNKNOWN:
@SuppressWarnings("rawtypes")
HandleImplementation queryBase = HandleAccessor.checkHandle(query.getHandle(), "validate");
reportFormat = queryBase.getFormat();
if (reportFormat == Format.UNKNOWN) {
reportFormat = Format.XML;
}
break;
case JSON:
case XML:
break;
default:
throw new UnsupportedOperationException("Only XML and JSON validation reports are possible.");
}
String mimetype = reportFormat.getDefaultMimetype();
reportBase.receiveContent(
services.search(requestLogger, reportBase.receiveAs(), query, mimetype, "validate")
);
return reportHandle;
}
@Override
public SuggestDefinition newSuggestDefinition() {
SuggestDefinition def = new SuggestDefinitionImpl();
return def;
}
@Override
public SuggestDefinition newSuggestDefinition(String optionsName) {
SuggestDefinition def = new SuggestDefinitionImpl();
def.setStringCriteria("");
def.setOptionsName(optionsName);
return def;
}
@Override
public SuggestDefinition newSuggestDefinition(String suggestString,
String optionsName) {
SuggestDefinition def = new SuggestDefinitionImpl();
def.setStringCriteria(suggestString);
def.setOptionsName(optionsName);
return def;
}
@Override
public RawCombinedQueryDefinition newRawCombinedQueryDefinitionAs(Format format, Object rawQuery) {
return newRawCombinedQueryDefinitionAs(format, rawQuery, null);
}
@Override
public RawCombinedQueryDefinition newRawCombinedQueryDefinitionAs(Format format, Object rawQuery, String optionsName) {
return newRawCombinedQueryDefinition(structuredWrite(format, rawQuery), optionsName);
}
@Override
public RawCombinedQueryDefinition newRawCombinedQueryDefinition(StructureWriteHandle handle) {
return new RawQueryDefinitionImpl.Combined(handle);
}
@Override
public RawCombinedQueryDefinition newRawCombinedQueryDefinition(StructureWriteHandle handle, String optionsName) {
return new RawQueryDefinitionImpl.Combined(handle, optionsName);
}
@Override
public RawStructuredQueryDefinition newRawStructuredQueryDefinitionAs(Format format, Object query) {
return newRawStructuredQueryDefinitionAs(format, query, null);
}
@Override
public RawStructuredQueryDefinition newRawStructuredQueryDefinitionAs(Format format, Object query, String optionsName) {
return newRawStructuredQueryDefinition(structuredWrite(format, query), optionsName);
}
@Override
public RawStructuredQueryDefinition newRawStructuredQueryDefinition(StructureWriteHandle handle) {
return new RawQueryDefinitionImpl.Structured(handle);
}
@Override
public RawStructuredQueryDefinition newRawStructuredQueryDefinition(StructureWriteHandle handle, String optionsName) {
return new RawQueryDefinitionImpl.Structured(handle, optionsName);
}
@Override
public RawQueryByExampleDefinition newRawQueryByExampleDefinitionAs(Format format, Object query) {
return newRawQueryByExampleDefinitionAs(format, query, null);
}
@Override
public RawQueryByExampleDefinition newRawQueryByExampleDefinitionAs(Format format, Object query, String optionsName) {
return newRawQueryByExampleDefinition(structuredWrite(format, query), optionsName);
}
@Override
public RawQueryByExampleDefinition newRawQueryByExampleDefinition(StructureWriteHandle handle) {
return new RawQueryDefinitionImpl.ByExample(handle);
}
@Override
public RawQueryByExampleDefinition newRawQueryByExampleDefinition(StructureWriteHandle handle, String optionsName) {
return new RawQueryDefinitionImpl.ByExample(handle, optionsName);
}
private StructureWriteHandle structuredWrite(Format format, Object query) {
Class as = query.getClass();
ContentHandle queryHandle = getHandleRegistry().makeHandle(as);
if (!StructureWriteHandle.class.isAssignableFrom(queryHandle.getClass())) {
throw new IllegalArgumentException(
"Handle "+queryHandle.getClass().getName()+
" does not provide structure write handle for "+as.getName()
);
}
Utilities.setHandleContent(queryHandle, query);
Utilities.setHandleStructuredFormat(queryHandle, format);
return (StructureWriteHandle) queryHandle;
}
}
