/**
* Copyright (c) 2013, Institute of Information Systems (Sven Groppe and contributors of LUPOSDATE), University of Luebeck
*
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification, are permitted provided that the
* following conditions are met:
*
* - Redistributions of source code must retain the above copyright notice, this list of conditions and the following
* disclaimer.
* - Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
* following disclaimer in the documentation and/or other materials provided with the distribution.
* - Neither the name of the University of Luebeck nor the names of its contributors may be used to endorse or promote
* products derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package lupos.engine.operators.singleinput.federated;
import java.io.IOException;
import java.util.Iterator;
import lupos.datastructures.bindings.Bindings;
import lupos.datastructures.items.Variable;
import lupos.datastructures.items.literal.LazyLiteral;
import lupos.datastructures.items.literal.Literal;
import lupos.datastructures.items.literal.URILiteral;
import lupos.datastructures.queryresult.ParallelIterator;
import lupos.datastructures.queryresult.QueryResult;
import lupos.endpoint.client.Client;
import lupos.misc.util.ImmutableIterator;
import lupos.optimizations.sparql2core_sparql.SPARQLParserVisitorImplementationDumper;
import lupos.sparql1_1.ASTSelectQuery;
import lupos.sparql1_1.ASTVar;
import lupos.sparql1_1.Node;
public class FederatedQueryFetchAsNeeded extends FederatedQueryWithoutSucceedingJoin {
public FederatedQueryFetchAsNeeded(final Node federatedQuery) {
super(federatedQuery);
}
@Override
public QueryResult process(final QueryResult bindings, final int operandID) {
return QueryResult.createInstance(new ImmutableIterator<Bindings>(){
private final Iterator<Bindings> bindingsIterator = bindings.iterator();
private Iterator<Bindings> currentIteratorQueryResult = this.nextIteratorQueryResult();
@Override
public boolean hasNext() {
if(this.currentIteratorQueryResult==null){
return false;
}
while(!this.currentIteratorQueryResult.hasNext()){
this.currentIteratorQueryResult = this.nextIteratorQueryResult();
if(this.currentIteratorQueryResult==null){
return false;
}
}
return true;
}
@Override
public Bindings next() {
if(this.hasNext()){
return this.currentIteratorQueryResult.next();
} else {
return null;
}
}
private Iterator<Bindings> nextIteratorQueryResult(){
try {
if(!this.bindingsIterator.hasNext()){
return null;
}
final Bindings bindingsTemp = this.bindingsIterator.next();
final String fQuery = FederatedQueryFetchAsNeeded.this.toStringQuery(bindingsTemp);
if (!FederatedQueryFetchAsNeeded.this.endpoint.isVariable()) {
return new IteratorQueryResultAndOneBindings(Client.submitQuery(((URILiteral)FederatedQueryFetchAsNeeded.this.endpoint).getString(), fQuery, FederatedQueryFetchAsNeeded.this.bindingsFactory), bindingsTemp);
} else {
Literal endpointURI = bindingsTemp.get((Variable) FederatedQueryFetchAsNeeded.this.endpoint);
if (endpointURI instanceof LazyLiteral) {
endpointURI = ((LazyLiteral) endpointURI).getLiteral();
}
if (endpointURI instanceof URILiteral) {
return new IteratorQueryResultAndOneBindings(Client.submitQuery(((URILiteral) endpointURI).getString(), fQuery, FederatedQueryFetchAsNeeded.this.bindingsFactory), bindingsTemp);
} else {
// ignore or error message?
}
}
} catch(final IOException e){
System.err.println(e);
e.printStackTrace();
}
// in case of error try next one:
return this.nextIteratorQueryResult();
}
});
}
public String toStringQuery(final Bindings bindings) {
final SPARQLParserVisitorImplementationDumper dumper = new SPARQLParserVisitorImplementationDumper() {
@Override
public String visit(final ASTSelectQuery node) {
String ret = "SELECT ";
int i = 0;
if (node.isDistinct()) {
ret += "DISTINCT ";
}
if (node.isReduced()) {
ret += "REDUCED ";
}
if (node.isSelectAll()) {
ret += "*";
} else {
while (i < node.jjtGetNumChildren()
&& node.jjtGetChild(i) instanceof ASTVar) {
// the only difference to its method in the super class:
// just ignore the variable if it is in the current bindings...
final Variable var = new Variable(((ASTVar)node.jjtGetChild(i)).getName());
if ((bindings.get(var) == null)){
ret += this.visitChild(node, i) + " ";
}
i++;
}
}
ret += "\n";
while (i < node.jjtGetNumChildren()) {
ret += this.visitChild(node, i++);
}
return ret;
}
@Override
public String visit(final ASTVar node) {
final Variable var = new Variable(node.getName());
if ((bindings.get(var) != null)) {
return bindings.get(var).toString();
} else {
return "?" + node.getName();
}
}
};
return "SELECT * " + this.federatedQuery.jjtGetChild(1).accept(dumper);
}
public static class IteratorQueryResultAndOneBindings implements ParallelIterator<Bindings>{
private final Iterator<Bindings> it;
private final Bindings bindings;
public IteratorQueryResultAndOneBindings(final QueryResult queryResult, final Bindings bindings){
this.it = queryResult.oneTimeIterator();
this.bindings = bindings;
}
@Override
public boolean hasNext() {
return this.it.hasNext();
}
@Override
public Bindings next() {
if(!this.hasNext()){
return null;
}
final Bindings result = this.it.next();
result.addAll(this.bindings);
return result;
}
@Override
public void remove() {
this.it.remove();
}
@Override
public void close() {
if(this.it instanceof ParallelIterator){
((ParallelIterator<Bindings>)this.it).close();
}
}
}
}