package cn.net.communion.dbdatasync.dbhelper.impl;
|
|
import cn.net.communion.dbdatasync.dbhelper.DbHelper;
|
import cn.net.communion.dbdatasync.entity.JobInfo;
|
import org.apache.log4j.Logger;
|
|
import java.sql.Connection;
|
import java.sql.PreparedStatement;
|
import java.sql.ResultSet;
|
import java.sql.SQLException;
|
import java.util.List;
|
|
public class PostgreSql implements DbHelper {
|
private Logger logger = Logger.getLogger(PostgreSql.class);
|
|
@Override
|
public String assembleSQL(String srcSql, Connection conn, JobInfo jobInfo) throws SQLException {
|
String[] fields = jobInfo.getDestTableFields().split(",");
|
String destTable = jobInfo.getDestTable();
|
|
PreparedStatement pst = conn.prepareStatement(srcSql);
|
ResultSet rs = pst.executeQuery();
|
StringBuilder sql = new StringBuilder();
|
sql.append("INSERT INTO ").append(destTable).append(" (").append(jobInfo.getDestTableFields()).append(") VALUES ");
|
|
long count = 0;
|
while (rs.next()) {
|
sql.append("(");
|
for (int index = 0; index < fields.length; index++) {
|
String fieldName = fields[index].trim();
|
String value = rs.getString(fieldName);
|
if (value == null) {
|
sql.append("NULL"); // 显式处理 NULL 值
|
} else if (isGeometryField(fieldName)) {
|
// 处理几何类型字段，使用 ST_GeomFromText 函数
|
logger.debug("Processing geometry field: " + fieldName + " with value: " + value);
|
sql.append(handleGeometryData(value));
|
} else {
|
sql.append("'").append(value).append("'"); // 字符串值加引号
|
}
|
if (index < fields.length - 1) {
|
sql.append(", ");
|
}
|
}
|
sql.append("),");
|
count++;
|
}
|
|
if (rs != null) {
|
rs.close();
|
}
|
if (pst != null) {
|
pst.close();
|
}
|
|
if (count > 0) {
|
sql = new StringBuilder(sql.substring(0, sql.length() - 1)); // 删除最后一个逗号
|
return sql.toString();
|
}
|
return null;
|
}
|
|
/**
|
* 判断是否为几何类型字段
|
*
|
* @param fieldName 字段名
|
* @return 是否为几何类型字段
|
*/
|
private boolean isGeometryField(String fieldName) {
|
// 常见的几何类型字段名
|
return fieldName.equalsIgnoreCase("geom") ||
|
fieldName.contains("geometry") ||
|
fieldName.contains("geom_");
|
}
|
|
/**
|
* 处理几何类型数据
|
*
|
* @param value 几何数据值
|
* @return 处理后的几何数据
|
*/
|
private String handleGeometryData(String value) {
|
logger.debug("原始几何数据值: " + value);
|
|
if (value == null || value.isEmpty()) {
|
logger.debug("几何数据为空，返回 NULL");
|
return "NULL";
|
}
|
|
// 尝试不同的几何数据格式处理
|
try {
|
// 直接尝试使用 ST_GeomFromText 函数来转换几何数据
|
// 不再检查是否为标准的 WKT 格式，因为 MySQL 的 ST_AsText 函数已经保证返回标准的 WKT 格式
|
logger.debug("尝试使用 ST_GeomFromText 函数转换几何数据: " + value.substring(0, Math.min(value.length(), 50)) + "...");
|
return "ST_GeomFromText('" + value + "')";
|
} catch (Exception e) {
|
logger.error("处理几何数据时出错: " + e.getMessage() + ", 原始值: " + value);
|
return "NULL"; // 出错时返回 NULL
|
}
|
}
|
|
|
|
@Override
|
public void executeSQL(String sql, Connection conn) throws SQLException {
|
logger.debug("Executing SQL: " + sql);
|
try (PreparedStatement pst = conn.prepareStatement(sql)) {
|
pst.execute();
|
conn.commit();
|
}
|
}
|
|
// 辅助方法：设置参数
|
private void setParameters(PreparedStatement pst, List<Object> params) throws SQLException {
|
for (int i = 0; i < params.size(); i++) {
|
pst.setObject(i + 1, params.get(i));
|
}
|
}
|
|
// 辅助方法：序列化 SQL 和参数
|
private String serializeSqlWithParams(List<String> sqlStatements, List<List<Object>> paramLists) {
|
StringBuilder result = new StringBuilder();
|
for (int i = 0; i < sqlStatements.size(); i++) {
|
result.append(sqlStatements.get(i)).append("; ");
|
}
|
return result.toString();
|
}
|
}
|
