批量查询
此示例主要使用setFetchSize调整客户端内存使用,原理是通过数据库游标来分批获取服务端数据,但会加大网络交互,可能会损失部分性能。由于游标事务内有效,故需要先关闭自动提交事务,最后执行手动提交事务。
代码运行的前提条件:根据实际情况添加opengaussjdbc.jar包(例如用户使用IDE执行代码,则需要在本地IDE添加opengaussjdbc.jar包)。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
// 认证用的用户名和密码直接写到代码中有很大的安全风险,建议在配置文件或者环境变量中存放(密码应密文存放,使用时解密),确保安全。 // 本示例以用户名和密码保存在环境变量中为例,运行本示例前请先在本地环境中设置环境变量(环境变量名称请根据自身情况进行设置)EXAMPLE_USERNAME_ENV和EXAMPLE_PASSWORD_ENV。 // $ip、$port、database需要用户自行修改。 import java.sql.Connection; import java.sql.DriverManager; import java.sql.Statement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.PreparedStatement; public class Batch { public static void main(String[] args) throws SQLException { String driver = "com.huawei.opengauss.jdbc.Driver"; String username = System.getenv("EXAMPLE_USERNAME_ENV"); String passwd = System.getenv("EXAMPLE_PASSWORD_ENV"); String sourceURL = "jdbc:opengauss://$ip:$port/database"; Connection conn = null; try { // 加载数据库驱动。 Class.forName(driver).newInstance(); } catch (Exception e) { e.printStackTrace(); } try { // 以非加密方式创建数据库连接。 conn = DriverManager.getConnection(sourceURL, username, passwd); System.out.println("Connection succeed!"); } catch (Exception e) { e.printStackTrace(); } // 关闭自动提交。 conn.setAutoCommit(false); // 建表。 Statement st = conn.createStatement(); st.execute("create table mytable (cal1 int);"); // 表中插入200行数据。 PreparedStatement pstm = conn.prepareStatement("insert into mytable values (?)"); for (int i = 0; i < 200; i++) { pstm.setInt(1, i + 1); pstm.addBatch(); } pstm.executeBatch(); conn.commit(); pstm.close(); // 打开游标,每次获取50行数据。 st.setFetchSize(50); int fetchCount = 0; ResultSet rs = st.executeQuery("SELECT * FROM mytable"); while (rs.next()) { fetchCount++; } System.out.println(fetchCount == 200); conn.commit(); rs.close(); // 关闭服务器游标。 st.setFetchSize(0); fetchCount = 0; rs = st.executeQuery("SELECT * FROM mytable"); while (rs.next()) { fetchCount++; } System.out.println(fetchCount == 200); conn.commit(); rs.close(); // 关闭语句对象、关闭数据库连接。 st.close(); conn.close(); } } |
上述示例的运行结果为:
Connection succeed! true true
执行完毕后可执行如下命令恢复自动提交事务。
conn.setAutoCommit(true);