org.broadleafcommerce.core.search.dao.SearchFacetDaoImpl Maven / Gradle / Ivy
/*
* Copyright 2008-2013 the original author or authors.
*
* Licensed 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.broadleafcommerce.core.search.dao;
import org.broadleafcommerce.common.persistence.EntityConfiguration;
import org.broadleafcommerce.core.catalog.domain.ProductImpl;
import org.broadleafcommerce.core.catalog.domain.Sku;
import org.broadleafcommerce.core.search.domain.SearchFacet;
import org.broadleafcommerce.core.search.domain.SearchFacetImpl;
import org.springframework.stereotype.Repository;
import java.util.List;
import javax.annotation.Resource;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Root;
@Repository("blSearchFacetDao")
public class SearchFacetDaoImpl implements SearchFacetDao {
@PersistenceContext(unitName = "blPU")
protected EntityManager em;
@Resource(name="blEntityConfiguration")
protected EntityConfiguration entityConfiguration;
@Override
public List readAllSearchFacets() {
CriteriaBuilder builder = em.getCriteriaBuilder();
CriteriaQuery criteria = builder.createQuery(SearchFacet.class);
Root facet = criteria.from(SearchFacetImpl.class);
criteria.select(facet);
criteria.where(
builder.equal(facet.get("showOnSearch").as(Boolean.class), true)
);
return em.createQuery(criteria).getResultList();
}
@Override
public List readDistinctValuesForField(String fieldName, Class fieldValueClass) {
CriteriaBuilder builder = em.getCriteriaBuilder();
CriteriaQuery criteria = builder.createQuery(fieldValueClass);
Root product = criteria.from(ProductImpl.class);
Path sku = product.get("defaultSku");
Path pathToUse;
if (fieldName.contains("defaultSku.")) {
pathToUse = sku;
fieldName = fieldName.substring("defaultSku.".length());
} else if (fieldName.contains("productAttributes.")) {
pathToUse = product.join("productAttributes");
fieldName = fieldName.substring("productAttributes.".length());
criteria.where(builder.equal(
builder.lower(pathToUse.get("name").as(String.class)), fieldName.toLowerCase()));
fieldName = "value";
} else if (fieldName.contains("product.")) {
pathToUse = product;
fieldName = fieldName.substring("product.".length());
} else {
throw new IllegalArgumentException("Invalid facet fieldName specified: " + fieldName);
}
criteria.where(pathToUse.get(fieldName).as(fieldValueClass).isNotNull());
criteria.distinct(true).select(pathToUse.get(fieldName).as(fieldValueClass));
return em.createQuery(criteria).getResultList();
}
public SearchFacet save(SearchFacet searchFacet) {
return em.merge(searchFacet);
}
}