#107 - Mapping - Add support for @PrimaryKeyJoinColumn/@PrimaryKeyJoinColumns

Adds Bi-directional support

src/test/java/org/tests/model/onetoone/OtoUBPrime.java

@@ -0,0 +1,65 @@
+package org.tests.model.onetoone;
+
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.OneToOne;
+import javax.persistence.Version;
+import java.util.UUID;
+
+@Entity
+public class OtoUBPrime {
+
+  @Id
+  UUID pid;
+
+  String name;
+
+  /**
+   * Master side of bi-directional PrimaryJoinColumn.
+   */
+  @OneToOne(mappedBy = "prime")
+  OtoUBPrimeExtra extra;
+
+  @Version
+  Long version;
+
+  public OtoUBPrime(String name) {
+    this.name = name;
+  }
+
+  public String toString() {
+    return "id:"+ pid +" name:"+name+" extra:"+extra;
+  }
+
+  public UUID getPid() {
+    return pid;
+  }
+
+  public void setPid(UUID pid) {
+    this.pid = pid;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  public void setName(String name) {
+    this.name = name;
+  }
+
+  public OtoUBPrimeExtra getExtra() {
+    return extra;
+  }
+
+  public void setExtra(OtoUBPrimeExtra extra) {
+    this.extra = extra;
+  }
+
+  public Long getVersion() {
+    return version;
+  }
+
+  public void setVersion(Long version) {
+    this.version = version;
+  }
+}

src/test/java/org/tests/model/onetoone/OtoUBPrimeExtra.java

@@ -0,0 +1,67 @@
+package org.tests.model.onetoone;
+
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.OneToOne;
+import javax.persistence.PrimaryKeyJoinColumn;
+import javax.persistence.Version;
+import java.util.UUID;
+
+@Entity
+public class OtoUBPrimeExtra {
+
+  @Id
+  UUID eid;
+
+  String extra;
+
+  /**
+   * Child side of bi-directional PrimaryJoinColumn.
+   */
+  @OneToOne
+  @PrimaryKeyJoinColumn
+  OtoUBPrime prime;
+
+  @Version
+  Long version;
+
+  public OtoUBPrimeExtra(String extra) {
+    this.extra = extra;
+  }
+
+  public String toString() {
+    return "exId:" + eid + " " + extra;
+  }
+
+  public UUID getEid() {
+    return eid;
+  }
+
+  public void setEid(UUID eid) {
+    this.eid = eid;
+  }
+
+  public String getExtra() {
+    return extra;
+  }
+
+  public void setExtra(String extra) {
+    this.extra = extra;
+  }
+
+  public OtoUBPrime getPrime() {
+    return prime;
+  }
+
+  public void setPrime(OtoUBPrime prime) {
+    this.prime = prime;
+  }
+
+  public Long getVersion() {
+    return version;
+  }
+
+  public void setVersion(Long version) {
+    this.version = version;
+  }
+}

src/test/java/org/tests/model/onetoone/TestOneToOnePrimaryKeyJoinBidi.java

@@ -0,0 +1,87 @@
+package org.tests.model.onetoone;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.Query;
+import org.ebeantest.LoggedSqlCollector;
+import org.junit.Test;
+
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestOneToOnePrimaryKeyJoinBidi extends BaseTestCase {
+
+
+  private OtoUBPrime insert(String desc) {
+    OtoUBPrime prime = new OtoUBPrime("u" + desc);
+    OtoUBPrimeExtra extra = new OtoUBPrimeExtra("v" + desc);
+    prime.setExtra(extra);
+    Ebean.save(prime);
+    return prime;
+  }
+
+  @Test
+  public void insertUpdateDelete() {
+
+    String desc = "" + System.currentTimeMillis();
+    OtoUBPrime p1 = insert(desc);
+
+    assertThat(p1.getExtra().getEid()).isEqualTo(p1.getPid()).as("Same id values");
+
+
+    Query<OtoUBPrime> query = Ebean.find(OtoUBPrime.class).setId(p1.getPid());
+
+    OtoUBPrime found = query.findOne();
+
+    assertThat(found).isNotNull();
+    assertThat(sqlOf(query)).contains("select t0.pid, t0.name, t0.version, t0.pid from oto_ubprime t0 where t0.pid = ?")
+      .as("we don't join to oto_ubprime_extra");
+
+    assertThat(found.getName()).isEqualTo("u" + desc);
+
+    Query<OtoUBPrime> queryWithFetch = Ebean.find(OtoUBPrime.class)
+      .setId(p1.getPid())
+      .fetch("extra");
+
+    OtoUBPrime oneWith = queryWithFetch.findOne();
+
+    assertThat(oneWith).isNotNull();
+    assertThat(sqlOf(queryWithFetch)).contains("select t0.pid, t0.name, t0.version, t1.eid, t1.extra, t1.version, t1.eid from oto_ubprime t0 left join oto_ubprime_extra t1 on t1.eid = t0.pid  where t0.pid = ?")
+      .as("we join to oto_prime_extra");
+
+
+    assertThat(oneWith.getExtra().getExtra()).isEqualTo("v" + desc);
+
+    thenUpdate(oneWith);
+    thenDelete(found);
+  }
+
+  private void thenUpdate(OtoUBPrime oneWith) {
+
+    OtoUBPrimeExtra extra = oneWith.getExtra();
+    extra.setExtra("modified");
+
+    Ebean.save(oneWith);
+
+    extra.setExtra("mod2");
+    oneWith.setName("mod2");
+
+    Ebean.save(oneWith);
+  }
+
+  private void thenDelete(OtoUBPrime found) {
+
+
+    OtoUBPrime bean = Ebean.find(OtoUBPrime.class, found.getPid());
+
+    LoggedSqlCollector.start();
+    Ebean.delete(bean);
+
+    List<String> sql = LoggedSqlCollector.stop();
+    assertThat(sql).hasSize(2);
+    assertThat(sql.get(0)).contains("delete from oto_ubprime_extra where");
+    assertThat(sql.get(1)).contains("delete from oto_ubprime where");
+
+  }
+}