mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-10-14 05:21:33 +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.
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
basedao "go.signoz.io/signoz/pkg/query-service/dao"
|
|
basedsql "go.signoz.io/signoz/pkg/query-service/dao/sqlite"
|
|
baseint "go.signoz.io/signoz/pkg/query-service/interfaces"
|
|
"go.signoz.io/signoz/pkg/sqlstore"
|
|
)
|
|
|
|
type modelDao struct {
|
|
*basedsql.ModelDaoSqlite
|
|
flags baseint.FeatureLookup
|
|
}
|
|
|
|
// SetFlagProvider sets the feature lookup provider
|
|
func (m *modelDao) SetFlagProvider(flags baseint.FeatureLookup) {
|
|
m.flags = flags
|
|
}
|
|
|
|
// CheckFeature confirms if a feature is available
|
|
func (m *modelDao) checkFeature(key string) error {
|
|
if m.flags == nil {
|
|
return fmt.Errorf("flag provider not set")
|
|
}
|
|
|
|
return m.flags.CheckFeature(key)
|
|
}
|
|
|
|
// InitDB creates and extends base model DB repository
|
|
func InitDB(sqlStore sqlstore.SQLStore) (*modelDao, error) {
|
|
dao, err := basedsql.InitDB(sqlStore)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// set package variable so dependent base methods (e.g. AuthCache) will work
|
|
basedao.SetDB(dao)
|
|
m := &modelDao{ModelDaoSqlite: dao}
|
|
return m, nil
|
|
}
|
|
|
|
func (m *modelDao) DB() *sqlx.DB {
|
|
return m.ModelDaoSqlite.DB()
|
|
}
|