mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-04 11:25:52 +08:00

* chore: add migration for pat to add default values * fix: minor changes * fix: don't panic in GetClickhouseColumnName * fix: use new function for pat * fix: address minor comments * fix: address comments * fix: remove generatepat * fix: minor changes * fix: remove extra check --------- Co-authored-by: Vibhu Pandey <vibhupandey28@gmail.com>
48 lines
1.6 KiB
Go
48 lines
1.6 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
type SQLStoreTxOptions = sql.TxOptions
|
|
|
|
type SQLStore interface {
|
|
// SQLDB returns the underlying sql.DB.
|
|
SQLDB() *sql.DB
|
|
|
|
// BunDB returns an instance of bun.DB. This is the recommended way to interact with the database.
|
|
BunDB() *bun.DB
|
|
|
|
// SQLxDB returns an instance of sqlx.DB. This is the legacy ORM used.
|
|
SQLxDB() *sqlx.DB
|
|
|
|
// Returns the dialect of the database.
|
|
Dialect() SQLDialect
|
|
|
|
// RunInTxCtx runs the given callback in a transaction. It creates and injects a new context with the transaction.
|
|
// If a transaction is present in the context, it will be used.
|
|
RunInTxCtx(ctx context.Context, opts *SQLStoreTxOptions, cb func(ctx context.Context) error) error
|
|
|
|
// BunDBCtx returns an instance of bun.IDB for the given context.
|
|
// If a transaction is present in the context, it will be used. Otherwise, the default will be used.
|
|
BunDBCtx(ctx context.Context) bun.IDB
|
|
}
|
|
|
|
type SQLStoreHook interface {
|
|
bun.QueryHook
|
|
}
|
|
|
|
type SQLDialect interface {
|
|
MigrateIntToTimestamp(context.Context, bun.IDB, string, string) error
|
|
MigrateIntToBoolean(context.Context, bun.IDB, string, string) error
|
|
AddNotNullDefaultToColumn(context.Context, bun.IDB, string, string, string, string) error
|
|
GetColumnType(context.Context, bun.IDB, string, string) (string, error)
|
|
ColumnExists(context.Context, bun.IDB, string, string) (bool, error)
|
|
RenameColumn(context.Context, bun.IDB, string, string, string) (bool, error)
|
|
RenameTableAndModifyModel(context.Context, bun.IDB, interface{}, interface{}, func(context.Context) error) error
|
|
}
|