From 57eb8f6712361a3bf75983ce153fac4846dc0273 Mon Sep 17 00:00:00 2001 From: Vikas Kushwaha Date: Tue, 11 Feb 2025 16:31:08 +0530 Subject: Initial commit --- internal/database/database.go | 50 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 internal/database/database.go (limited to 'internal/database') diff --git a/internal/database/database.go b/internal/database/database.go new file mode 100644 index 0000000..c67c75e --- /dev/null +++ b/internal/database/database.go @@ -0,0 +1,50 @@ +package database + +import ( + "fmt" + "log" + "os" + "time" + + "github.com/glebarez/sqlite" + "gorm.io/gorm" + "gorm.io/gorm/logger" +) + +var DB *gorm.DB + +func Open() { + + newLogger := logger.New( + log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer + logger.Config{ + SlowThreshold: time.Millisecond * 10, // Slow SQL threshold + LogLevel: logger.Info, // Log level + IgnoreRecordNotFoundError: false, // Ignore ErrRecordNotFound error for logger + Colorful: true, // Disable color + }, + ) + + db, err := gorm.Open(sqlite.Open("database.sqlite"), &gorm.Config{ + Logger: newLogger, + }) + + if err != nil { + fmt.Println("storage err: ", err) + } + + sqlDB, err := db.DB() + if err != nil { + fmt.Println("storage err: ", err) + } + + sqlDB.SetMaxIdleConns(3) + sqlDB.SetMaxOpenConns(100) + sqlDB.SetConnMaxLifetime(time.Hour) + + DB = db +} + +func Get() *gorm.DB { + return DB +} -- cgit v1.2.3