org.squashtest.tm.service.internal.repository.hibernate.HibernateRequirementFolderDao.java Source code

Introduction

Here is the source code for org.squashtest.tm.service.internal.repository.hibernate.HibernateRequirementFolderDao.java

Source

/**
 *     This file is part of the Squashtest platform.
 *     Copyright (C) 2010 - 2016 Henix, henix.fr
 *
 *     See the NOTICE file distributed with this work for additional
 *     information regarding copyright ownership.
 *
 *     This is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Lesser General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     this software 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 Lesser General Public License for more details.
 *
 *     You should have received a copy of the GNU Lesser General Public License
 *     along with this software.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.squashtest.tm.service.internal.repository.hibernate;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.hibernate.Query;
import org.hibernate.SQLQuery;
import org.hibernate.type.LongType;
import org.springframework.stereotype.Repository;
import org.squashtest.tm.domain.requirement.RequirementFolder;
import org.squashtest.tm.domain.requirement.RequirementLibraryNode;
import org.squashtest.tm.service.internal.repository.RequirementFolderDao;

@Repository
public class HibernateRequirementFolderDao extends HibernateEntityDao<RequirementFolder>
        implements RequirementFolderDao {
    @SuppressWarnings("rawtypes")
    @Override
    public List<RequirementLibraryNode> findAllContentById(final long folderId) {
        SetQueryParametersCallback setParams = new SetFolderIdParameter(folderId);
        return executeListNamedQuery("requirementFolder.findAllContentById", setParams);

    }

    @SuppressWarnings("rawtypes")
    @Override
    public RequirementFolder findByContent(final RequirementLibraryNode node) {
        SetQueryParametersCallback callback = new SetNodeContentParameter(node);

        return executeEntityNamedQuery("requirementFolder.findByContent", callback);
    }

    @Override
    public List<String> findNamesInFolderStartingWith(final long folderId, final String nameStart) {
        SetQueryParametersCallback newCallBack1 = new ContainerIdNameStartParameterCallback(folderId, nameStart);
        return executeListNamedQuery("requirementFolder.findNamesInFolderStartingWith", newCallBack1);
    }

    @Override
    public List<String> findNamesInLibraryStartingWith(final long libraryId, final String nameStart) {
        SetQueryParametersCallback newCallBack1 = new ContainerIdNameStartParameterCallback(libraryId, nameStart);
        return executeListNamedQuery("requirementFolder.findNamesInLibraryStartingWith", newCallBack1);
    }

    @Override
    public List<String> findNamesInNodeStartingWith(final long nodeId, final String nameStart) {
        SetQueryParametersCallback newCallBack1 = new ContainerIdNameStartParameterCallback(nodeId, nameStart);
        return executeListNamedQuery("requirementLibraryNode.findNamesInNodeStartingWith", newCallBack1);

    }

    @SuppressWarnings("unchecked")
    @Override
    public List<Long[]> findPairedContentForList(final List<Long> ids) {

        if (ids.isEmpty()) {
            return Collections.emptyList();
        }

        SQLQuery query = currentSession()
                .createSQLQuery(NativeQueries.REQUIREMENT_FOLDER_SQL_FIND_PAIRED_CONTENT_FOR_FOLDERS);
        query.setParameterList("folderIds", ids, LongType.INSTANCE);
        query.addScalar("ancestor_id", LongType.INSTANCE);
        query.addScalar("descendant_id", LongType.INSTANCE);

        List<Object[]> result = query.list();

        return toArrayOfLong(result);
    }

    @SuppressWarnings("unchecked")
    @Override
    public List<Long> findContentForList(List<Long> ids) {
        if (ids.isEmpty()) {
            return Collections.emptyList();
        }

        SQLQuery query = currentSession()
                .createSQLQuery(NativeQueries.REQUIREMENT_FOLDER_SQL_FIND_CONTENT_FOR_FOLDER);
        query.setParameterList("folderIds", ids, LongType.INSTANCE);
        query.addScalar("descendant_id", LongType.INSTANCE);

        return query.list();
    }

    private List<Long[]> toArrayOfLong(List<Object[]> input) {
        List<Long[]> result = new ArrayList<>();

        for (Object[] pair : input) {
            Long[] newPair = new Long[] { (Long) pair[0], (Long) pair[1] };
            result.add(newPair);
        }

        return result;
    }

    @Override
    public RequirementFolder findParentOf(final Long id) {
        SetQueryParametersCallback newCallBack = new ContentIdParametterCallback(id);
        return executeEntityNamedQuery("requirementFolder.findParentOf", newCallBack);
    }

    private static final class ContentIdParametterCallback implements SetQueryParametersCallback {
        private long contentId;

        private ContentIdParametterCallback(long contentId) {
            this.contentId = contentId;
        }

        @Override
        public void setQueryParameters(Query query) {
            query.setParameter("contentId", contentId, LongType.INSTANCE);
        }
    }

}