/*
* 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.deltaspike.data.impl.builder.result;
import java.util.Iterator;
import java.util.List;
import javax.persistence.FlushModeType;
import javax.persistence.LockModeType;
import javax.persistence.Query;
import javax.persistence.metamodel.SingularAttribute;
import org.apache.deltaspike.data.api.QueryResult;
import org.apache.deltaspike.data.impl.builder.OrderDirection;
import org.apache.deltaspike.data.impl.builder.QueryBuilder;
import org.apache.deltaspike.data.impl.builder.postprocessor.CountQueryPostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.FirstResultPostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.FlushModePostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.HintPostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.LockModePostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.MaxResultPostProcessor;
import org.apache.deltaspike.data.impl.builder.postprocessor.OrderByQueryStringPostProcessor;
import org.apache.deltaspike.data.impl.handler.CdiQueryInvocationContext;
import org.apache.deltaspike.data.impl.handler.QueryStringPostProcessor;
public class DefaultQueryResult<T> implements QueryResult<T>
{
private final QueryBuilder builder;
private final CdiQueryInvocationContext context;
private int page = 0;
private int pageSize = 10;
public DefaultQueryResult(QueryBuilder builder, CdiQueryInvocationContext context)
{
this.builder = builder;
this.context = context;
}
@Override
public <X> QueryResult<T> orderAsc(SingularAttribute<T, X> attribute)
{
context.addQueryStringPostProcessor(new OrderByQueryStringPostProcessor(attribute, OrderDirection.ASC));
return this;
}
@Override
public QueryResult<T> orderAsc(String attribute)
{
context.addQueryStringPostProcessor(new OrderByQueryStringPostProcessor(attribute, OrderDirection.ASC));
return this;
}
@Override
public <X> QueryResult<T> orderDesc(SingularAttribute<T, X> attribute)
{
context.addQueryStringPostProcessor(new OrderByQueryStringPostProcessor(attribute, OrderDirection.DESC));
return this;
}
@Override
public QueryResult<T> orderDesc(String attribute)
{
context.addQueryStringPostProcessor(new OrderByQueryStringPostProcessor(attribute, OrderDirection.DESC));
return this;
}
@Override
public <X> QueryResult<T> changeOrder(final SingularAttribute<T, X> attribute)
{
changeOrder(new ChangeOrder()
{
@Override
public boolean matches(OrderByQueryStringPostProcessor orderBy)
{
return orderBy.matches(attribute);
}
@Override
public void addDefault()
{
orderAsc(attribute);
}
});
return this;
}
@Override
public QueryResult<T> changeOrder(final String attribute)
{
changeOrder(new ChangeOrder()
{
@Override
public boolean matches(OrderByQueryStringPostProcessor orderBy)
{
return orderBy.matches(attribute);
}
@Override
public void addDefault()
{
orderAsc(attribute);
}
});
return this;
}
@Override
public QueryResult<T> clearOrder()
{
for (Iterator<QueryStringPostProcessor> it = context.getQueryStringPostProcessors().iterator(); it.hasNext();)
{
if (it.next() instanceof OrderByQueryStringPostProcessor)
{
it.remove();
}
}
return this;
}
@Override
public QueryResult<T> maxResults(int max)
{
context.addJpaQueryPostProcessor(new MaxResultPostProcessor(max));
pageSize = max;
return this;
}
@Override
public QueryResult<T> firstResult(int first)
{
context.addJpaQueryPostProcessor(new FirstResultPostProcessor(first));
return this;
}
@Override
public QueryResult<T> lockMode(LockModeType lockMode)
{
context.addJpaQueryPostProcessor(new LockModePostProcessor(lockMode));
return this;
}
@Override
public QueryResult<T> flushMode(FlushModeType flushMode)
{
context.addJpaQueryPostProcessor(new FlushModePostProcessor(flushMode));
return this;
}
@Override
public QueryResult<T> hint(String hint, Object value)
{
context.addJpaQueryPostProcessor(new HintPostProcessor(hint, value));
return this;
}
@Override
@SuppressWarnings("unchecked")
public List<T> getResultList()
{
return ((Query) builder.execute(context)).getResultList();
}
@Override
@SuppressWarnings("unchecked")
public T getSingleResult()
{
return (T) ((Query) builder.execute(context)).getSingleResult();
}
@Override
public long count()
{
CountQueryPostProcessor counter = new CountQueryPostProcessor();
context.addJpaQueryPostProcessor(counter);
try
{
Long result = (Long) ((Query) builder.execute(context)).getSingleResult();
return result.intValue();
}
catch (RuntimeException e)
{
throw e;
}
catch (Exception e)
{
throw new RuntimeException(e);
}
finally
{
context.removeJpaQueryPostProcessor(counter);
}
}
@Override
public QueryResult<T> withPageSize(int pageSize)
{
return maxResults(pageSize);
}
@Override
public QueryResult<T> toPage(int page)
{
this.page = page;
return firstResult(pageSize * page);
}
@Override
public QueryResult<T> nextPage()
{
page = page + 1;
return firstResult(pageSize * page);
}
@Override
public QueryResult<T> previousPage()
{
page = page > 0 ? page - 1 : page;
return firstResult(pageSize * page);
}
@Override
public int countPages()
{
return (int) Math.ceil((double) count() / pageSize);
}
@Override
public int currentPage()
{
return page;
}
@Override
public int pageSize()
{
return pageSize;
}
private <X> QueryResult<T> changeOrder(ChangeOrder changeOrder)
{
for (QueryStringPostProcessor processor : context.getQueryStringPostProcessors())
{
if (processor instanceof OrderByQueryStringPostProcessor)
{
OrderByQueryStringPostProcessor orderBy = (OrderByQueryStringPostProcessor) processor;
if (changeOrder.matches(orderBy))
{
orderBy.changeDirection();
return this;
}
}
}
changeOrder.addDefault();
return this;
}
private abstract static class ChangeOrder
{
public abstract boolean matches(OrderByQueryStringPostProcessor orderBy);
public abstract void addDefault();
}
}