From 28a1a481c7d95de382c1285ece832655ab1964cc Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Mon, 14 Aug 2023 10:03:05 +0100
Subject: [PATCH] Register YugabyteDB extension.

---
 src/main/java/me/kavin/piped/utils/LiquibaseHelper.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/java/me/kavin/piped/utils/LiquibaseHelper.java b/src/main/java/me/kavin/piped/utils/LiquibaseHelper.java
index f412ff8..a97d09c 100644
--- a/src/main/java/me/kavin/piped/utils/LiquibaseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/LiquibaseHelper.java
@@ -26,6 +26,9 @@ public class LiquibaseHelper {
         // ensure postgres driver is loaded
         DriverManager.registerDriver(new org.postgresql.Driver());
 
+        // register YugabyteDB database
+        DatabaseFactory.getInstance().register(new liquibase.ext.yugabytedb.database.YugabyteDBDatabase());
+
         Database database = DatabaseFactory.getInstance().findCorrectDatabaseImplementation(new JdbcConnection(DriverManager.getConnection(url, username, password)));
 
         try (Liquibase liquibase = new Liquibase("changelog/db.changelog-master.xml", new ClassLoaderResourceAccessor(), database)) {