Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.mycore.datamodel.classifications2.impl.MCRCategLinkServiceImpl Maven / Gradle / Ivy
/*
* This file is part of *** M y C o R e ***
* See http://www.mycore.de/ for details.
*
* MyCoRe is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* MyCoRe is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with MyCoRe. If not, see LINK_CLASS = MCRCategoryLinkImpl.class;
private static final String NAMED_QUERY_NAMESPACE = "MCRCategoryLink.";
private static MCRCache categCache = new MCRCache<>(
MCRConfiguration2.getInt("MCR.Classifications.LinkServiceImpl.CategCache.Size").orElse(1000),
"MCRCategLinkService category cache");
private static MCRCategoryDAO DAO = MCRCategoryDAOFactory.getInstance();
public MCRCategLinkServiceImpl() {
HIB_CONNECTION_INSTANCE = MCRHIBConnection.instance();
}
@Override
public Map countLinks(MCRCategory parent, boolean childrenOnly) {
return countLinksForType(parent, null, childrenOnly);
}
@Override
public Map countLinksForType(MCRCategory parent, String type, boolean childrenOnly) {
boolean restrictedByType = type != null;
String queryName;
if (childrenOnly) {
queryName = restrictedByType ? "NumberByTypePerChildOfParentID" : "NumberPerChildOfParentID";
} else {
queryName = restrictedByType ? "NumberByTypePerClassID" : "NumberPerClassID";
}
Map countLinks = new HashMap<>();
Collection ids = childrenOnly ? getAllChildIDs(parent) : getAllCategIDs(parent);
for (MCRCategoryID id : ids) {
// initialize all categIDs with link count of zero
countLinks.put(id, 0);
}
//have to use rootID here if childrenOnly=false
//old classification browser/editor could not determine links correctly otherwise
if (!childrenOnly) {
parent = parent.getRoot();
} else if (!(parent instanceof MCRCategoryImpl) || ((MCRCategoryImpl) parent).getInternalID() == 0) {
parent = MCRCategoryDAOImpl.getByNaturalID(MCREntityManagerProvider.getCurrentEntityManager(),
parent.getId());
}
LOGGER.info("parentID:{}", parent.getId());
String classID = parent.getId().getRootID();
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + queryName);
// query can take long time, please cache result
q.setCacheable(true);
q.setReadOnly(true);
q.setParameter("classID", classID);
if (childrenOnly) {
q.setParameter("parentID", ((MCRCategoryImpl) parent).getInternalID());
}
if (restrictedByType) {
q.setParameter("type", type);
}
// get object count for every category (not accumulated)
@SuppressWarnings("unchecked")
List result = (List) q.getResultList();
for (Object[] sr : result) {
MCRCategoryID key = new MCRCategoryID(classID, sr[0].toString());
Number value = (Number) sr[1];
countLinks.put(key, value);
}
return countLinks;
}
@Override
public void deleteLink(MCRCategLinkReference reference) {
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + "deleteByObjectID");
q.setParameter("id", reference.getObjectID());
q.setParameter("type", reference.getType());
int deleted = q.executeUpdate();
LOGGER.debug("Number of Links deleted: {}", deleted);
}
@Override
public void deleteLinks(final Collection ids) {
if (ids.isEmpty()) {
return;
}
HashMap> typeMap = new HashMap<>();
//prepare
Collection objectIds = new LinkedList<>();
String currentType = ids.iterator().next().getType();
typeMap.put(currentType, objectIds);
//collect per type
for (MCRCategLinkReference ref : ids) {
if (!currentType.equals(ref.getType())) {
currentType = ref.getType();
objectIds = typeMap.computeIfAbsent(ref.getType(), k -> new LinkedList<>());
}
objectIds.add(ref.getObjectID());
}
EntityManager em = MCREntityManagerProvider.getCurrentEntityManager();
javax.persistence.Query q = em.createNamedQuery(NAMED_QUERY_NAMESPACE + "deleteByObjectCollection");
int deleted = 0;
for (Map.Entry> entry : typeMap.entrySet()) {
q.setParameter("ids", entry.getValue());
q.setParameter("type", entry.getKey());
deleted += q.executeUpdate();
}
LOGGER.debug("Number of Links deleted: {}", deleted);
}
@Override
@SuppressWarnings("unchecked")
public Collection getLinksFromCategory(MCRCategoryID id) {
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + "ObjectIDByCategory");
q.setCacheable(true);
q.setParameter("id", id);
q.setReadOnly(true);
return (Collection) q.getResultList();
}
@Override
@SuppressWarnings("unchecked")
public Collection getLinksFromCategoryForType(MCRCategoryID id, String type) {
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + "ObjectIDByCategoryAndType");
q.setCacheable(true);
q.setParameter("id", id);
q.setParameter("type", type);
q.setReadOnly(true);
return (Collection) q.getResultList();
}
@Override
@SuppressWarnings("unchecked")
public Collection getLinksFromReference(MCRCategLinkReference reference) {
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + "categoriesByObjectID");
q.setCacheable(true);
q.setParameter("id", reference.getObjectID());
q.setParameter("type", reference.getType());
q.setReadOnly(true);
return (Collection) q.getResultList();
}
@Override
public void setLinks(MCRCategLinkReference objectReference, Collection categories) {
EntityManager entityManager = MCREntityManagerProvider.getCurrentEntityManager();
categories
.stream()
.distinct()
.forEach(categID -> {
final MCRCategory category = getMCRCategory(entityManager, categID);
if (category == null) {
throw new MCRPersistenceException("Could not link to unknown category " + categID);
}
MCRCategoryLinkImpl link = new MCRCategoryLinkImpl(category, objectReference);
if (LOGGER.isDebugEnabled()) {
MCRCategory linkedCategory = link.getCategory();
StringBuilder debugMessage = new StringBuilder("Adding Link from ").append(linkedCategory.getId());
if (linkedCategory instanceof MCRCategoryImpl) {
debugMessage.append("(").append(((MCRCategoryImpl) linkedCategory).getInternalID())
.append(") ");
}
debugMessage.append("to ").append(objectReference);
LOGGER.debug(debugMessage.toString());
}
entityManager.persist(link);
LOGGER.debug("===DONE: {}", link.id);
});
}
private static MCRCategory getMCRCategory(EntityManager entityManager, MCRCategoryID categID) {
MCRCategory categ = categCache.getIfUpToDate(categID, DAO.getLastModified());
if (categ != null) {
return categ;
}
categ = MCRCategoryDAOImpl.getByNaturalID(entityManager, categID);
if (categ == null) {
return null;
}
categCache.put(categID, categ);
return categ;
}
@Override
public Map hasLinks(MCRCategory category) {
if (category == null) {
return hasLinksForClassifications();
}
MCRCategoryImpl rootImpl = (MCRCategoryImpl) MCRCategoryDAOFactory.getInstance()
.getCategory(category.getRoot().getId(), -1);
if (rootImpl == null) {
//Category does not exist, so it has no links
return getNoLinksMap(category);
}
HashMap boolMap = new HashMap<>();
final BitSet linkedInternalIds = getLinkedInternalIds();
storeHasLinkValues(boolMap, linkedInternalIds, rootImpl);
return boolMap;
}
@SuppressWarnings("unchecked")
private Map hasLinksForClassifications() {
HashMap boolMap = new HashMap<>() {
private static final long serialVersionUID = 1L;
@Override
public Boolean get(Object key) {
return Optional.ofNullable(super.get(key)).orElse(Boolean.FALSE);
}
};
Query linkedClassifications = MCRHIBConnection.instance()
.getNamedQuery(NAMED_QUERY_NAMESPACE + "linkedClassifications");
linkedClassifications.setReadOnly(true);
((List) linkedClassifications.getResultList()).stream().map(MCRCategoryID::rootID)
.forEach(id -> boolMap.put(id, true));
return boolMap;
}
private Map getNoLinksMap(MCRCategory category) {
HashMap boolMap = new HashMap<>();
for (MCRCategoryID categID : getAllCategIDs(category)) {
boolMap.put(categID, false);
}
return boolMap;
}
private void storeHasLinkValues(HashMap boolMap, BitSet internalIds,
MCRCategoryImpl parent) {
final int internalID = parent.getInternalID();
if (internalID < internalIds.size() && internalIds.get(internalID)) {
addParentHasValues(boolMap, parent);
} else {
boolMap.put(parent.getId(), false);
}
for (MCRCategory child : parent.getChildren()) {
storeHasLinkValues(boolMap, internalIds, (MCRCategoryImpl) child);
}
}
private void addParentHasValues(HashMap boolMap, MCRCategory parent) {
boolMap.put(parent.getId(), true);
if (parent.isCategory() && !Optional.ofNullable(boolMap.get(parent.getParent().getId())).orElse(false)) {
addParentHasValues(boolMap, parent.getParent());
}
}
private BitSet getLinkedInternalIds() {
EntityManager em = MCREntityManagerProvider.getCurrentEntityManager();
CriteriaBuilder cb = em.getCriteriaBuilder();
CriteriaQuery query = cb.createQuery(Number.class);
Root li = query.from(LINK_CLASS);
Path internalId = li.get(MCRCategoryLinkImpl_.category).get(MCRCategoryImpl_.internalID);
List result = em
.createQuery(
query.select(internalId)
.orderBy(cb.desc(internalId)))
.getResultList();
int maxSize = result.size() == 0 ? 1 : result.get(0).intValue() + 1;
BitSet linkSet = new BitSet(maxSize);
for (Number internalID : result) {
linkSet.set(internalID.intValue(), true);
}
return linkSet;
}
private static Collection getAllCategIDs(MCRCategory category) {
return MCRStreamUtils.flatten(category, MCRCategory::getChildren, Collection::parallelStream)
.map(MCRCategory::getId)
.collect(Collectors.toCollection(HashSet::new));
}
private static Collection getAllChildIDs(MCRCategory category) {
return category.getChildren()
.stream()
.map(MCRCategory::getId)
.collect(Collectors.toCollection(HashSet::new));
}
@Override
public boolean hasLink(MCRCategory mcrCategory) {
return !hasLinks(mcrCategory).isEmpty();
}
@Override
public boolean isInCategory(MCRCategLinkReference reference, MCRCategoryID id) {
Query q = HIB_CONNECTION_INSTANCE.getNamedQuery(NAMED_QUERY_NAMESPACE + "CategoryAndObjectID");
q.setCacheable(true);
q.setReadOnly(true);
q.setParameter("rootID", id.getRootID());
q.setParameter("categID", id.getID());
q.setParameter("objectID", reference.getObjectID());
q.setParameter("type", reference.getType());
return !q.getResultList().isEmpty();
}
@Override
public Collection getReferences(String type) {
EntityManager em = MCREntityManagerProvider.getCurrentEntityManager();
CriteriaBuilder cb = em.getCriteriaBuilder();
CriteriaQuery query = cb.createQuery(MCRCategLinkReference.class);
Root li = query.from(LINK_CLASS);
Path objectReferencePath = li.get(MCRCategoryLinkImpl_.objectReference);
return em
.createQuery(
query.select(objectReferencePath)
.where(cb.equal(objectReferencePath.get(MCRCategLinkReference_.type), type)))
.setHint(QueryHints.READ_ONLY, "true")
.getResultList();
}
@Override
public Collection getTypes() {
EntityManager em = MCREntityManagerProvider.getCurrentEntityManager();
TypedQuery q = em.createNamedQuery(NAMED_QUERY_NAMESPACE + "types", String.class);
return q.getResultList();
}
@Override
public Collection getLinks(String type) {
EntityManager em = MCREntityManagerProvider.getCurrentEntityManager();
TypedQuery q = em.createNamedQuery(NAMED_QUERY_NAMESPACE + "links", MCRCategoryLink.class);
q.setParameter("type", type);
return q.getResultList();
}
}
