mirror of
https://github.com/SigNoz/signoz.git
synced 2025-12-17 23:47:12 +00:00
19 lines
336 B
Go
19 lines
336 B
Go
|
package sqlstore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
|
||
|
|
"github.com/uptrace/bun"
|
||
|
|
"github.com/uptrace/bun/schema"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewBunDB(sqldb *sql.DB, dialect schema.Dialect, hooks []SQLStoreHook, opts ...bun.DBOption) *bun.DB {
|
||
|
|
bunDB := bun.NewDB(sqldb, dialect, opts...)
|
||
|
|
|
||
|
|
for _, hook := range hooks {
|
||
|
|
bunDB.AddQueryHook(hook)
|
||
|
|
}
|
||
|
|
|
||
|
|
return bunDB
|
||
|
|
}
|