github.com/adharshmk96/stk@v1.2.3/pkg/sqlMigrator/selector.go (about)

     1  package sqlmigrator
     2  
     3  func SelectDatabase(database string) Database {
     4  	switch database {
     5  	case "postgres", "postgresql":
     6  		return PostgresDB
     7  	case "mysql":
     8  		return MySQLDB
     9  	case "sqlite", "sqlite3":
    10  		return SQLiteDB
    11  	default:
    12  		return SQLiteDB
    13  	}
    14  }
    15  
    16  func SelectExtention(database Database) string {
    17  	var ext string
    18  	switch database {
    19  	case PostgresDB:
    20  		ext = "sql"
    21  	case MySQLDB:
    22  		ext = "sql"
    23  	case SQLiteDB:
    24  		ext = "sqlite"
    25  	default:
    26  		ext = "sql"
    27  	}
    28  
    29  	return ext
    30  }
    31  
    32  func SelectSubDirectory(database Database) string {
    33  	var subDir string
    34  	switch database {
    35  	case PostgresDB:
    36  		subDir = "postgres"
    37  	case MySQLDB:
    38  		subDir = "mysql"
    39  	case SQLiteDB:
    40  		subDir = "sqlite"
    41  	default:
    42  		subDir = "sqlite"
    43  	}
    44  
    45  	return subDir
    46  }