Add support for yugabyte

lakhansamani · lakhansamani · commit 6c2a4c3256b2 · 2022-02-12T13:19:31.000+05:30
Resolves #119
diff --git 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
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -13,4 +13,6 @@ const (`
`13`	`13`	`DbTypeArangodb = "arangodb"`
`14`	`14`	`// DbTypeMongodb is the mongodb database type`
`15`	`15`	`DbTypeMongodb = "mongodb"`
	`16`	`+ // DbTypeYugabyte is the yugabyte database type`
	`17`	`+ DbTypeYugabyte = "yugabyte"`
`16`	`18`	`)`