Fix copy to replicated table display copy number is incorrect.

Reproduece SQL:
CREATE TABLE test_rep(a int) DISTRIBUTED REPLICATED;
INSERT INTO test_rep SELECT i FROM generate_series(1, 99) i;
COPY test_rep TO 'tmp/rep.csv';
COPY test_rep FROM 'tmp/rep.csv';

The display copy row numbers will be RowNums * numsegments which is
incorrect. In this commit, we fix this.

Authored-by: Zhang Wenchao [email protected]

Author: zhangwenchao-123

src/backend/commands/copy.c

@@ -440,6 +440,11 @@ DoCopy(ParseState *pstate, const CopyStmt *stmt,
 				*processed = CopyDispatchOnSegment(stmt);
 			else
 				*processed = CopyFrom(cstate);	/* copy from file to database */
+
+			/* Handle copy to replicated table returns processed number */
+			if (Gp_role == GP_ROLE_DISPATCH && cstate->rel->rd_cdbpolicy &&
+				cstate->rel->rd_cdbpolicy->ptype == POLICYTYPE_REPLICATED)
+				*processed = *processed / cstate->rel->rd_cdbpolicy->numsegments;
 		}
 		PG_CATCH();
 		{

src/test/isolation2/expected/ao_unique_index.out

@@ -390,20 +390,20 @@ CREATE
 1: BEGIN;
 BEGIN
 1: COPY unique_index_ao_row FROM PROGRAM 'seq 1 10';
-COPY 30
+COPY 10
 -- concurrent tx inserting conflicting row should block.
 2&: COPY unique_index_ao_row FROM PROGRAM 'seq 1 1';  <waiting ...>
 -- concurrent tx inserting non-conflicting rows should be successful.
 3: COPY unique_index_ao_row FROM PROGRAM 'seq 11 20';
-COPY 30
+COPY 10
 -- inserting a conflicting row in the same transaction should ERROR out.
 1: COPY unique_index_ao_row FROM PROGRAM 'seq 1 1';
 ERROR:  duplicate key value violates unique constraint "unique_index_ao_row_a_key"
 DETAIL:  Key (a)=(1) already exists.
 CONTEXT:  COPY unique_index_ao_row, line 1
 -- now that tx 1 was aborted, tx 2 is successful.
 2<:  <... completed>
-COPY 3
+COPY 1
 1: END;
 END
 

src/test/isolation2/expected/aocs_unique_index.out

@@ -390,20 +390,20 @@ CREATE
 1: BEGIN;
 BEGIN
 1: COPY unique_index_ao_column FROM PROGRAM 'seq 1 10';
-COPY 30
+COPY 10
 -- concurrent tx inserting conflicting row should block.
 2&: COPY unique_index_ao_column FROM PROGRAM 'seq 1 1';  <waiting ...>
 -- concurrent tx inserting non-conflicting rows should be successful.
 3: COPY unique_index_ao_column FROM PROGRAM 'seq 11 20';
-COPY 30
+COPY 10
 -- inserting a conflicting row in the same transaction should ERROR out.
 1: COPY unique_index_ao_column FROM PROGRAM 'seq 1 1';
 ERROR:  duplicate key value violates unique constraint "unique_index_ao_column_a_key"
 DETAIL:  Key (a)=(1) already exists.
 CONTEXT:  COPY unique_index_ao_column, line 1
 -- now that tx 1 was aborted, tx 2 is successful.
 2<:  <... completed>
-COPY 3
+COPY 1
 1: END;
 END