implemented refresh row

Dave Cramer · Dave Cramer · commit 0daee96ed18c · 2002-06-16T18:26:53.000Z
diff --git a/src/interfaces/jdbc/org/postgresql/jdbc2/UpdateableResultSet.java b/src/interfaces/jdbc/org/postgresql/jdbc2/UpdateableResultSet.java
@@ -76,7 +76,10 @@ Object getValue() throws SQLException
 	protected java.sql.PreparedStatement deleteStatement = null;
 
 
-  private java.sql.Statement currentStatement = null;
+  /**
+   * PreparedStatement used to refresh data
+   */
+  private java.sql.PreparedStatement selectStatement = null;
 
 
 	/**
@@ -617,6 +620,78 @@ public synchronized void updateObject(int columnIndex, Object x, int scale) thro
 
 	}
 
+
+  public void refreshRow() throws SQLException
+  {
+		if ( !isUpdateable() )
+    {
+			throw new PSQLException( "postgresql.updateable.notupdateable" );
+		}
+
+    try
+    {
+      StringBuffer selectSQL = new StringBuffer( "select ");
+
+      final int numColumns = java.lang.reflect.Array.getLength(fields);
+
+      for (int i=0; i < numColumns ; i++ )
+      {
+
+        selectSQL.append( fields[i].getName() );
+
+        if ( i < numColumns - 1 )
+        {
+
+          selectSQL.append(", ");
+        }
+
+      }
+      selectSQL.append(" from " ).append(tableName).append(" where ");
+
+      int numKeys = primaryKeys.size();
+
+      for ( int i = 0; i < numKeys; i++ )
+      {
+
+        PrimaryKey primaryKey = ((PrimaryKey)primaryKeys.get(i));
+        selectSQL.append(primaryKey.name).append("= ?");
+
+        if ( i < numKeys -1 )
+        {
+          selectSQL.append(" and ");
+        }
+      }
+      if ( Driver.logDebug ) Driver.debug("selecting "+ selectSQL.toString());
+      selectStatement = ((java.sql.Connection)connection).prepareStatement(selectSQL.toString());
+
+
+      for( int j=0, i=1; j < numKeys; j++, i++)
+      {
+        selectStatement.setObject( i, ((PrimaryKey)primaryKeys.get(j)).getValue() );
+      }
+
+      org.postgresql.jdbc2.ResultSet rs = (org.postgresql.jdbc2.ResultSet) selectStatement.executeQuery();
+
+      if( rs.first() )
+      {
+        rowBuffer = rs.rowBuffer;
+      }
+
+      rows.setElementAt( rowBuffer, current_row );
+      if ( Driver.logDebug ) Driver.debug("done updates");
+
+      rs.close();
+      selectStatement.close();
+      selectStatement = null;
+
+    }
+    catch (Exception e)
+    {
+      if ( Driver.logDebug ) Driver.debug(e.getClass().getName()+e);
+      throw new SQLException( e.getMessage() );
+    }
+
+  }
   /**
    *
    * @throws SQLException
diff --git a/src/interfaces/jdbc/org/postgresql/test/jdbc2/UpdateableResultTest.java b/src/interfaces/jdbc/org/postgresql/test/jdbc2/UpdateableResultTest.java
@@ -57,7 +57,7 @@ public void testUpdateable()
         rs.updateString("name", "paul");
 
         rs.insertRow();
-
+        rs.refreshRow();
         assertTrue( rs.getInt("id") == 3 );
         assertTrue( rs.getString("name").equals("paul"));
         assertTrue( rs.getString("notselected") == null );
