BagriXQExpression.java

package com.bagri.xqj;

import static com.bagri.support.util.XQUtils.getXQException;
import static com.bagri.xqj.BagriXQErrors.ex_expression_closed;

import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.util.ArrayList;
import java.util.List;

import javax.xml.xquery.XQConstants;
import javax.xml.xquery.XQException;
import javax.xml.xquery.XQExpression;
import javax.xml.xquery.XQResultSequence;
import javax.xml.xquery.XQStaticContext;

import com.bagri.core.api.ResultCursor;
import com.bagri.core.api.BagriException;
import com.bagri.support.util.XMLUtils;

public class BagriXQExpression extends BagriXQDynamicContext implements XQExpression {
	
	private List<XQResultSequence> results = new ArrayList<>();

	BagriXQExpression(XQStaticContext context) {
		super(context);
	}

	BagriXQExpression(BagriXQConnection connection) {
		super(connection);
	}

	BagriXQExpression(BagriXQConnection connection, XQStaticContext context) {
		super(connection, context);
	}

	@Override
	public void close() throws XQException {
		super.close();
		for (XQResultSequence sq: results) {
			sq.close();
		}
		results.clear();
	}
	
	@Override
	public void executeCommand(String cmd) throws XQException {
		
		checkState(ex_expression_closed);
		if (cmd == null) {
			throw new XQException("Provided command is null");
		}
		connection.executeCommand(cmd, getBindings());
	}

	@Override
	public void executeCommand(Reader cmd) throws XQException {
		
		String str;
		try {
			str = XMLUtils.textToString(cmd);
		} catch (IOException ex) {
			throw new XQException(ex.getMessage());
		}
		executeCommand(str);
	}

	@Override
	public XQResultSequence executeQuery(String query) throws XQException {
		
		checkState(ex_expression_closed);
		if (query == null) {
			throw new XQException("Provided query is null");
		}
		
		ResultCursor result = connection.executeQuery(query, context);
		XQResultSequence sequence;
		if (context.getScrollability() == XQConstants.SCROLLTYPE_SCROLLABLE) {
			try {
				sequence = new ScrollableXQResultSequence(this, result.getList());
			} catch (BagriException ex) {
				throw getXQException(ex); 
			}
		} else {
			sequence = new IterableXQResultSequence(this, result); 
		}
		results.add(sequence);
		return sequence;
	}

	@Override
	public XQResultSequence executeQuery(Reader query) throws XQException {
		
		String str;
		try {
			str = XMLUtils.textToString(query);
		} catch (IOException ex) {
			throw new XQException(ex.getMessage());
		}
		return executeQuery(str);
	}

	@Override
	public XQResultSequence executeQuery(InputStream query) throws XQException {
		
		String str;
		try {
			str = XMLUtils.textToString(query);
		} catch (IOException ex) {
			throw new XQException(ex.getMessage());
		}
		return executeQuery(str);
	}
	
}