#1261 - ENH: Support use of getGeneratedKeys with SqlUpdate

2024-04-21 10:51:47 +00:00 · 2018-02-16 09:45:49 +13:00
parent 9926cad7f1
commit 5237e8ba66
12 changed files with 315 additions and 13 deletions
@@ -0,0 +1,57 @@
+package org.tests.update;
+
+import io.ebean.annotation.Index;
+import io.ebean.annotation.WhenModified;
+
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.Table;
+import java.time.Instant;
+
+@Entity
+@Table(name = "e_person_online")
+public class EPersonOnline {
+
+  @Id
+  Long id;
+
+  @Index(unique = true)
+  String email;
+
+  boolean online;
+
+  @WhenModified
+  Instant whenUpdated;
+
+  public Long getId() {
+    return id;
+  }
+
+  public void setId(Long id) {
+    this.id = id;
+  }
+
+  public String getEmail() {
+    return email;
+  }
+
+  public void setEmail(String email) {
+    this.email = email;
+  }
+
+  public boolean isOnline() {
+    return online;
+  }
+
+  public void setOnline(boolean online) {
+    this.online = online;
+  }
+
+  public Instant getWhenUpdated() {
+    return whenUpdated;
+  }
+
+  public void setWhenUpdated(Instant whenUpdated) {
+    this.whenUpdated = whenUpdated;
+  }
+}
@@ -0,0 +1,128 @@
+package org.tests.update;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.SqlUpdate;
+import io.ebean.annotation.ForPlatform;
+import io.ebean.annotation.Platform;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestSqlUpdateUpsert extends BaseTestCase {
+
+  @ForPlatform(Platform.H2)
+  @Test
+  public void h2Merge() {
+
+    String sql = "merge into e_person_online (email, online, when_updated) key(email) values (?, ?, now())";
+
+    String email = "baz@one.com";
+
+    Object key = Ebean.createSqlUpdate(sql)
+      .setGetGeneratedKeys(true)
+      .setParameter(1, email)
+      .setParameter(2, true)
+      .executeGetKey();
+
+    EPersonOnline found = Ebean.find(EPersonOnline.class, key);
+    assertThat(found).isNotNull();
+    assertThat(found.getEmail()).isEqualTo(email);
+    assertThat(found.isOnline()).isTrue();
+
+    String sqlNamed = "merge into e_person_online (email, online, when_updated) key(email) values (:email, :online, now())";
+
+    SqlUpdate sqlUpdate2 = Ebean.createSqlUpdate(sqlNamed)
+      .setGetGeneratedKeys(true)
+      .setParameter("email", email)
+      .setParameter("online", false);
+
+    Object key2 = sqlUpdate2.executeGetKey();
+    assertThat(key2).isNull();
+
+
+    EPersonOnline found2 = Ebean.find(EPersonOnline.class).where().eq("email", email).findOne();
+    assertThat(found2).isNotNull();
+    assertThat(found2.getId()).isEqualTo(key);
+    assertThat(found2.getEmail()).isEqualTo(email);
+    assertThat(found2.isOnline()).isFalse();
+    assertThat(found2.getWhenUpdated()).isGreaterThan(found.getWhenUpdated());
+  }
+
+  @ForPlatform(Platform.POSTGRES)
+  @Test
+  public void postgresUpsert() {
+
+    String sql = "insert into e_person_online (email, online, when_updated) values (?, ?, now()) on conflict (email) do update set when_updated=now(), online = ?";
+
+    String email = "foo@one.com";
+
+    Object key = Ebean.createSqlUpdate(sql)
+      .setGetGeneratedKeys(true)
+      .setParameter(1, email)
+      .setParameter(2, true)
+      .setParameter(3, true)
+      .executeGetKey();
+
+    EPersonOnline found = Ebean.find(EPersonOnline.class, key);
+    assertThat(found).isNotNull();
+    assertThat(found.getEmail()).isEqualTo("foo@one.com");
+    assertThat(found.isOnline()).isTrue();
+
+
+    String sqlNamed = "insert into e_person_online (email, online, when_updated) values (:email, :online, now()) on conflict (email) do update set when_updated=now(), online = :online";
+    SqlUpdate sqlUpdate2 = Ebean.createSqlUpdate(sqlNamed)
+      .setGetGeneratedKeys(true)
+      .setParameter("email", email)
+      .setParameter("online", false);
+
+    Object key2 = sqlUpdate2.executeGetKey();
+
+    EPersonOnline found2 = Ebean.find(EPersonOnline.class, key2);
+    assertThat(found2).isNotNull();
+    assertThat(found2.getId()).isEqualTo(key);
+    assertThat(found2.getEmail()).isEqualTo("foo@one.com");
+    assertThat(found2.isOnline()).isFalse();
+    assertThat(found2.getWhenUpdated()).isGreaterThan(found.getWhenUpdated());
+
+  }
+
+  @ForPlatform(Platform.MYSQL)
+  @Test
+  public void mySqlUpsert() {
+
+    String email = "bar@one.com";
+
+    String sql = "insert into e_person_online (email, online, when_updated) values (?, ?, current_time) on duplicate key update when_updated=current_time, online = ?";
+    SqlUpdate sqlUpdate = Ebean.createSqlUpdate(sql)
+      .setGetGeneratedKeys(true)
+      .setParameter(1, email)
+      .setParameter(2, true)
+      .setParameter(3, true);
+
+    Object key = sqlUpdate.executeGetKey();
+    assertThat(key).isNotNull();
+
+    EPersonOnline found = Ebean.find(EPersonOnline.class, key);
+    assertThat(found).isNotNull();
+    assertThat(found.getEmail()).isEqualTo("bar@one.com");
+    assertThat(found.isOnline()).isTrue();
+
+
+    String sqlNamed = "insert into e_person_online (email, online, when_updated) values (:email, :online, current_time) on duplicate key update when_updated=current_time, online = :online";
+    SqlUpdate sqlUpdate2 = Ebean.createSqlUpdate(sqlNamed)
+      .setGetGeneratedKeys(true)
+      .setParameter("email", email)
+      .setParameter("online", false);
+
+    sqlUpdate2.execute();
+    Object key2 = sqlUpdate2.getGeneratedKey();
+
+    EPersonOnline found2 = Ebean.find(EPersonOnline.class, key2);
+    assertThat(found2).isNotNull();
+    assertThat(found2.getId()).isEqualTo(key);
+    assertThat(found2.getEmail()).isEqualTo("bar@one.com");
+    assertThat(found2.isOnline()).isFalse();
+
+  }
+}