fixed some comments

flow/activities/flowable.go

@@ -658,8 +658,7 @@ func (a *FlowableActivity) replicateQRepPartition(ctx context.Context,
 	errGroup, errCtx := errgroup.WithContext(ctx)
 	stream := model.NewQRecordStream(bufferSize)
 	errGroup.Go(func() error {
-		pgConn := srcConn.(*connpostgres.PostgresConnector)
-		tmp, err := pgConn.PullQRepRecordStream(errCtx, config, partition, stream)
+		tmp, err := srcConn.PullQRepRecordStream(errCtx, config, partition, stream)
 		numRecords := int64(tmp)
 		if err != nil {
 			a.Alerter.LogFlowError(ctx, config.FlowJobName, err)

flow/cmd/peer_data.go

@@ -201,17 +201,11 @@ func (h *FlowRequestHandler) GetAllTables(
 		if err != nil {
 			return &protos.AllTablesResponse{Tables: nil}, err
 		}
-
 		defer rows.Close()
-		var tables []string
-		for rows.Next() {
-			var table pgtype.Text
-			err := rows.Scan(&table)
-			if err != nil {
-				return &protos.AllTablesResponse{Tables: nil}, err
-			}
-
-			tables = append(tables, table.String)
+
+		tables, err := pgx.CollectRows[string](rows, pgx.RowTo)
+		if err != nil {
+			return nil, fmt.Errorf("failed to fetch all tables: %w", err)
 		}
 		return &protos.AllTablesResponse{Tables: tables}, nil
 	}

flow/connectors/core.go

@@ -142,7 +142,7 @@ type QRepPullConnector interface {
 	// GetQRepPartitions returns the partitions for a given table that haven't been synced yet.
 	GetQRepPartitions(ctx context.Context, config *protos.QRepConfig, last *protos.QRepPartition) ([]*protos.QRepPartition, error)
 
-	// PullQRepRecords returns the records for a given partition.
+	// PullQRepRecordStream streams the records for a given partition.
 	PullQRepRecordStream(ctx context.Context, config *protos.QRepConfig,
 		partition *protos.QRepPartition, stream *model.QRecordStream) (int, error)
 }

flow/connectors/postgres/qrep_query_executor.go

@@ -226,7 +226,7 @@ func (qe *QRepQueryExecutor) processFetchedRows(
 	defer func() {
 		rows.Close()
 
-		// description of .Close() says it should only be called once rows are closed
+		// description of .Err() says it should only be called once rows are closed
 		if rows.Err() != nil {
 			stream.Records <- model.QRecordOrError{
 				Err: rows.Err(),