mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-04 11:25:52 +08:00
34 lines
558 B
Go
34 lines
558 B
Go
package dao
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"go.signoz.io/query-service/dao/sqlite"
|
|
)
|
|
|
|
var db ModelDao
|
|
|
|
func InitDao(engine, path string) error {
|
|
var err error
|
|
|
|
switch engine {
|
|
case "sqlite":
|
|
db, err = sqlite.InitDB(path)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to initialize DB")
|
|
}
|
|
default:
|
|
return fmt.Errorf("RelationalDB type: %s is not supported in query service", engine)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func DB() ModelDao {
|
|
if db == nil {
|
|
// Should never reach here
|
|
panic("GetDB called before initialization")
|
|
}
|
|
return db
|
|
}
|