gorm.go 665 B

123456789101112131415161718192021222324252627282930
  1. package db
  2. import (
  3. "fmt"
  4. "os"
  5. "gorm.io/driver/mysql"
  6. "gorm.io/gorm"
  7. )
  8. var Gorm *gorm.DB
  9. func init() {
  10. var err error
  11. var dsn string
  12. connectionName := os.Getenv("CLOUDSQL_CONNECTION_NAME")
  13. if connectionName != "" {
  14. dbName := os.Getenv("DB_NAME")
  15. dbUser := os.Getenv("DB_USER")
  16. dbPwd := os.Getenv("DB_PASS")
  17. dsn = fmt.Sprintf("%s:%s@tcp(%s)/%s?parseTime=true", dbUser, dbPwd, connectionName, dbName)
  18. } else {
  19. // dsn := "root:123456@tcp(localhost:3306)/rental?charset=utf8"
  20. dsn = "homapp:PaqvTcSXUzNc9M9@tcp(34.105.87.213:3306)/homapp?charset=utf8"
  21. }
  22. Gorm, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
  23. if err != nil {
  24. panic(err)
  25. }
  26. }