Add UPDATE tab SET ROW (col, ...) = (val, ...) for updating

multiple columns

Susanne Ebrecht

diff --git a/doc/src/sgml/ref/update.sgml b/doc/src/sgml/ref/update.sgml
index 5d1265e945f7ad4e578d5b1635a1ba3a10448307..7b5d17a1fc89dcb4c89f1b87940dfa8080ffbda1 100644 (file)
--- a/doc/src/sgml/ref/update.sgml
+++ b/doc/src/sgml/ref/update.sgml
@@ -1,5 +1,5 @@
 <!--
-$PostgreSQL: pgsql/doc/src/sgml/ref/update.sgml,v 1.38 2006/08/12 02:52:03 tgl Exp $
+$PostgreSQL: pgsql/doc/src/sgml/ref/update.sgml,v 1.39 2006/09/02 20:34:47 momjian Exp $
 PostgreSQL documentation
 -->
 
@@ -21,7 +21,8 @@ PostgreSQL documentation
  <refsynopsisdiv>
 <synopsis>
 UPDATE [ ONLY ] <replaceable class="PARAMETER">table</replaceable> [ [ AS ] <replaceable class="parameter">alias</replaceable> ]
-    SET <replaceable class="PARAMETER">column</replaceable> = { <replaceable class="PARAMETER">expression</replaceable> | DEFAULT } [, ...]
+    [ SET <replaceable class="PARAMETER">column</replaceable> = { <replaceable class="PARAMETER">expression</replaceable> | DEFAULT } [, ...] |
+      SET ( <replaceable class="PARAMETER">column</replaceable> [, ...] ) = ( { <replaceable class="PARAMETER">expression</replaceable> | DEFAULT } [, ...] ) [, ...] ]
     [ FROM <replaceable class="PARAMETER">fromlist</replaceable> ]
     [ WHERE <replaceable class="PARAMETER">condition</replaceable> ]
     [ RETURNING * | <replaceable class="parameter">output_expression</replaceable> [ AS <replaceable class="parameter">output_name</replaceable> ] [, ...] ]
@@ -250,6 +251,10 @@ UPDATE films SET kind = 'Dramatic' WHERE kind = 'Drama';
 <programlisting>
 UPDATE weather SET temp_lo = temp_lo+1, temp_hi = temp_lo+15, prcp = DEFAULT
   WHERE city = 'San Francisco' AND date = '2003-07-03';
+</programlisting>
+<programlisting>
+UPDATE weather SET (temp_lo, temp_hi, prcp) = (temp_lo+1, temp_lo+15, DEFAULT)
+  WHERE city = 'San Francisco' AND date = '2003-07-03';
 </programlisting>
   </para>
 

diff --git a/src/backend/parser/gram.y b/src/backend/parser/gram.y
index f53ba9ce60d21c1f355366cc14cfeedf481b44dd..eab360bc7748feb8679c404b1ba5b0d885606d89 100644 (file)
--- a/src/backend/parser/gram.y
+++ b/src/backend/parser/gram.y
@@ -11,7 +11,7 @@
  *
  *
  * IDENTIFICATION
- *   $PostgreSQL: pgsql/src/backend/parser/gram.y,v 2.560 2006/09/02 18:17:17 momjian Exp $
+ *   $PostgreSQL: pgsql/src/backend/parser/gram.y,v 2.561 2006/09/02 20:34:47 momjian Exp $
  *
  * HISTORY
  *   AUTHOR            DATE            MAJOR EVENT
@@ -237,7 +237,8 @@ static void doNegateFloat(Value *v);
                name_list from_clause from_list opt_array_bounds
                qualified_name_list any_name any_name_list
                any_operator expr_list attrs
-               target_list update_target_list insert_column_list
+               target_list update_col_list update_target_list
+               update_value_list insert_column_list
                values_list def_list indirection opt_indirection
                group_clause TriggerFuncArgs select_limit
                opt_select_limit opclass_item_list
@@ -308,7 +309,8 @@ static void doNegateFloat(Value *v);
 %type <jexpr>  joined_table
 %type <range>  relation_expr
 %type <range>  relation_expr_opt_alias
-%type <target> target_el update_target_el insert_column_item
+%type <target> target_el update_target_el update_col_list_el insert_column_item
+%type <list>   update_target_lists_list update_target_lists_el
 
 %type <typnam> Typename SimpleTypename ConstTypename
                GenericType Numeric opt_float
@@ -5537,6 +5539,20 @@ UpdateStmt: UPDATE relation_expr_opt_alias
                    n->returningList = $7;
                    $$ = (Node *)n;
                }
+            | UPDATE relation_expr_opt_alias
+           SET update_target_lists_list
+           from_clause
+           where_clause
+           returning_clause
+               {
+                   UpdateStmt *n = makeNode(UpdateStmt);
+                   n->relation = $2;
+                   n->targetList = $4;
+                   n->fromClause = $5;
+                   n->whereClause = $6;
+                   n->returningList = $7;
+                   $$ = (Node *)n;
+               }
        ;
 
 
@@ -5941,6 +5957,60 @@ values_item:
            | DEFAULT               { $$ = (Node *) makeNode(SetToDefault); }
        ;
 
+update_target_lists_list:
+           update_target_lists_el { $$ = $1; }
+           | update_target_lists_list ',' update_target_lists_el { $$ = list_concat($1, $3); }
+       ;
+       
+update_target_lists_el:
+           '(' update_col_list ')' '=' '(' update_value_list ')'
+               {
+                   ListCell *col_cell;
+                   ListCell *val_cell;
+
+                   if (list_length($2) != list_length($6))
+                   {
+                       ereport(ERROR, 
+                           (errcode(ERRCODE_SYNTAX_ERROR),
+                            errmsg("number of columns does not match to number of values")));
+                   }
+
+                   for (col_cell = list_head($2), val_cell = list_head($6);
+                        col_cell != NULL && val_cell != NULL;
+                        col_cell = lnext(col_cell), val_cell = lnext(val_cell))
+                   {
+                       /* merge update_value_list with update_col_list */
+                       ResTarget *res_col = (ResTarget *) lfirst(col_cell);
+                       ResTarget *res_val = (ResTarget *) lfirst(val_cell);
+
+                       res_col->val = (Node *)copyObject(res_val);
+                   }
+                   
+                   $$ = $2;
+               }
+       ;
+
+update_col_list:
+           update_col_list_el { $$ = list_make1($1); }
+           | update_col_list ',' update_col_list_el { $$ = lappend($1, $3); }
+       ;
+
+update_col_list_el:
+           ColId opt_indirection
+               {
+                   $$ = makeNode(ResTarget);
+                   $$->name = $1;
+                   $$->indirection = $2;
+                   $$->val = NULL;
+                   $$->location = @1;
+               }
+       ;
+
+update_value_list:
+           values_item { $$ = list_make1($1); }
+           | update_value_list ',' values_item { $$ = lappend($1, $3); }
+       ;
+
 
 /*****************************************************************************
  *
@@ -8253,7 +8323,7 @@ target_el:    a_expr AS ColLabel
        ;
 
 update_target_list:
-           update_target_el                        { $$ = list_make1($1); }
+           update_target_el              { $$ = list_make1($1); }
            | update_target_list ',' update_target_el { $$ = lappend($1,$3); }
        ;