/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.ignite.internal.visor.query;
import java.util.List;
import java.util.concurrent.ConcurrentMap;
import javax.cache.Cache;
import org.apache.ignite.IgniteException;
import org.apache.ignite.internal.processors.task.GridInternal;
import org.apache.ignite.internal.util.typedef.internal.S;
import org.apache.ignite.internal.util.typedef.internal.U;
import org.apache.ignite.internal.visor.VisorJob;
import org.apache.ignite.internal.visor.VisorOneNodeTask;
/**
* Task for collecting next page previously executed SQL or SCAN query.
*/
@GridInternal
public class VisorQueryNextPageTask extends VisorOneNodeTask<VisorQueryNextPageTaskArg, VisorQueryResult> {
/** */
private static final long serialVersionUID = 0L;
/** {@inheritDoc} */
@Override protected VisorQueryNextPageJob job(VisorQueryNextPageTaskArg arg) {
return new VisorQueryNextPageJob(arg, debug);
}
/**
* Job for collecting next page previously executed SQL or SCAN query.
*/
private static class VisorQueryNextPageJob extends VisorJob<VisorQueryNextPageTaskArg, VisorQueryResult> {
/** */
private static final long serialVersionUID = 0L;
/**
* Create job with specified argument.
*
* @param arg Job argument.
* @param debug Debug flag.
*/
private VisorQueryNextPageJob(VisorQueryNextPageTaskArg arg, boolean debug) {
super(arg, debug);
}
/** {@inheritDoc} */
@Override protected VisorQueryResult run(VisorQueryNextPageTaskArg arg) {
return arg.getQueryId().startsWith(VisorQueryUtils.SCAN_QRY_NAME) ? nextScanPage(arg) : nextSqlPage(arg);
}
/**
* Collect data from SQL query.
*
* @param arg Query name and page size.
* @return Query result with next page.
*/
private VisorQueryResult nextSqlPage(VisorQueryNextPageTaskArg arg) {
long start = U.currentTimeMillis();
ConcurrentMap<String, VisorQueryCursor<List<?>>> storage = ignite.cluster().nodeLocalMap();
String qryId = arg.getQueryId();
VisorQueryCursor<List<?>> cur = storage.get(qryId);
if (cur == null)
throw new IgniteException("SQL query results are expired.");
List<Object[]> nextRows = VisorQueryUtils.fetchSqlQueryRows(cur, arg.getPageSize());
boolean hasMore = cur.hasNext();
if (hasMore)
cur.accessed(true);
else {
storage.remove(qryId);
cur.close();
}
return new VisorQueryResult(ignite.localNode().id(), qryId, null, nextRows, hasMore,
U.currentTimeMillis() - start);
}
/**
* Collect data from SCAN query
*
* @param arg Query name and page size.
* @return Next page with data.
*/
private VisorQueryResult nextScanPage(VisorQueryNextPageTaskArg arg) {
long start = U.currentTimeMillis();
ConcurrentMap<String, VisorQueryCursor<Cache.Entry<Object, Object>>> storage = ignite.cluster().nodeLocalMap();
String qryId = arg.getQueryId();
VisorQueryCursor<Cache.Entry<Object, Object>> cur = storage.get(qryId);
if (cur == null)
throw new IgniteException("Scan query results are expired.");
List<Object[]> rows = VisorQueryUtils.fetchScanQueryRows(cur, arg.getPageSize());
boolean hasMore = cur.hasNext();
if (hasMore)
cur.accessed(true);
else {
storage.remove(qryId);
cur.close();
}
return new VisorQueryResult(ignite.localNode().id(), qryId, null, rows, hasMore,
U.currentTimeMillis() - start);
}
/** {@inheritDoc} */
@Override public String toString() {
return S.toString(VisorQueryNextPageJob.class, this);
}
}
}