zerolin1024 commited on
Commit
9b5cfec
·
verified ·
1 Parent(s): c66aafc

Upload 2 files

Browse files
Files changed (1) hide show
  1. database/database.go +8 -2
database/database.go CHANGED
@@ -11,6 +11,7 @@ import (
11
  "github.com/glebarez/sqlite"
12
  "gorm.io/driver/mysql"
13
  "gorm.io/gorm"
 
14
  )
15
 
16
  var DB *gorm.DB
@@ -18,9 +19,14 @@ var DB *gorm.DB
18
  func ConnectDatabase() {
19
  var err error
20
  dsn := os.Getenv("DATABASE_URI")
 
 
 
 
 
21
  if dsn == "" {
22
  log.Println("DATABASE_URI is not set, falling back to sqlite.")
23
- DB, err = gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
24
  } else {
25
  // Parse the aiven-provided URI
26
  parsedURL, err := url.Parse(dsn)
@@ -41,7 +47,7 @@ func ConnectDatabase() {
41
  path,
42
  )
43
 
44
- DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
45
  }
46
 
47
  if err != nil {
 
11
  "github.com/glebarez/sqlite"
12
  "gorm.io/driver/mysql"
13
  "gorm.io/gorm"
14
+ "gorm.io/gorm/logger"
15
  )
16
 
17
  var DB *gorm.DB
 
19
  func ConnectDatabase() {
20
  var err error
21
  dsn := os.Getenv("DATABASE_URI")
22
+
23
+ gormConfig := &gorm.Config{
24
+ Logger: logger.Default.LogMode(logger.Silent),
25
+ }
26
+
27
  if dsn == "" {
28
  log.Println("DATABASE_URI is not set, falling back to sqlite.")
29
+ DB, err = gorm.Open(sqlite.Open("test.db"), gormConfig)
30
  } else {
31
  // Parse the aiven-provided URI
32
  parsedURL, err := url.Parse(dsn)
 
47
  path,
48
  )
49
 
50
+ DB, err = gorm.Open(mysql.Open(dsn), gormConfig)
51
  }
52
 
53
  if err != nil {