From 6c2a4c3256b21e3be6f9f305277a6445bcd8c9a3 Mon Sep 17 00:00:00 2001
From: Lakhan Samani <lakhan.m.samani@gmail.com>
Date: Sat, 12 Feb 2022 13:19:31 +0530
Subject: [PATCH] Add support for yugabyte

Resolves #119
---
 server/constants/db_types.go   | 2 ++
 server/db/providers/sql/sql.go | 3 +--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/server/constants/db_types.go b/server/constants/db_types.go
index 9cdef99..190e174 100644
--- a/server/constants/db_types.go
+++ b/server/constants/db_types.go
@@ -13,4 +13,6 @@ const (
 	DbTypeArangodb = "arangodb"
 	// DbTypeMongodb is the mongodb database type
 	DbTypeMongodb = "mongodb"
+	// DbTypeYugabyte is the yugabyte database type
+	DbTypeYugabyte = "yugabyte"
 )
diff --git a/server/db/providers/sql/sql.go b/server/db/providers/sql/sql.go
index a75408a..f9f4713 100644
--- a/server/db/providers/sql/sql.go
+++ b/server/db/providers/sql/sql.go
@@ -41,9 +41,8 @@ func NewProvider() (*provider, error) {
 			TablePrefix: models.Prefix,
 		},
 	}
-
 	switch envstore.EnvInMemoryStoreObj.GetStringStoreEnvVariable(constants.EnvKeyDatabaseType) {
-	case constants.DbTypePostgres:
+	case constants.DbTypePostgres, constants.DbTypeYugabyte:
 		sqlDB, err = gorm.Open(postgres.Open(envstore.EnvInMemoryStoreObj.GetStringStoreEnvVariable(constants.EnvKeyDatabaseURL)), ormConfig)
 		break
 	case constants.DbTypeSqlite: