Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

it is a report #680

Closed
wants to merge 1 commit into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
213 changes: 107 additions & 106 deletions db.go
Original file line number Diff line number Diff line change
@@ -1,108 +1,109 @@
package main

import (
"log"
"math/rand"
"os"
"path/filepath"
"time"

"gorm.io/driver/mysql"
"gorm.io/driver/postgres"
"gorm.io/driver/sqlite"
"gorm.io/driver/sqlserver"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)

var DB *gorm.DB

func init() {
var err error
if DB, err = OpenTestConnection(); err != nil {
log.Printf("failed to connect database, got error %v\n", err)
os.Exit(1)
} else {
sqlDB, err := DB.DB()
if err == nil {
err = sqlDB.Ping()
}

if err != nil {
log.Printf("failed to connect database, got error %v\n", err)
}

RunMigrations()
if DB.Dialector.Name() == "sqlite" {
DB.Exec("PRAGMA foreign_keys = ON")
}

DB.Logger = DB.Logger.LogMode(logger.Info)
}
}

func OpenTestConnection() (db *gorm.DB, err error) {
dbDSN := os.Getenv("GORM_DSN")
switch os.Getenv("GORM_DIALECT") {
case "mysql":
log.Println("testing mysql...")
if dbDSN == "" {
dbDSN = "gorm:gorm@tcp(localhost:9910)/gorm?charset=utf8&parseTime=True&loc=Local"
}
db, err = gorm.Open(mysql.Open(dbDSN), &gorm.Config{})
case "postgres":
log.Println("testing postgres...")
if dbDSN == "" {
dbDSN = "user=gorm password=gorm host=localhost dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai"
}
db, err = gorm.Open(postgres.Open(dbDSN), &gorm.Config{})
case "sqlserver":
// CREATE LOGIN gorm WITH PASSWORD = 'LoremIpsum86';
// CREATE DATABASE gorm;
// USE gorm;
// CREATE USER gorm FROM LOGIN gorm;
// sp_changedbowner 'gorm';
log.Println("testing sqlserver...")
if dbDSN == "" {
dbDSN = "sqlserver://gorm:LoremIpsum86@localhost:9930?database=gorm"
}
db, err = gorm.Open(sqlserver.Open(dbDSN), &gorm.Config{})
default:
log.Println("testing sqlite3...")
db, err = gorm.Open(sqlite.Open(filepath.Join(os.TempDir(), "gorm.db")), &gorm.Config{})
}

if debug := os.Getenv("DEBUG"); debug == "true" {
db.Logger = db.Logger.LogMode(logger.Info)
} else if debug == "false" {
db.Logger = db.Logger.LogMode(logger.Silent)
}

return
}

