@@ -23,13 +23,13 @@ import (
2323 "sync"
2424 "time"
2525
26- "github.com/casbin/casbin/v3"
27- "github.com/casbin/casbin/v3/model"
28- "github.com/casbin/casbin/v3/persist"
29- "github.com/glebarez/sqlite"
26+ "github.com/casbin/casbin/v2"
27+ "github.com/casbin/casbin/v2/model"
28+ "github.com/casbin/casbin/v2/persist"
3029 "github.com/pkg/errors"
3130 "gorm.io/driver/mysql"
3231 "gorm.io/driver/postgres"
32+ "gorm.io/driver/sqlite"
3333 "gorm.io/driver/sqlserver"
3434 "gorm.io/gorm"
3535 "gorm.io/gorm/clause"
@@ -326,16 +326,17 @@ func openDBConnection(driverName, dataSourceName string) (*gorm.DB, error) {
326326 config := & gorm.Config {
327327 Logger : logger .Default .LogMode (logger .Silent ),
328328 }
329- if driverName == "postgres" {
329+ switch driverName {
330+ case "postgres" :
330331 db , err = gorm .Open (postgres .Open (dataSourceName ), config )
331- } else if driverName == "mysql" {
332+ case "mysql" :
332333 db , err = gorm .Open (mysql .Open (dataSourceName ), config )
333- } else if driverName == "sqlserver" {
334+ case "sqlserver" :
334335 db , err = gorm .Open (sqlserver .Open (dataSourceName ), config )
335- } else if driverName == "sqlite3" {
336+ case "sqlite" :
336337 db , err = gorm .Open (sqlite .Open (dataSourceName ), config )
337- } else {
338- return nil , errors .New ("Database dialect '" + driverName + "' is not supported. Supported databases are postgres, mysql, sqlserver and sqlite3 " )
338+ default :
339+ return nil , errors .New ("Database dialect '" + driverName + "' is not supported. Supported databases are postgres, mysql, sqlserver and sqlite " )
339340 }
340341 if err != nil {
341342 return nil , err
@@ -356,7 +357,7 @@ func (a *Adapter) createDatabase() error {
356357 return nil
357358 }
358359 }
359- } else if a .driverName != "sqlite3 " && a .driverName != "sqlserver" {
360+ } else if a .driverName != "sqlite " && a .driverName != "sqlserver" {
360361 err = db .Exec ("CREATE DATABASE IF NOT EXISTS " + a .databaseName ).Error
361362 }
362363 if err != nil {
@@ -378,13 +379,14 @@ func (a *Adapter) Open() error {
378379 if err = a .createDatabase (); err != nil {
379380 return err
380381 }
381- if a .driverName == "postgres" {
382+ switch a .driverName {
383+ case "postgres" :
382384 db , err = openDBConnection (a .driverName , a .dataSourceName + " dbname=" + a .databaseName )
383- } else if a . driverName == "sqlite3" {
385+ case "sqlite" :
384386 db , err = openDBConnection (a .driverName , a .dataSourceName )
385- } else if a . driverName == "sqlserver" {
387+ case "sqlserver" :
386388 db , err = openDBConnection (a .driverName , a .dataSourceName + "?database=" + a .databaseName )
387- } else {
389+ default :
388390 db , err = openDBConnection (a .driverName , a .dataSourceName + a .databaseName )
389391 }
390392 if err != nil {
@@ -468,9 +470,7 @@ func (a *Adapter) dropTable() error {
468470func (a * Adapter ) truncateTable () error {
469471 var sql string
470472 switch a .db .Config .Name () {
471- case sqlite .DriverName :
472- sql = fmt .Sprintf ("delete from %s" , a .getFullTableName ())
473- case "sqlite3" :
473+ case "sqlite" :
474474 sql = fmt .Sprintf ("delete from %s" , a .getFullTableName ())
475475 case "postgres" :
476476 sql = fmt .Sprintf ("truncate table %s RESTART IDENTITY" , a .getFullTableName ())
0 commit comments