Add primaryKey info to daos

civicrm · Jul 4, 2021 · eda5da7 · eda5da7
1 parent b675a45
commit eda5da7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/CRM/Core/DAO.php b/CRM/Core/DAO.php
@@ -32,6 +32,13 @@
  */
 class CRM_Core_DAO extends DB_DataObject {
 
+  /**
+   * Primary key field(s).
+   *
+   * @var string[]
+   */
+  public static $_primaryKey = ['id'];
+
   /**
    * How many times has this instance been cloned.
    *

diff --git a/Civi/Api4/Generic/AbstractEntity.php b/Civi/Api4/Generic/AbstractEntity.php
@@ -152,6 +152,7 @@ public static function getInfo() {
       $dao = \CRM_Core_DAO_AllCoreTables::getFullName($info['name']);
       if ($dao) {
         $info['paths'] = $dao::getEntityPaths();
+        $info['primary_key'] = $dao::$_primaryKey;
         $info['icon'] = $dao::$_icon;
         $info['label_field'] = $dao::$_labelField;
         $info['dao'] = $dao;

diff --git a/xml/templates/dao.tpl b/xml/templates/dao.tpl
@@ -24,6 +24,16 @@ class {$table.className} extends CRM_Core_DAO {ldelim}
       */
       public static $_tableName = '{$table.name}';
 
+   {* Only print this variable if it's different than the default in CRM_Core_DAO *}
+   {if count($table.primaryKey.field) !== 1 || $table.primaryKey.field.0 !== 'id'}
+     /**
+      * Primary key field(s).
+      *
+      * @var string[]
+      */
+      public static $_primaryKey = [{if $table.primaryKey.field}'{"', '"|implode:$table.primaryKey.field}'{/if}];
+   {/if}
+
    {if $table.icon}
      /**
       * Icon associated with this entity.