mirror of
https://github.com/dancojocaru2000/CfrTrainInfoTelegramBot.git
synced 2025-02-23 17:49:40 +02:00
29 lines
400 B
Go
29 lines
400 B
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
db *gorm.DB
|
||
|
|
mutex sync.RWMutex
|
||
|
|
)
|
||
|
|
|
||
|
|
func SetDatabase(d *gorm.DB) {
|
||
|
|
db = d
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadDB[T any](callback func(*gorm.DB) (T, error)) (T, error) {
|
||
|
|
mutex.RLock()
|
||
|
|
defer mutex.RUnlock()
|
||
|
|
return callback(db)
|
||
|
|
}
|
||
|
|
|
||
|
|
func WriteDB[T any](callback func(*gorm.DB) (T, error)) (T, error) {
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
return callback(db)
|
||
|
|
}
|