org.swiftboot.data.model.interceptor.IdInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of swiftboot-data Show documentation

Data access component based on Spring-Data-JPA

There is a newer version: 2.4.8

package org.swiftboot.data.model.interceptor;

import org.hibernate.EmptyInterceptor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.swiftboot.data.model.entity.IdPersistable;
import org.swiftboot.data.model.id.IdPopulator;

import javax.annotation.Resource;
import java.util.Iterator;

/**
 * Interceptor for populating id to entities before Hibernate flushing them.
 *
 * @author swiftech
 * @since 2.0.1
 */
public class IdInterceptor extends EmptyInterceptor {

    private static final Logger log = LoggerFactory.getLogger(IdInterceptor.class);

    @Resource
    private IdPopulator idPopulator;

    @Override
    public void preFlush(Iterator entities) {
        log.trace("preFlush()");
        try {
            while (entities.hasNext()) {
                Object e = entities.next();
                if (e instanceof IdPersistable) {
                    IdPersistable entity = (IdPersistable) e;
                    idPopulator.populate(entity, false); // avoid re-populate id for children entities.
                }
            }
        } catch (Exception e) {
            log.warn("Exception throws when trying to set id for entities: " + e.getLocalizedMessage());
        }
    }
}