View Javadoc
1   /**
2    *    Copyright 2006-2016 the original author or authors.
3    *
4    *    Licensed under the Apache License, Version 2.0 (the "License");
5    *    you may not use this file except in compliance with the License.
6    *    You may obtain a copy of the License at
7    *
8    *       http://www.apache.org/licenses/LICENSE-2.0
9    *
10   *    Unless required by applicable law or agreed to in writing, software
11   *    distributed under the License is distributed on an "AS IS" BASIS,
12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *    See the License for the specific language governing permissions and
14   *    limitations under the License.
15   */
16  package org.mybatis.generator.config.xml;
17  
18  import java.io.IOException;
19  import java.io.InputStream;
20  
21  import org.mybatis.generator.codegen.XmlConstants;
22  import org.xml.sax.EntityResolver;
23  import org.xml.sax.InputSource;
24  import org.xml.sax.SAXException;
25  
26  /**
27   * @author Jeff Butler
28   */
29  public class ParserEntityResolver implements EntityResolver {
30  
31      /**
32  	 *  
33  	 */
34      public ParserEntityResolver() {
35          super();
36      }
37  
38      /*
39       * (non-Javadoc)
40       * 
41       * @see org.xml.sax.EntityResolver#resolveEntity(java.lang.String,
42       * java.lang.String)
43       */
44      public InputSource resolveEntity(String publicId, String systemId)
45              throws SAXException, IOException {
46          if (XmlConstants.IBATOR_CONFIG_PUBLIC_ID.equalsIgnoreCase(publicId)) {
47              InputStream is = getClass().getClassLoader().getResourceAsStream(
48                      "org/mybatis/generator/config/xml/ibator-config_1_0.dtd"); //$NON-NLS-1$
49              InputSource ins = new InputSource(is);
50  
51              return ins;
52          } else if (XmlConstants.MYBATIS_GENERATOR_CONFIG_PUBLIC_ID
53                  .equalsIgnoreCase(publicId)) {
54              InputStream is = getClass()
55                      .getClassLoader()
56                      .getResourceAsStream(
57                              "org/mybatis/generator/config/xml/mybatis-generator-config_1_0.dtd"); //$NON-NLS-1$
58              InputSource ins = new InputSource(is);
59  
60              return ins;
61          } else {
62              return null;
63          }
64      }
65  }