File tree 1 file changed +19
-1
lines changed
1 file changed +19
-1
lines changed Original file line number Diff line number Diff line change @@ -6,10 +6,13 @@ package db
6
6
7
7
import (
8
8
"fmt"
9
+ "github.com/gitpod-io/gitpod/common-go/log"
9
10
driver_mysql "github.com/go-sql-driver/mysql"
11
+ "github.com/sirupsen/logrus"
10
12
"github.com/stretchr/testify/require"
11
13
"gorm.io/driver/mysql"
12
14
"gorm.io/gorm"
15
+ "gorm.io/gorm/logger"
13
16
"testing"
14
17
"time"
15
18
)
@@ -38,7 +41,22 @@ func Connect(p ConnectionParams) (*gorm.DB, error) {
38
41
}
39
42
40
43
// refer to https://github.com/go-sql-driver/mysql#dsn-data-source-name for details
41
- return gorm .Open (mysql .Open (cfg .FormatDSN ()), & gorm.Config {})
44
+ return gorm .Open (mysql .Open (cfg .FormatDSN ()), & gorm.Config {
45
+ Logger : logger .New (log .Log , logger.Config {
46
+ SlowThreshold : 200 * time .Millisecond ,
47
+ Colorful : false ,
48
+ LogLevel : (func () logger.LogLevel {
49
+ switch log .Log .Level {
50
+ case logrus .PanicLevel , logrus .FatalLevel , logrus .ErrorLevel :
51
+ return logger .Error
52
+ case logrus .WarnLevel :
53
+ return logger .Warn
54
+ default :
55
+ return logger .Info
56
+ }
57
+ })(),
58
+ }),
59
+ })
42
60
}
43
61
44
62
func ConnectForTests (t * testing.T ) * gorm.DB {
You can’t perform that action at this time.
0 commit comments