org.yx.db.mapper.SqlXmlBuilderFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
Show all versions of sumk Show documentation
A quick developing framewort for internet company
/**
* Copyright (C) 2016 - 2030 youtongluan.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.yx.db.mapper;
import java.io.InputStream;
import java.io.StringReader;
import java.nio.charset.StandardCharsets;
import java.util.function.Supplier;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import org.xml.sax.InputSource;
import org.yx.exception.SumkException;
import org.yx.log.Logs;
import org.yx.util.IOUtil;
public class SqlXmlBuilderFactory implements Supplier {
protected final DocumentBuilderFactory dbf;
protected String sumkDTD;
public SqlXmlBuilderFactory() {
dbf = DocumentBuilderFactory.newInstance();
dbf.setIgnoringComments(true);
dbf.setIgnoringElementContentWhitespace(true);
dbf.setExpandEntityReferences(false);
dbf.setCoalescing(true);
}
private DocumentBuilder create() throws Exception {
DocumentBuilder dbd = dbf.newDocumentBuilder();
dbd.setEntityResolver((publicId, systemId) -> {
if (systemId == null) {
return null;
}
String dtd = this.sumkDTD;
if (dtd == null) {
InputStream in = SqlXmlParser.class.getClassLoader().getResourceAsStream("META-INF/sql.dtd");
if (in == null) {
return null;
}
byte[] bs = IOUtil.readAllBytes(in, true);
if (bs == null || bs.length == 0) {
return null;
}
dtd = new String(bs, StandardCharsets.UTF_8);
this.sumkDTD = dtd;
}
InputSource source = new InputSource(systemId);
source.setCharacterStream(new StringReader(dtd));
source.setEncoding("UTF-8");
return source;
});
return dbd;
}
@Override
public DocumentBuilder get() {
try {
return this.create();
} catch (Exception e) {
Logs.db().error(e.toString(), e);
throw new SumkException(-7623435, "创建XmlDocumentBuilder失败");
}
}
public DocumentBuilderFactory getDocumentBuilderFactory() {
return dbf;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy