Pgjdbc试用copy

无需多言，请看代码，细节写在备注里了

package org.postgresql.newTest;

import org.postgresql.copy.CopyManager;
import org.postgresql.core.BaseConnection;

import java.io.IOException;
import java.io.StringReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;

public class CopyManagerTest {

  private static final String DB_URL = "jdbc:postgresql://localhost:5866/mydb";
  private static final String USER = "testuser";
  private static final String PASSWORD = "helloworld";

  public void copy(String tabName, List<LinkedHashMap> list) {
    StringReader reader = null;
    try (Connection connection = getConnection()) {
      CopyManager copyManager = new CopyManager((BaseConnection) connection);
      reader = createStringReader(list);
      //格式为value,value,value\n
      copyManager.copyIn("COPY " + tabName + " FROM STDIN WITH CSV DELIMITER AS ',' NULL AS 'null'", reader);
    } catch (Exception e) {
      handleException(e);
    } finally {
      if (reader != null) {
        try {
          reader.close();
        } catch (Exception e) {
          e.printStackTrace();
        }
      }
    }
  }

  private Connection getConnection() throws SQLException, ClassNotFoundException {
    Class.forName("org.postgresql.Driver");
    return DriverManager.getConnection(DB_URL, USER, PASSWORD);
  }

  private StringReader createStringReader(List<LinkedHashMap> list) {
    StringBuilder sb = new StringBuilder();
    for (LinkedHashMap map : list) {
      Object[] values = map.values().toArray();
      for (int i = 0; i < values.length; i++) {
        if (values[i] == null) {
          sb.append("null");
        } else {
          sb.append(values[i].toString());
        }
        //为了规整格式，格式为value,value,value\n
        //如：0,Shanghai,1724380743590,0
        if (i != values.length - 1) {
          sb.append(',');
        }
      }
      sb.append('\n');
    }
    return new StringReader(sb.toString());
  }

  private void handleException(Exception e) {
    e.printStackTrace();
    System.err.println("Error occurred while copying data: " + e.getMessage());
  }

  public static void main(String[] args) {
    List<LinkedHashMap> list = new ArrayList<>();
    //因为只是简单的测试，使用了LinkedHashMap暂存，保证了写入文件的顺序插入
    for (int i = 0; i < 400; i++) {
      LinkedHashMap<String, Object> linkedHashMap = new LinkedHashMap<>();
      linkedHashMap.put("id", i);
      linkedHashMap.put("name", "h_" + i);
      linkedHashMap.put("value", Long.valueOf(i % 100));
      list.add(linkedHashMap);
    }
    new CopyManagerTest().copy("t_test", list);
  }
}