org.riotfamily.core.dao.hibernate.HqlParentChildDao.java Source code

Introduction

Here is the source code for org.riotfamily.core.dao.hibernate.HqlParentChildDao.java

Source

/* 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.riotfamily.core.dao.hibernate;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.hibernate.Query;
import org.hibernate.SessionFactory;
import org.riotfamily.common.beans.property.PropertyUtils;
import org.riotfamily.common.hibernate.HibernateUtils;
import org.riotfamily.core.dao.CutAndPaste;
import org.riotfamily.core.dao.Hierarchy;
import org.riotfamily.core.dao.ListParams;

/**
 * ParentChildDao implementation based on Hibernate.
 */
public class HqlParentChildDao extends HqlDao implements Hierarchy, CutAndPaste {

    private String parentProperty;

    private boolean parentPropertyMapped;

    public HqlParentChildDao(SessionFactory sessionFactory) {
        super(sessionFactory);
    }

    public String getParentProperty() {
        return parentProperty;
    }

    public void setParentProperty(String parentProperty) {
        this.parentProperty = parentProperty;
    }

    @Override
    protected final void initDao() throws Exception {
        if (parentProperty != null) {
            parentPropertyMapped = HibernateUtils.isPersistentProperty(getSessionFactory(), getEntityClass(),
                    parentProperty);
        }
        initParentChildDao();
    }

    protected void initParentChildDao() throws Exception {
    }

    public Object getParent(Object entity) {
        return PropertyUtils.getProperty(entity, parentProperty);
    }

    @Override
    public void save(Object entity, Object parent) {
        PropertyUtils.setProperty(entity, parentProperty, parent);
        super.save(entity, parent);
    }

    @Override
    protected String getWhereClause(Object parent, ListParams params) {
        StringBuilder sb = new StringBuilder();
        if (parentPropertyMapped) {
            sb.append("this.");
            sb.append(parentProperty);
            if (parent == null) {
                sb.append(" is null");
            } else {
                sb.append(" = :parent ");
            }
        }
        HqlUtils.appendHql(sb, "and", super.getWhereClause(parent, params));

        return sb.toString().replaceAll(":parent\\.(\\w+)", ":parent_$1");
    }

    @Override
    protected void setQueryParameters(Query query, Object parent, ListParams params) {

        super.setQueryParameters(query, parent, params);
        if (parent != null) {
            for (String param : query.getNamedParameters()) {
                Matcher m = Pattern.compile("parent(?:_(\\w+))?").matcher(param);
                if (m.matches()) {
                    Object value = parent;
                    String nested = m.group(1);
                    if (nested != null) {
                        value = PropertyUtils.getProperty(parent, nested);
                    }
                    query.setParameter(param, value);
                }
            }
        }
    }

    public boolean canCut(Object entity) {
        return true;
    }

    public void cut(Object entity, Object parent) {
    }

    public boolean canPasteCut(Object entity, Object target) {
        return true;
    }

    public void pasteCut(Object entity, Object parent) {
        PropertyUtils.setProperty(entity, parentProperty, parent);
        setPositionIfNeeded(entity, parent);
    }

}