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: