Revert API part of "Add page_props table access class"
authorAnomie <bjorsch@wikimedia.org>
Tue, 19 Jan 2016 17:49:01 +0000 (17:49 +0000)
committerBrad Jorsch <bjorsch@wikimedia.org>
Tue, 19 Jan 2016 17:49:43 +0000 (12:49 -0500)
This partially reverts commit e48030a7aebb47eeb702d351716ba8304308b02f.

Change-Id: I31869acdaed8c2f5acfd9780887fbd3ab175bbe9

includes/api/ApiQueryPageProps.php

index c2a8df7..1f992f8 100644 (file)
@@ -40,44 +40,63 @@ class ApiQueryPageProps extends ApiQueryBase {
        public function execute() {
                # Only operate on existing pages
                $pages = $this->getPageSet()->getGoodTitles();
+               if ( !count( $pages ) ) {
+                       # Nothing to do
+                       return;
+               }
 
                $this->params = $this->extractRequestParams();
+
+               $this->addTables( 'page_props' );
+               $this->addFields( array( 'pp_page', 'pp_propname', 'pp_value' ) );
+               $this->addWhereFld( 'pp_page', array_keys( $pages ) );
+
                if ( $this->params['continue'] ) {
-                       $continueValue = intval( $this->params['continue'] );
-                       $filteredPages = array();
-                       foreach ( $pages as $id => $page ) {
-                               if ( $id >= $continueValue ) {
-                                       $filteredPages[$id] = $page;
-                               }
-                       }
-                       $pages = $filteredPages;
+                       $this->addWhere( 'pp_page >=' . intval( $this->params['continue'] ) );
                }
 
-               if ( !count( $pages ) ) {
-                       # Nothing to do
-                       return;
+               if ( $this->params['prop'] ) {
+                       $this->addWhereFld( 'pp_propname', $this->params['prop'] );
+               }
+
+               # Force a sort order to ensure that properties are grouped by page
+               # But only if pp_page is not constant in the WHERE clause.
+               if ( count( $pages ) > 1 ) {
+                       $this->addOption( 'ORDER BY', 'pp_page' );
                }
 
-               $pageProps = PageProps::getInstance();
+               $res = $this->select( __METHOD__ );
+               $currentPage = 0; # Id of the page currently processed
                $props = array();
                $result = $this->getResult();
-               if ( $this->params['prop'] ) {
-                       $propnames = $this->params['prop'];
-                       $properties = array();
-                       foreach ( $propnames as $propname ) {
-                               $values = $pageProps->getProperty( $pages, $propname );
-                               foreach ( $values as $page => $value ) {
-                                       if ( !isset( $properties[$page] ) ) {
-                                               $properties[$page] = array();
+
+               foreach ( $res as $row ) {
+                       if ( $currentPage != $row->pp_page ) {
+                               # Different page than previous row, so add the properties to
+                               # the result and save the new page id
+
+                               if ( $currentPage ) {
+                                       if ( !$this->addPageProps( $result, $currentPage, $props ) ) {
+                                               # addPageProps() indicated that the result did not fit
+                                               # so stop adding data. Reset props so that it doesn't
+                                               # get added again after loop exit
+
+                                               $props = array();
+                                               break;
                                        }
-                                       $properties[$page][$propname] = $value;
+
+                                       $props = array();
                                }
+
+                               $currentPage = $row->pp_page;
                        }
-               } else {
-                       $properties = $pageProps->getProperties( $pages );
+
+                       $props[$row->pp_propname] = $row->pp_value;
                }
-               foreach ( $properties as $page => $props ) {
-                       $this->addPageProps( $result, $page, $props );
+
+               if ( count( $props ) ) {
+                       # Add any remaining properties to the results
+                       $this->addPageProps( $result, $currentPage, $props );
                }
        }