feat: implement addOrUpgradeById methods in DatabaseConnector.java

Panzer1119 · Aug 3, 2021 · af2e4a2 · af2e4a2
1 parent b345bf5
commit af2e4a2
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/main/java/de/codemakers/database/connectors/DatabaseConnector.java b/src/main/java/de/codemakers/database/connectors/DatabaseConnector.java
@@ -154,4 +154,20 @@ public void delete(Object object) {
         HibernateUtil.delete(this, object);
     }
 
+    public <T extends IEntity<Integer, T>> Optional<T> addOrUpgradeById(T entity, Function<Integer, Optional<T>> entityGetterFunction) {
+        return HibernateUtil.addOrUpgradeById(this, entity, entityGetterFunction);
+    }
+
+    public <I, T extends IEntity<I, T>> Optional<T> addOrUpgradeById(T entity, Function<I, Optional<T>> entityGetterFunction, Class<I> idClazz) {
+        return HibernateUtil.addOrUpgradeById(this, entity, entityGetterFunction, idClazz);
+    }
+
+    public <I, T extends IEntity<I, T>> Optional<T> addOrUpgradeById(T entity, Function<I, Optional<T>> entityGetterFunction, Class<I> idClazz, Function<T, I> idGetterFunction) {
+        return HibernateUtil.addOrUpgradeById(this, entity, entityGetterFunction, idClazz, idGetterFunction);
+    }
+
+    public <I, M, T extends IEntity<I, T>> Optional<T> addOrUpgrade(T entity, Function<M, Optional<T>> entityGetterFunction, Class<M> middleClazz, Function<T, M> middleGetterFunction) {
+        return HibernateUtil.addOrUpgrade(this, entity, entityGetterFunction, middleClazz, middleGetterFunction);
+    }
+
 }