/* * eGov suite of products aim to improve the internal efficiency,transparency, * accountability and the service delivery of the government organizations. * * Copyright (C) <2015> eGovernments Foundation * * The updated version of eGov suite of products as by eGovernments Foundation * is available at http://www.egovernments.org * * This program 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 * any later version. * * This program 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 this program. If not, see http://www.gnu.org/licenses/ or * http://www.gnu.org/licenses/gpl.html . * * In addition to the terms of the GPL license to be adhered to in using this * program, the following additional terms are to be complied with: * * 1) All versions of this program, verbatim or modified must carry this * Legal Notice. * * 2) Any misrepresentation of the origin of the material is prohibited. It * is required that all modified versions of this material be marked in * reasonable ways as different from the original version. * * 3) This license does not grant any rights to any user of the program * with regards to rights under trademark law for use of the trade names * or trademarks of eGovernments Foundation. * * In case of any queries, you can reach eGovernments Foundation at contact@egovernments.org. */ package org.egov.infra.persistence.validator; import org.apache.commons.lang3.reflect.FieldUtils; import org.egov.infra.exception.ApplicationRuntimeException; import org.egov.infra.persistence.validator.annotation.CompositeUnique; import org.hibernate.Criteria; import org.hibernate.Session; import org.hibernate.criterion.Conjunction; import org.hibernate.criterion.Projections; import org.hibernate.criterion.Restrictions; import javax.persistence.EntityManager; import javax.persistence.PersistenceContext; import javax.validation.ConstraintValidator; import javax.validation.ConstraintValidatorContext; public class CompositeUniqueCheckValidator implements ConstraintValidator<CompositeUnique, Object> { private CompositeUnique unique; @PersistenceContext private EntityManager entityManager; @Override public void initialize(final CompositeUnique unique) { this.unique = unique; } @Override public boolean isValid(final Object arg0, final ConstraintValidatorContext constraintValidatorContext) { try { final Number id = (Number) FieldUtils.readField(arg0, unique.id(), true); final boolean isValid = checkCompositeUniqueKey(arg0, id); if (!isValid && unique.enableDfltMsg()) for (final String fieldName : unique.fields()) constraintValidatorContext.buildConstraintViolationWithTemplate(unique.message()).addPropertyNode(fieldName) .addConstraintViolation(); return isValid; } catch (final IllegalAccessException e) { throw new ApplicationRuntimeException("Error while validating composite unique key", e); } } private boolean checkCompositeUniqueKey(final Object arg0, final Number id) throws IllegalAccessException { final Criteria criteria = entityManager.unwrap(Session.class) .createCriteria(unique.isSuperclass() ? arg0.getClass().getSuperclass() : arg0.getClass()); final Conjunction conjunction = Restrictions.conjunction(); for (final String fieldName : unique.fields()) { final Object fieldValue = FieldUtils.readField(arg0, fieldName, true); if (unique.checkForNull() && fieldValue == null) conjunction.add(Restrictions.isNull(fieldName)); else if (fieldValue instanceof String) conjunction.add(Restrictions.eq(fieldName, fieldValue).ignoreCase()); else conjunction.add(Restrictions.eq(fieldName, fieldValue)); } if (id != null) conjunction.add(Restrictions.ne(unique.id(), id)); return criteria.add(conjunction).setProjection(Projections.id()).setMaxResults(1).uniqueResult() == null; } }