001    /**
002     *   GRANITE DATA SERVICES
003     *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004     *
005     *   This file is part of the Granite Data Services Platform.
006     *
007     *   Granite Data Services is free software; you can redistribute it and/or
008     *   modify it under the terms of the GNU Lesser General Public
009     *   License as published by the Free Software Foundation; either
010     *   version 2.1 of the License, or (at your option) any later version.
011     *
012     *   Granite Data Services is distributed in the hope that it will be useful,
013     *   but WITHOUT ANY WARRANTY; without even the implied warranty of
014     *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015     *   General Public License for more details.
016     *
017     *   You should have received a copy of the GNU Lesser General Public
018     *   License along with this library; if not, write to the Free Software
019     *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020     *   USA, or see <http://www.gnu.org/licenses/>.
021     */
022    package org.granite.tide.data;
023    
024    import java.util.HashMap;
025    import java.util.Map;
026    
027    import javax.persistence.EntityExistsException;
028    import javax.persistence.EntityNotFoundException;
029    import javax.persistence.NoResultException;
030    import javax.persistence.NonUniqueResultException;
031    import javax.persistence.OptimisticLockException;
032    import javax.persistence.PersistenceException;
033    import javax.persistence.RollbackException;
034    import javax.persistence.TransactionRequiredException;
035    
036    import org.granite.messaging.service.ExceptionConverter;
037    import org.granite.messaging.service.ServiceException;
038    
039    
040    public class PersistenceExceptionConverter implements ExceptionConverter {
041        
042        public static final String ENTITY_EXISTS = "Persistence.EntityExists";
043        public static final String ENTITY_NOT_FOUND = "Persistence.EntityNotFound";
044        public static final String NON_UNIQUE_RESULT = "Persistence.NonUnique";
045        public static final String NO_RESULT = "Persistence.NoResult";
046        public static final String OPTIMISTIC_LOCK = "Persistence.OptimisticLock";
047        public static final String TRANSACTION_REQUIRED = "Persistence.TransactionRequired";
048        public static final String ROLLBACK = "Persistence.Rollback";
049        public static final String OTHER = "Persistence.Error";
050        
051    
052        public boolean accepts(Throwable t, Throwable finalException) {
053            return t.getClass().equals(EntityExistsException.class) 
054                || t.getClass().equals(EntityNotFoundException.class)
055                || t.getClass().equals(NonUniqueResultException.class)
056                || t.getClass().equals(NoResultException.class)
057                || t.getClass().equals(OptimisticLockException.class)
058                || t.getClass().equals(TransactionRequiredException.class)
059                || t.getClass().equals(RollbackException.class)
060                || PersistenceException.class.isAssignableFrom(t.getClass());
061        }
062    
063        public ServiceException convert(Throwable t, String detail, Map<String, Object> extendedData) {
064            String error = null;
065            Map<String, Object> ex = null;
066            if (t.getClass().equals(EntityExistsException.class))
067                error = ENTITY_EXISTS;
068            else if (t.getClass().equals(EntityNotFoundException.class))
069                error = ENTITY_NOT_FOUND;
070            else if (t.getClass().equals(NonUniqueResultException.class))
071                error = NON_UNIQUE_RESULT;
072            else if (t.getClass().equals(NoResultException.class))
073                error = NO_RESULT;
074            else if (t.getClass().equals(OptimisticLockException.class)) {
075                error = OPTIMISTIC_LOCK;
076                ex = new HashMap<String, Object>();
077                ex.put("entity", ((OptimisticLockException)t).getEntity());
078            }
079            else if (t.getClass().equals(TransactionRequiredException.class))
080                error = TRANSACTION_REQUIRED;
081            else if (t.getClass().equals(RollbackException.class))
082                error = ROLLBACK;
083            else
084                error = OTHER; 
085            
086            ServiceException se = new ServiceException(error, t.getMessage(), detail, t);
087            if (ex != null && !ex.isEmpty())
088                se.getExtendedData().putAll(ex);
089            return se;
090        }
091    
092    }