EclipseLink: add PrimaryKey to policy mapping records JPA model (#1403)

Author: HonahX

extension/persistence/jpa-model/build.gradle.kts

@@ -24,6 +24,8 @@ plugins {
 
 dependencies {
   implementation(project(":polaris-core"))
+  implementation(libs.eclipselink)
+
   compileOnly(libs.jakarta.validation.api)
   compileOnly(libs.jakarta.annotation.api)
   compileOnly(libs.jakarta.persistence.api)

extension/persistence/jpa-model/src/main/java/org/apache/polaris/jpa/models/ModelPolicyMappingRecord.java

@@ -18,12 +18,14 @@
  */
 package org.apache.polaris.jpa.models;
 
+import jakarta.persistence.Column;
 import jakarta.persistence.Entity;
 import jakarta.persistence.Id;
 import jakarta.persistence.Index;
 import jakarta.persistence.Table;
 import jakarta.persistence.Version;
 import org.apache.polaris.core.policy.PolarisPolicyMappingRecord;
+import org.eclipse.persistence.annotations.PrimaryKey;
 
 @Entity
 @Table(
@@ -33,6 +35,14 @@
           name = "POLICY_MAPPING_RECORDS_BY_POLICY_INDEX",
           columnList = "policyCatalogId,policyId,targetCatalogId,targetId")
     })
+@PrimaryKey(
+    columns = {
+      @Column(name = "targetCatalogId"),
+      @Column(name = "targetId"),
+      @Column(name = "policyTypeCode"),
+      @Column(name = "policyCatalogId"),
+      @Column(name = "policyId")
+    })
 public class ModelPolicyMappingRecord {
   // id of the catalog where target entity resides
   @Id private long targetCatalogId;

getting-started/assets/eclipselink/persistence.xml

@@ -28,6 +28,7 @@
     <class>org.apache.polaris.jpa.models.ModelEntityChangeTracking</class>
     <class>org.apache.polaris.jpa.models.ModelEntityDropped</class>
     <class>org.apache.polaris.jpa.models.ModelGrantRecord</class>
+    <class>org.apache.polaris.jpa.models.ModelPolicyMappingRecord</class>
     <class>org.apache.polaris.jpa.models.ModelPrincipalSecrets</class>
     <class>org.apache.polaris.jpa.models.ModelSequenceId</class>
     <shared-cache-mode>NONE</shared-cache-mode>