mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-31 03:41:58 +08:00
eba2049a4d
All necessary changes so that whatever initalize SQL commans run, they are moved to bun so that it works with both sqlite and postgres.
32 lines
486 B
Go
32 lines
486 B
Go
package dao
|
|
|
|
import (
|
|
"go.signoz.io/signoz/pkg/query-service/dao/sqlite"
|
|
"go.signoz.io/signoz/pkg/sqlstore"
|
|
)
|
|
|
|
var db ModelDao
|
|
|
|
func InitDao(sqlStore sqlstore.SQLStore) error {
|
|
var err error
|
|
db, err = sqlite.InitDB(sqlStore)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SetDB is used by ee for setting modelDAO
|
|
func SetDB(m ModelDao) {
|
|
db = m
|
|
}
|
|
|
|
func DB() ModelDao {
|
|
if db == nil {
|
|
// Should never reach here
|
|
panic("GetDB called before initialization")
|
|
}
|
|
return db
|
|
}
|