CRM_Utils_SQL_Select - Allow fluent query execution

CRM/Core/DAO.php

@@ -1032,6 +1032,22 @@ public function fetchAll() {
     return $result;
   }
 
+  /**
+   * Returns a singular value.
+   *
+   * @return mixed|NULL
+   */
+  public function fetchValue() {
+    $result = $this->getDatabaseResult();
+    $row = $result->fetchRow();
+    $ret = NULL;
+    if ($row) {
+      $ret = $row[0];
+    }
+    $this->free();
+    return $ret;
+  }
+
   /**
    * Get all the result records as mapping between columns.
    *

CRM/Utils/SQL/Select.php

@@ -593,6 +593,25 @@ public function toSQL() {
     return $sql;
   }
 
+  /**
+   * @return CRM_Core_DAO
+   */
+  public function execute($daoName = NULL, $i18nRewrite = TRUE) {
+    // Don't pass through $params. toSQL() handles interpolation.
+    $params = array();
+
+    // Don't pass through $abort, $trapException. Just use straight-up exceptions.
+    $abort = TRUE;
+    $trapException = FALSE;
+    $errorScope = CRM_Core_TemporaryErrorScope::useException();
+
+    // Don't pass through freeDAO. You can do it yourself.
+    $freeDAO = FALSE;
+
+    return CRM_Core_DAO::executeQuery($this->toSQL(), $params, $abort, $daoName,
+      $freeDAO, $i18nRewrite, $trapException);
+  }
+
   /**
    * Has an offset been set.
    *

tests/phpunit/CRM/Utils/SQL/SelectTest.php

@@ -10,6 +10,39 @@ public function testGetDefault() {
     $this->assertLike('SELECT * FROM foo bar', $select->toSQL());
   }
 
+  public function testExecute_OK() {
+    // We need some SQL query.
+    $select = CRM_Utils_SQL_Select::from('civicrm_contact')
+      ->select('count(*) as cnt');
+    $this->assertLike('SELECT count(*) as cnt FROM civicrm_contact', $select->toSQL());
+
+    // Try with typical fetch().
+    $rows = 0;
+    $dao = $select->execute();
+    while ($dao->fetch()) {
+      $rows++;
+      $this->assertTrue(is_numeric($dao->cnt), "Expect query to execute");
+    }
+    $this->assertEquals(1, $rows);
+
+    // Try with fetchValue().
+    $this->assertTrue(is_numeric($select->execute()->fetchValue()));
+
+    // Try with fetchAll()
+    $records = $select->execute()->fetchAll();
+    $this->assertTrue(is_numeric($records[0]['cnt']));
+  }
+
+  public function testExecute_Error() {
+    try {
+      CRM_Utils_SQL_Select::from('civicrm_contact')->select('snarb;barg')->execute();
+      $this->fail('Expected an exception');
+    }
+    catch (PEAR_Exception $e) {
+      $this->assertTrue(TRUE, "Received expected exception");
+    }
+  }
+
   public function testGetFields() {
     $select = CRM_Utils_SQL_Select::from('foo')
       ->select('bar')