com.jn.sqlhelper.dialect.urlparser.PostgreSQLUrlParser Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlhelper-dialect Show documentation
Show all versions of sqlhelper-dialect Show documentation
Database dialects ( supports pagination, UrlParser, SqlStatementParser, WallFilter, BatchExecutor for Test)
based Java, you can use it in any ORM framework.
package com.jn.sqlhelper.dialect.urlparser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class PostgreSQLUrlParser extends CommonUrlParser {
private static final String URL_PREFIX = "jdbc:postgresql:";
private static final String LOADBALANCE_URL_PREFIX = "jdbc:postgresql:loadbalance:";
private static final Logger logger = LoggerFactory.getLogger(PostgreSQLUrlParser.class);
private static final List URL_SCHEMAS = Arrays.asList(new String[]{URL_PREFIX});
@Override
public List getUrlSchemas() {
return URL_SCHEMAS;
}
public PostgreSQLUrlParser() {
}
@Override
public String getName() {
return "postgresql";
}
@Override
public DatabaseInfo parse(final String jdbcUrl) {
if (jdbcUrl == null) {
logger.info("jdbcUrl may not be null");
return UnKnownDatabaseInfo.INSTANCE;
}
if (!jdbcUrl.startsWith("jdbc:postgresql:")) {
logger.info("jdbcUrl has invalid prefix.(url:{}, prefix:{})", (Object) jdbcUrl, (Object) "jdbc:postgresql:");
return UnKnownDatabaseInfo.INSTANCE;
}
DatabaseInfo result = null;
try {
result = this.parse0(jdbcUrl);
} catch (Exception e) {
logger.info("PostgreJdbcUrl parse error. url: {}, Caused: {}", new Object[]{jdbcUrl, e.getMessage(), e});
result = UnKnownDatabaseInfo.createUnknownDataBase("postgresql", jdbcUrl);
}
return result;
}
private DatabaseInfo parse0(final String url) {
if (this.isLoadbalanceUrl(url)) {
return this.parseLoadbalancedUrl(url);
}
return this.parseNormal(url);
}
private DatabaseInfo parseLoadbalancedUrl(final String url) {
final StringMaker maker = new StringMaker(url);
maker.after("jdbc:postgresql:");
final String host = maker.after("//").before('/').value();
final String[] parsedHost = host.split(",");
final List hostList = Arrays.asList(parsedHost);
final String databaseId = maker.next().afterLast('/').before('?').value();
final String normalizedUrl = maker.clear().before('?').value();
return new DefaultDatabaseInfo(getName(), url, normalizedUrl, hostList, databaseId);
}
private boolean isLoadbalanceUrl(final String url) {
return url.regionMatches(true, 0, "jdbc:postgresql:loadbalance:", 0, "jdbc:postgresql:loadbalance:".length());
}
private DatabaseInfo parseNormal(final String url) {
final StringMaker maker = new StringMaker(url);
maker.after("jdbc:postgresql:");
final String host = maker.after("//").before('/').value();
final List hostList = new ArrayList(1);
hostList.add(host);
final String databaseId = maker.next().afterLast('/').before('?').value();
final String normalizedUrl = maker.clear().before('?').value();
return new DefaultDatabaseInfo(getName(), url, normalizedUrl, hostList, databaseId);
}
}