    001// Copyright 2011, 2014 The Apache Software Foundation
    002//
    003// Licensed under the Apache License, Version 2.0 (the "License");
    004// you may not use this file except in compliance with the License.
    005// You may obtain a copy of the License at
    006//
    007// http://www.apache.org/licenses/LICENSE-2.0
    008//
    009// Unless required by applicable law or agreed to in writing, software
    010// distributed under the License is distributed on an "AS IS" BASIS,
    011// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012// See the License for the specific language governing permissions and
    013// limitations under the License.
    014
    015package org.apache.tapestry5.internal.jpa;
    016
    017import org.apache.tapestry5.commons.util.CollectionFactory;
    018import org.apache.tapestry5.jpa.TapestryPersistenceUnitInfo;
    019import org.xml.sax.Attributes;
    020import org.xml.sax.ContentHandler;
    021import org.xml.sax.Locator;
    022import org.xml.sax.SAXException;
    023
    024import javax.persistence.SharedCacheMode;
    025import javax.persistence.ValidationMode;
    026import javax.persistence.spi.PersistenceUnitTransactionType;
    027
    028import java.util.Arrays;
    029import java.util.Collections;
    030import java.util.List;
    031import java.util.Set;
    032
    033public class PersistenceContentHandler implements ContentHandler
    034{
    035    private static final List<String> NAMESPACE_URIS = Arrays.asList(
    036                new String[]{"http://java.sun.com/xml/ns/persistence", "http://xmlns.jcp.org/xml/ns/persistence"});
    037    private static final String ELEMENT_PERSISTENCE_UNIT = "persistence-unit";
    038    private static final String ELEMENT_PROVIDER = "provider";
    039    private static final String ELEMENT_JTA_DATA_SOURCE = "jta-data-source";
    040    private static final String ELEMENT_NON_JTA_DATA_SOURCE = "non-jta-data-source";
    041    private static final String ELEMENT_MAPPING_FILE = "mapping-file";
    042    private static final String ELEMENT_JAR_FILE = "jar-file";
    043    private static final String ELEMENT_CLASS = "class";
    044    private static final String ELEMENT_EXCLUDE_UNLISTED_CLASSES = "exclude-unlisted-classes";
    045    private static final String ELEMENT_CACHING = "shared-cache-mode";
    046    private static final String ELEMENT_VALIDATION_MODE = "validation-mode";
    047    private static final String ELEMENT_PROPERTY = "property";
    048    private static final String ATTRIBUTE_NAME = "name";
    049    private static final String ATTRIBUTE_VALUE = "value";
    050    private static final String ATTRIBUTE_VERSION = "value";
    051    private static final String ATTRIBUTE_TRANSACTION_TYPE = "transaction-type";
    052
    053    private final List<TapestryPersistenceUnitInfo> persistenceUnits = CollectionFactory.newList();
    054    private PersistenceUnitInfoImpl persistenceUnitInfo;
    055    private StringBuilder characters;
    056
    057    public List<TapestryPersistenceUnitInfo> getPersistenceUnits()
    058    {
    059        return persistenceUnits;
    060    }
    061
    062    @Override
    063    public void setDocumentLocator(final Locator locator)
    064    {
    065    }
    066
    067    @Override
    068    public void startDocument() throws SAXException
    069    {
    070    }
    071
    072    @Override
    073    public void endDocument() throws SAXException
    074    {
    075    }
    076
    077    @Override
    078    public void startPrefixMapping(final String prefix, final String uri) throws SAXException
    079    {
    080    }
    081
    082    @Override
    083    public void endPrefixMapping(final String prefix) throws SAXException
    084    {
    085    }
    086
    087    @Override
    088    public void startElement(final String namespaceURI, final String localName, final String qName,
    089                             final Attributes atts) throws SAXException
    090    {
    091        if (NAMESPACE_URIS.contains(namespaceURI))
    092        {
    093            if (ELEMENT_PERSISTENCE_UNIT.equals(localName))
    094            {
    095                persistenceUnitInfo = new PersistenceUnitInfoImpl(atts.getValue(ATTRIBUTE_NAME));
    096                persistenceUnitInfo
    097                        .setPersistenceXMLSchemaVersion(atts.getValue(ATTRIBUTE_VERSION));
    098
    099                final String transactionType = atts.getValue(ATTRIBUTE_TRANSACTION_TYPE);
    100
    101                if (transactionType != null)
    102                {
    103                    persistenceUnitInfo.transactionType(PersistenceUnitTransactionType
    104                            .valueOf(transactionType));
    105                }
    106            } else if (ELEMENT_PROPERTY.equals(localName))
    107            {
    108                final String name = atts.getValue(ATTRIBUTE_NAME);
    109                final String value = atts.getValue(ATTRIBUTE_VALUE);
    110                persistenceUnitInfo.getProperties().setProperty(name, value);
    111            }
    112        }
    113    }
    114
    115    @Override
    116    public void endElement(final String namespaceURI, final String localName, final String qName)
    117            throws SAXException
    118    {
    119
    120        final String string = characters.toString().trim();
    121        characters = null;
    122
    123        if (NAMESPACE_URIS.contains(namespaceURI))
    124        {
    125            if (ELEMENT_PROVIDER.equals(localName))
    126            {
    127                persistenceUnitInfo.persistenceProviderClassName(string);
    128            }
    129            else if (ELEMENT_CLASS.equals(localName))
    130            {
    131                persistenceUnitInfo.addManagedClassName(string);
    132            }
    133            else if (ELEMENT_CACHING.equals(localName))
    134            {
    135                persistenceUnitInfo.sharedCacheMode(toEnum(SharedCacheMode.class, string));
    136            }
    137            else if (ELEMENT_VALIDATION_MODE.equals(localName))
    138            {
    139                persistenceUnitInfo.validationMode(toEnum(ValidationMode.class, string));
    140            }
    141            else if (ELEMENT_MAPPING_FILE.equals(localName))
    142            {
    143                persistenceUnitInfo.addMappingFileName(string);
    144            }
    145            else if (ELEMENT_JAR_FILE.equals(localName))
    146            {
    147                persistenceUnitInfo.addJarFileUrl(string);
    148            }
    149            else if (ELEMENT_NON_JTA_DATA_SOURCE.equals(localName))
    150            {
    151                persistenceUnitInfo.nonJtaDataSource(string);
    152            }
    153            else if (ELEMENT_JTA_DATA_SOURCE.equals(localName))
    154            {
    155                persistenceUnitInfo.jtaDataSource(string);
    156            }
    157            else if (ELEMENT_PERSISTENCE_UNIT.equals(localName))
    158            {
    159                if (persistenceUnitInfo != null)
    160                {
    161                    persistenceUnits.add(persistenceUnitInfo);
    162                    persistenceUnitInfo = null;
    163                }
    164            } else if (ELEMENT_EXCLUDE_UNLISTED_CLASSES.equals(localName))
    165            {
    166                persistenceUnitInfo.excludeUnlistedClasses(Boolean.valueOf(string));
    167            }
    168        }
    169    }
    170
    171    @Override
    172    public void characters(final char[] ch, final int start, final int length) throws SAXException
    173    {
    174        final String s = new String(ch, start, length);
    175
    176        if (characters == null)
    177        {
    178            characters = new StringBuilder(s);
    179        } else
    180        {
    181            characters.append(s);
    182        }
    183
    184    }
    185
    186    @Override
    187    public void ignorableWhitespace(final char[] ch, final int start, final int length)
    188            throws SAXException
    189    {
    190    }
    191
    192    @Override
    193    public void processingInstruction(final String target, final String data) throws SAXException
    194    {
    195    }
    196
    197    @Override
    198    public void skippedEntity(final String name) throws SAXException
    199    {
    200    }
    201
    202    private <T extends Enum<T>> T toEnum(final Class<T> enumType, final String value)
    203    {
    204        return Enum.valueOf(enumType, value);
    205    }
    206}