feat: add support for cockroachdb

This commit is contained in:
Lakhan Samani 2022-06-09 23:43:21 +05:30
parent e0d8644264
commit dff50097e8
2 changed files with 3 additions and 1 deletions

View File

@ -21,4 +21,6 @@ const (
DbTypeCassandraDB = "cassandradb" DbTypeCassandraDB = "cassandradb"
// DbTypeScyllaDB is the scylla database type // DbTypeScyllaDB is the scylla database type
DbTypeScyllaDB = "scylladb" DbTypeScyllaDB = "scylladb"
// DbTypeCockroachDB is the cockroach database type
DbTypeCockroachDB = "cockroachdb"
) )

View File

@ -46,7 +46,7 @@ func NewProvider() (*provider, error) {
dbURL := memorystore.RequiredEnvStoreObj.GetRequiredEnv().DatabaseURL dbURL := memorystore.RequiredEnvStoreObj.GetRequiredEnv().DatabaseURL
switch dbType { switch dbType {
case constants.DbTypePostgres, constants.DbTypeYugabyte: case constants.DbTypePostgres, constants.DbTypeYugabyte, constants.DbTypeCockroachDB:
sqlDB, err = gorm.Open(postgres.Open(dbURL), ormConfig) sqlDB, err = gorm.Open(postgres.Open(dbURL), ormConfig)
case constants.DbTypeSqlite: case constants.DbTypeSqlite:
sqlDB, err = gorm.Open(sqlite.Open(dbURL), ormConfig) sqlDB, err = gorm.Open(sqlite.Open(dbURL), ormConfig)