From 447b83515ae498f46e9d959e7c07f733647fdf06 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Mon, 09 Feb 2026 10:00:24 +0800
Subject: [PATCH] init
---
src/main/java/cn/net/communion/dbdatasync/dbhelper/impl/PostgreSql.java | 128 ++++++++++++++++++++++++++++++++++++++++++
1 files changed, 128 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cn/net/communion/dbdatasync/dbhelper/impl/PostgreSql.java b/src/main/java/cn/net/communion/dbdatasync/dbhelper/impl/PostgreSql.java
new file mode 100644
index 0000000..b900236
--- /dev/null
+++ b/src/main/java/cn/net/communion/dbdatasync/dbhelper/impl/PostgreSql.java
@@ -0,0 +1,128 @@
+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();
+ }
+}
--
Gitblit v1.9.3