Merge pull request #37 from fossyy/staging

Staging
This commit is contained in:
2024-09-03 13:47:14 +07:00
committed by GitHub
2 changed files with 8 additions and 4 deletions

View File

@ -64,6 +64,10 @@ func NewMYSQLdb(username, password, host, port, dbName string) Database {
Logger: gormLogger.Default.LogMode(gormLogger.Silent), Logger: gormLogger.Default.LogMode(gormLogger.Silent),
}) })
if err != nil {
panic("failed to connect database: " + err.Error())
}
initDB.Raw("SELECT count(*) FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = ?", dbName).Scan(&count) initDB.Raw("SELECT count(*) FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = ?", dbName).Scan(&count)
if count <= 0 { if count <= 0 {
if err := initDB.Exec("CREATE DATABASE IF NOT EXISTS " + dbName).Error; err != nil { if err := initDB.Exec("CREATE DATABASE IF NOT EXISTS " + dbName).Error; err != nil {

View File

@ -66,25 +66,25 @@ func CheckPasswordHash(password, hash string) bool {
} }
func ValidatePassword(password string) bool { func ValidatePassword(password string) bool {
if len(password) < 6 { if len(password) < 8 {
return false return false
} }
var ( var (
hasNumber int hasNumber bool
hasUppercase bool hasUppercase bool
) )
for _, char := range password { for _, char := range password {
switch { switch {
case unicode.IsNumber(char): case unicode.IsNumber(char):
hasNumber++ hasNumber = true
case unicode.IsUpper(char): case unicode.IsUpper(char):
hasUppercase = true hasUppercase = true
} }
} }
return hasNumber >= 3 && hasUppercase return hasNumber && hasUppercase
} }
func ConvertFileSize(byte int64) string { func ConvertFileSize(byte int64) string {