org.squashtest.tm.domain.requirement.RequirementVersionAttachmentBridge.java Source code

Introduction

Here is the source code for org.squashtest.tm.domain.requirement.RequirementVersionAttachmentBridge.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.domain.requirement;

import org.apache.commons.lang3.StringUtils;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.hibernate.Session;
import org.hibernate.search.bridge.LuceneOptions;
import org.squashtest.tm.domain.search.SessionFieldBridge;

public class RequirementVersionAttachmentBridge extends SessionFieldBridge {

    private static final Integer EXPECTED_LENGTH = 7;

    private String padRawValue(long rawValue) {
        return StringUtils.leftPad(Long.toString(rawValue), EXPECTED_LENGTH, '0');
    }

    @Override
    protected void writeFieldToDocument(String name, Session session, Object value, Document document,
            LuceneOptions luceneOptions) {
        RequirementVersion reqVer = (RequirementVersion) value;

        long count = (Long) session.getNamedQuery("requirementVersion.countAttachments").setReadOnly(true)
                .setParameter("id", reqVer.getId()).uniqueResult();

        Field field = new Field(name, padRawValue(count), luceneOptions.getStore(), luceneOptions.getIndex(),
                luceneOptions.getTermVector());
        field.setBoost(luceneOptions.getBoost());

        document.add(field);
    }
}