func RunMigrations() {
var err error
allModels := []interface{}{&User{}, &Account{}, &Pet{}, &Company{}, &Toy{}, &Language{}}
rand.Seed(time.Now().UnixNano())
rand.Shuffle(len(allModels), func(i, j int) { allModels[i], allModels[j] = allModels[j], allModels[i] })

DB.Migrator().DropTable("user_friends", "user_speaks")

if err = DB.Migrator().DropTable(allModels...); err != nil {
log.Printf("Failed to drop table, got error %v\n", err)
os.Exit(1)
}

if err = DB.AutoMigrate(allModels...); err != nil {
log.Printf("Failed to auto migrate, but got error %v\n", err)
os.Exit(1)
}

for _, m := range allModels {
if !DB.Migrator().HasTable(m) {
log.Printf("Failed to create table for %#v\n", m)
os.Exit(1)
}
}
}
//
//import (
// "log"
// "math/rand"
// "os"
// "path/filepath"
// "time"
//
// "gorm.io/driver/mysql"
// "gorm.io/driver/postgres"
// "gorm.io/driver/sqlite"
// "gorm.io/driver/sqlserver"
// "gorm.io/gorm"
// "gorm.io/gorm/logger"
//)
//
//var DB *gorm.DB
//
//func init() {
// var err error
// if DB, err = OpenTestConnection(); err != nil {
// log.Printf("failed to connect database, got error %v\n", err)
// os.Exit(1)
// } else {
// sqlDB, err := DB.DB()
// if err == nil {
// err = sqlDB.Ping()
// }
//
// if err != nil {
// log.Printf("failed to connect database, got error %v\n", err)
// }
//
// RunMigrations()
// if DB.Dialector.Name() == "sqlite" {
// DB.Exec("PRAGMA foreign_keys = ON")
// }
//
// DB.Logger = DB.Logger.LogMode(logger.Info)
// }
//}
//
//func OpenTestConnection() (db *gorm.DB, err error) {
// dbDSN := os.Getenv("GORM_DSN")
// switch os.Getenv("GORM_DIALECT") {
// case "mysql":
// log.Println("testing mysql...")
// if dbDSN == "" {
// dbDSN = "gorm:gorm@tcp(localhost:9910)/gorm?charset=utf8&parseTime=True&loc=Local"
// }
// db, err = gorm.Open(mysql.Open(dbDSN), &gorm.Config{})
// case "postgres":
// log.Println("testing postgres...")
// if dbDSN == "" {
// dbDSN = "user=gorm password=gorm host=localhost dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai"
// }
// db, err = gorm.Open(postgres.Open(dbDSN), &gorm.Config{})
// case "sqlserver":
// // CREATE LOGIN gorm WITH PASSWORD = 'LoremIpsum86';
// // CREATE DATABASE gorm;
// // USE gorm;
// // CREATE USER gorm FROM LOGIN gorm;
// // sp_changedbowner 'gorm';
// log.Println("testing sqlserver...")
// if dbDSN == "" {
// dbDSN = "sqlserver://gorm:LoremIpsum86@localhost:9930?database=gorm"
// }
// db, err = gorm.Open(sqlserver.Open(dbDSN), &gorm.Config{})
// default:
// log.Println("testing sqlite3...")
// db, err = gorm.Open(sqlite.Open(filepath.Join(os.TempDir(), "gorm.db")), &gorm.Config{})
// }
//
// if debug := os.Getenv("DEBUG"); debug == "true" {
// db.Logger = db.Logger.LogMode(logger.Info)
// } else if debug == "false" {
// db.Logger = db.Logger.LogMode(logger.Silent)
// }
//
// return
//}
//
//func RunMigrations() {
// var err error
// allModels := []interface{}{&User{}, &Account{}, &Pet{}, &Company{}, &Toy{}, &Language{}}
// rand.Seed(time.Now().UnixNano())
// rand.Shuffle(len(allModels), func(i, j int) { allModels[i], allModels[j] = allModels[j], allModels[i] })
//
// DB.Migrator().DropTable("user_friends", "user_speaks")
//
// if err = DB.Migrator().DropTable(allModels...); err != nil {
// log.Printf("Failed to drop table, got error %v\n", err)
// os.Exit(1)
// }
//
// if err = DB.AutoMigrate(allModels...); err != nil {
// log.Printf("Failed to auto migrate, but got error %v\n", err)
// os.Exit(1)
// }
//
// for _, m := range allModels {
// if !DB.Migrator().HasTable(m) {
// log.Printf("Failed to create table for %#v\n", m)
// os.Exit(1)
// }
// }
//}
37 changes: 10 additions & 27 deletions docker-compose.yml
Original file line number Diff line number Diff line change
@@ -1,31 +1,14 @@
version: '3'
version: "3"

services:
mysql:
image: 'mysql/mysql-server:latest'
mariadb:
image: "mariadb:10.11.3"
restart: 'always'
ports:
- 9910:3306
- '13306:3306'
volumes:
- "/var/lib/mysql/data:/var/lib/mysql"
- "/var/lib/mysql/logs:/var/log/mysql"
- "/var/docker/mariadb/conf:/etc/mysql"
environment:
- MYSQL_DATABASE=gorm
- MYSQL_USER=gorm
- MYSQL_PASSWORD=gorm
- MYSQL_RANDOM_ROOT_PASSWORD="yes"
postgres:
image: 'postgres:latest'
ports:
- 9920:5432
environment:
- TZ=Asia/Shanghai
- POSTGRES_DB=gorm
- POSTGRES_USER=gorm
- POSTGRES_PASSWORD=gorm
mssql:
image: '${MSSQL_IMAGE:-mcmoe/mssqldocker}:latest'
ports:
- 9930:1433
environment:
- ACCEPT_EULA=Y
- SA_PASSWORD=LoremIpsum86
- MSSQL_DB=gorm
- MSSQL_USER=gorm
- MSSQL_PASSWORD=LoremIpsum86
MYSQL_ROOT_PASSWORD: "123456"
19 changes: 3 additions & 16 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -3,26 +3,13 @@ module gorm.io/playground
go 1.20

require (
github.com/google/uuid v1.3.0
gorm.io/driver/mysql v1.5.1
gorm.io/driver/postgres v1.5.2
gorm.io/driver/sqlite v1.5.3
gorm.io/driver/sqlserver v1.5.1
gorm.io/gorm v1.25.4
gorm.io/gorm v1.25.5
)

