/* * 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.Unique; import org.hibernate.Criteria; import org.hibernate.Session; 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 UniqueCheckValidator implements ConstraintValidator<Unique, Object> { private Unique unique; @PersistenceContext private EntityManager entityManager; @Override public void initialize(final Unique 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); boolean isValid = true; for (final String fieldName : unique.fields()) if (!checkUnique(arg0, id, fieldName)) { isValid = false; if (unique.enableDfltMsg()) constraintValidatorContext.buildConstraintViolationWithTemplate(unique.message()) .addPropertyNode(fieldName) .addConstraintViolation(); } return isValid; } catch (final IllegalAccessException e) { throw new ApplicationRuntimeException("Error while validating unique key", e); } } private boolean checkUnique(final Object arg0, final Number id, final String fieldName) throws IllegalAccessException { final Criteria criteria = entityManager.unwrap(Session.class) .createCriteria(unique.isSuperclass() ? arg0.getClass().getSuperclass() : arg0.getClass()); final Object fieldValue = FieldUtils.readField(arg0, fieldName, true); if (fieldValue instanceof String) criteria.add(Restrictions.eq(fieldName, fieldValue).ignoreCase()); else criteria.add(Restrictions.eq(fieldName, fieldValue)); if (id != null) criteria.add(Restrictions.ne(unique.id(), id)); return criteria.setProjection(Projections.id()).setMaxResults(1).uniqueResult() == null; } }