Files
authorizer/server/db/providers/sql/provider.go

72 lines
2.0 KiB
Go
Raw Permalink Normal View History

2022-01-21 12:18:07 +05:30
package sql
import (
2022-02-05 09:00:56 +05:30
"log"
"os"
"time"
2022-01-21 12:18:07 +05:30
"github.com/authorizerdev/authorizer/server/constants"
"github.com/authorizerdev/authorizer/server/db/models"
2022-05-29 17:22:46 +05:30
"github.com/authorizerdev/authorizer/server/memorystore"
2022-01-21 12:18:07 +05:30
"gorm.io/driver/mysql"
"gorm.io/driver/postgres"
"gorm.io/driver/sqlite"
"gorm.io/driver/sqlserver"
"gorm.io/gorm"
2022-02-05 09:00:56 +05:30
"gorm.io/gorm/logger"
2022-01-21 12:18:07 +05:30
"gorm.io/gorm/schema"
)
type provider struct {
db *gorm.DB
}
// NewProvider returns a new SQL provider
func NewProvider() (*provider, error) {
var sqlDB *gorm.DB
var err error
2022-02-05 09:00:56 +05:30
customLogger := logger.New(
log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
logger.Config{
SlowThreshold: time.Second, // Slow SQL threshold
LogLevel: logger.Silent, // Log level
IgnoreRecordNotFoundError: true, // Ignore ErrRecordNotFound error for logger
Colorful: false, // Disable color
},
)
2022-01-21 12:18:07 +05:30
ormConfig := &gorm.Config{
2022-02-05 09:00:56 +05:30
Logger: customLogger,
2022-01-21 12:18:07 +05:30
NamingStrategy: schema.NamingStrategy{
TablePrefix: models.Prefix,
},
AllowGlobalUpdate: true,
2022-01-21 12:18:07 +05:30
}
2022-05-29 17:22:46 +05:30
2022-05-31 08:14:03 +05:30
dbType := memorystore.RequiredEnvStoreObj.GetRequiredEnv().DatabaseType
dbURL := memorystore.RequiredEnvStoreObj.GetRequiredEnv().DatabaseURL
2022-05-29 17:22:46 +05:30
switch dbType {
2022-06-09 23:43:21 +05:30
case constants.DbTypePostgres, constants.DbTypeYugabyte, constants.DbTypeCockroachDB:
2022-05-29 17:22:46 +05:30
sqlDB, err = gorm.Open(postgres.Open(dbURL), ormConfig)
2022-01-21 12:18:07 +05:30
case constants.DbTypeSqlite:
2022-05-29 17:22:46 +05:30
sqlDB, err = gorm.Open(sqlite.Open(dbURL), ormConfig)
case constants.DbTypeMysql, constants.DbTypeMariaDB, constants.DbTypePlanetScaleDB:
2022-05-29 17:22:46 +05:30
sqlDB, err = gorm.Open(mysql.Open(dbURL), ormConfig)
2022-01-21 12:18:07 +05:30
case constants.DbTypeSqlserver:
2022-05-29 17:22:46 +05:30
sqlDB, err = gorm.Open(sqlserver.Open(dbURL), ormConfig)
2022-01-21 12:18:07 +05:30
}
if err != nil {
return nil, err
}
err = sqlDB.AutoMigrate(&models.User{}, &models.VerificationRequest{}, &models.Session{}, &models.Env{}, &models.Webhook{}, models.WebhookLog{}, models.EmailTemplate{}, &models.OTP{})
2022-03-02 17:42:31 +05:30
if err != nil {
return nil, err
}
2022-01-21 12:18:07 +05:30
return &provider{
db: sqlDB,
}, nil
}