require (
github.com/go-sql-driver/mysql v1.7.1 // indirect
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect
github.com/golang-sql/sqlexp v0.1.0 // indirect
github.com/jackc/pgpassfile v1.0.0 // indirect
github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
github.com/jackc/pgx/v5 v5.4.3 // indirect
github.com/go-sql-driver/mysql v1.7.0 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
github.com/mattn/go-sqlite3 v1.14.17 // indirect
github.com/microsoft/go-mssqldb v1.5.0 // indirect
golang.org/x/crypto v0.12.0 // indirect
golang.org/x/text v0.12.0 // indirect
)

replace gorm.io/gorm => ./gorm
60 changes: 58 additions & 2 deletions main.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,63 @@
package main

import "fmt"
import (
"fmt"
"github.com/google/uuid"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"gorm.io/gorm/logger"
"strings"
)

type TestUser struct {
Id int `json:"id"`
Username string `json:"username" gorm:"unique;index" validate:"max=12"`
Password string `json:"password" gorm:"not null;" validate:"min=8,max=20"`
DisplayName string `json:"display_name" gorm:"index" validate:"max=20"`
Role int `json:"role" gorm:"type:int;default:1"` // admin, common
Status int `json:"status" gorm:"type:int;default:1"` // enabled, disabled
Email string `json:"email" gorm:"index" validate:"max=50"`
GitHubId string `json:"github_id" gorm:"column:github_id;index"`
WeChatId string `json:"wechat_id" gorm:"column:wechat_id;index"`
VerificationCode string `json:"verification_code" gorm:"-:all"` // this field is only for Email verification, don't save it to database!
AccessToken string `json:"access_token" gorm:"type:char(32);column:access_token;uniqueIndex"` // this token is for system management
Quota int `json:"quota" gorm:"type:int;default:0"`
UsedQuota int `json:"used_quota" gorm:"type:int;default:0;column:used_quota"` // used quota
RequestCount int `json:"request_count" gorm:"type:int;default:0;"` // request number
Group string `json:"group" gorm:"type:varchar(32);default:'default'"`
AffCode string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"`
InviterId int `json:"inviter_id" gorm:"type:int;column:inviter_id;index"`
}

func main() {
fmt.Println("vim-go")
dsn := "root:123456@tcp(localhost:13306)/chat?charset=utf8mb4&parseTime=True&loc=Local&timeout=10s"
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
PrepareStmt: true, // precompile SQL
Logger: logger.Default.LogMode(logger.Info),
})
if err != nil {
fmt.Println(err)
}
db = db.Set("gorm:table_options", "CHARSET=utf8mb4")
err = db.AutoMigrate(&TestUser{})
if err != nil {
fmt.Println(err)
}

var user TestUser
if err := db.Take(&user).Error; err != nil {
rootUser := TestUser{
Username: "root",
Password: "$2a$10$OoNwis82Y7/yjN80Ces/FR1CZhK5YjOo7xn6pOTewobO",
Role: 1,
Status: 1,
DisplayName: "Root User",
AccessToken: strings.Replace(uuid.New().String(), "-", "", -1),
Quota: 100000000,
}
err2 := db.Create(&rootUser).Error
fmt.Println("-----------------------WHY???------------------------")
fmt.Println("root user", err2)
}

}
2 changes: 1 addition & 1 deletion test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ dialects=("sqlite" "mysql" "postgres" "sqlserver")
if [ "$GORM_ENABLE_CACHE" = "" ]
then
rm -rf gorm
if [ "$GORM_DIALECT" = "" ] || [ "$GORM_DIALECT" = "${dialect}" ]
fi

[ -d gorm ] || (echo "git clone --depth 1 -b $(cat main_test.go | grep GORM_BRANCH | awk '{print $3}') $(cat main_test.go | grep GORM_REPO | awk '{print $3}')"; git clone --depth 1 -b $(cat main_test.go | grep GORM_BRANCH | awk '{print $3}') $(cat main_test.go | grep GORM_REPO | awk '{print $3}'))
Expand All @@ -27,7 +28,6 @@ if [[ -z $GITHUB_ACTION ]]; then
fi

for dialect in "${dialects[@]}" ; do
if [ "$GORM_DIALECT" = "" ] || [ "$GORM_DIALECT" = "${dialect}" ]
then
if [[ $(grep TEST_DRIVER main_test.go) =~ "${dialect}" ]]
then
Expand Down
Loading