| /* |
| * Copyright (c) 2009-2013 Eike Stepper (Berlin, Germany) and others. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * Simon McDuff - initial API and implementation |
| * Eike Stepper - maintenance |
| */ |
| package org.eclipse.emf.cdo.server.internal.net4j.protocol; |
| |
| import org.eclipse.emf.cdo.common.id.CDOIDReference; |
| import org.eclipse.emf.cdo.common.protocol.CDODataInput; |
| import org.eclipse.emf.cdo.common.protocol.CDODataOutput; |
| import org.eclipse.emf.cdo.common.protocol.CDOProtocolConstants; |
| import org.eclipse.emf.cdo.common.util.CDOQueryInfo; |
| import org.eclipse.emf.cdo.internal.common.CDOQueryInfoImpl; |
| import org.eclipse.emf.cdo.server.internal.net4j.bundle.OM; |
| import org.eclipse.emf.cdo.spi.server.InternalQueryManager; |
| import org.eclipse.emf.cdo.spi.server.InternalQueryResult; |
| import org.eclipse.emf.cdo.spi.server.InternalView; |
| |
| import org.eclipse.net4j.util.om.trace.ContextTracer; |
| |
| import java.io.IOException; |
| |
| /** |
| * @author Simon McDuff |
| */ |
| public class QueryIndication extends CDOServerReadIndication |
| { |
| private static final ContextTracer TRACER = new ContextTracer(OM.DEBUG_PROTOCOL, QueryIndication.class); |
| |
| private boolean xrefs; |
| |
| private boolean disableResponseFlushing; |
| |
| private InternalQueryResult queryResult; |
| |
| public QueryIndication(CDOServerProtocol protocol) |
| { |
| super(protocol, CDOProtocolConstants.SIGNAL_QUERY); |
| } |
| |
| @Override |
| protected void indicating(CDODataInput in) throws IOException |
| { |
| int viewID = in.readInt(); |
| InternalView view = getSession().getView(viewID); |
| |
| CDOQueryInfo queryInfo = new CDOQueryInfoImpl(in); |
| xrefs = queryInfo.getQueryLanguage().equals(CDOProtocolConstants.QUERY_LANGUAGE_XREFS); |
| |
| Object param = queryInfo.getParameters().get(CDOQueryInfo.PARAM_DISABLE_RESPONSE_FLUSHING); |
| disableResponseFlushing = xrefs || Boolean.TRUE.equals(param); |
| |
| InternalQueryManager queryManager = getRepository().getQueryManager(); |
| queryResult = queryManager.execute(view, queryInfo); |
| } |
| |
| @Override |
| protected void responding(CDODataOutput out) throws IOException |
| { |
| // Return queryID immediately. |
| out.writeInt(queryResult.getQueryID()); |
| flushUnlessDisabled(); |
| |
| int numberOfResults = 0; |
| for (;;) |
| { |
| Object object; |
| |
| try |
| { |
| if (!queryResult.hasNext()) |
| { |
| break; |
| } |
| |
| object = queryResult.next(); |
| } |
| catch (Throwable ex) |
| { |
| object = ex; |
| } |
| |
| out.writeBoolean(true); |
| |
| if (xrefs) |
| { |
| CDOIDReference idReference = (CDOIDReference)object; |
| out.writeCDOIDReference(idReference); |
| } |
| else |
| { |
| out.writeCDORevisionOrPrimitive(object); |
| |
| if (object instanceof Throwable) |
| { |
| break; |
| } |
| } |
| |
| ++numberOfResults; |
| |
| try |
| { |
| if (queryResult.peek() == null) |
| { |
| flushUnlessDisabled(); |
| } |
| } |
| catch (IOException ex) |
| { |
| throw ex; |
| } |
| catch (Throwable ex) |
| { |
| // Ignore execution exceptions from peek(); they're handle |
| } |
| } |
| |
| if (TRACER.isEnabled()) |
| { |
| TRACER.trace("Query returned " + numberOfResults + " results"); //$NON-NLS-1$ //$NON-NLS-2$ |
| } |
| |
| // Query is done successfully |
| out.writeBoolean(false); |
| } |
| |
| private void flushUnlessDisabled() throws IOException |
| { |
| if (!disableResponseFlushing) |
| { |
| flush(); |
| } |
| } |
| } |