Add support for yugabyte

Resolves #119
This commit is contained in:
Lakhan Samani 2022-02-12 13:19:31 +05:30
parent 51532657d7
commit 6c2a4c3256
2 changed files with 3 additions and 2 deletions

View File

@ -13,4 +13,6 @@ const (
DbTypeArangodb = "arangodb" DbTypeArangodb = "arangodb"
// DbTypeMongodb is the mongodb database type // DbTypeMongodb is the mongodb database type
DbTypeMongodb = "mongodb" DbTypeMongodb = "mongodb"
// DbTypeYugabyte is the yugabyte database type
DbTypeYugabyte = "yugabyte"
) )

View File

@ -41,9 +41,8 @@ func NewProvider() (*provider, error) {
TablePrefix: models.Prefix, TablePrefix: models.Prefix,
}, },
} }
switch envstore.EnvInMemoryStoreObj.GetStringStoreEnvVariable(constants.EnvKeyDatabaseType) { 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) sqlDB, err = gorm.Open(postgres.Open(envstore.EnvInMemoryStoreObj.GetStringStoreEnvVariable(constants.EnvKeyDatabaseURL)), ormConfig)
break break
case constants.DbTypeSqlite: case constants.DbTypeSqlite: