Fix: if tableam implement relation_acquire_sample_rows, then just use it

Author: lss602726449

src/backend/commands/analyze.c

@@ -355,11 +355,15 @@ analyze_rel_internal(Oid relid, RangeVar *relation,
 		onerel->rd_rel->relkind == RELKIND_MATVIEW ||
 		onerel->rd_rel->relkind == RELKIND_DIRECTORY_TABLE)
 	{
+		/* Regular table, so we'll use the regular row acquisition function */
+		if (onerel->rd_tableam)
+			acquirefunc = onerel->rd_tableam->relation_acquire_sample_rows;
 		/*
 		 * If the TableAmRoutine's gp_acquire_sample_rows_func if NULL, we use
 		 * gp_acquire_sample_rows_func as default.
 		 */
-		acquirefunc = gp_acquire_sample_rows_func;
+		if (acquirefunc == NULL)
+			acquirefunc = gp_acquire_sample_rows_func;
 
 		/* Also get regular table's size */
 		relpages = AcquireNumberOfBlocks(onerel);
@@ -2079,8 +2083,16 @@ acquire_inherited_sample_rows(Relation onerel, int elevel,
 			childrel->rd_rel->relkind == RELKIND_MATVIEW ||
 			childrel->rd_rel->relkind == RELKIND_DIRECTORY_TABLE)
 		{
-			/* use relation_acquire_sample_rows as default. */
-			acquirefunc = gp_acquire_sample_rows_func;
+			/* Regular table, so use the regular row acquisition function */
+			if (childrel->rd_tableam)
+				acquirefunc = childrel->rd_tableam->relation_acquire_sample_rows;
+
+			/*
+			 * If the TableAmRoutine's relation_acquire_sample_rows if NULL, we use
+			 * relation_acquire_sample_rows as default.
+			 */
+			if (acquirefunc == NULL)
+				acquirefunc = gp_acquire_sample_rows_func;
 
 			relpages = AcquireNumberOfBlocks(childrel);
 		}