Return a bitmap representing generated columns being updated */
| 1376 | |
| 1377 | /* Return a bitmap representing generated columns being updated */ |
| 1378 | Bitmapset * |
| 1379 | ExecGetExtraUpdatedCols(ResultRelInfo *relinfo, EState *estate) |
| 1380 | { |
| 1381 | /* see ExecGetInsertedCols() */ |
| 1382 | if (relinfo->ri_RangeTableIndex != 0) |
| 1383 | { |
| 1384 | RangeTblEntry *rte = exec_rt_fetch(relinfo->ri_RangeTableIndex, estate); |
| 1385 | |
| 1386 | return rte->extraUpdatedCols; |
| 1387 | } |
| 1388 | else if (relinfo->ri_RootResultRelInfo) |
| 1389 | { |
| 1390 | ResultRelInfo *rootRelInfo = relinfo->ri_RootResultRelInfo; |
| 1391 | RangeTblEntry *rte = exec_rt_fetch(rootRelInfo->ri_RangeTableIndex, estate); |
| 1392 | |
| 1393 | if (relinfo->ri_RootToPartitionMap != NULL) |
| 1394 | return execute_attr_map_cols(relinfo->ri_RootToPartitionMap->attrMap, |
| 1395 | rte->extraUpdatedCols); |
| 1396 | else |
| 1397 | return rte->extraUpdatedCols; |
| 1398 | } |
| 1399 | else |
| 1400 | return NULL; |
| 1401 | } |
| 1402 | |
| 1403 | /* Return columns being updated, including generated columns */ |
| 1404 | Bitmapset * |
no test coverage detected