mirror of
https://github.com/SigNoz/signoz.git
synced 2025-12-22 09:56:57 +00:00
32 lines
990 B
Go
32 lines
990 B
Go
|
package telemetrystore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/ClickHouse/clickhouse-go/v2"
|
||
|
|
"github.com/ClickHouse/clickhouse-go/v2/lib/driver"
|
||
|
|
)
|
||
|
|
|
||
|
|
type TelemetryStore interface {
|
||
|
|
ClickHouseDB() clickhouse.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
type TelemetryStoreHook interface {
|
||
|
|
BeforeQuery(ctx context.Context, query string, args ...interface{}) (context.Context, string, []interface{})
|
||
|
|
AfterQuery(ctx context.Context, query string, args []interface{}, rows driver.Rows, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func WrapBeforeQuery(hooks []TelemetryStoreHook, ctx context.Context, query string, args ...interface{}) (context.Context, string, []interface{}) {
|
||
|
|
for _, hook := range hooks {
|
||
|
|
ctx, query, args = hook.BeforeQuery(ctx, query, args...)
|
||
|
|
}
|
||
|
|
return ctx, query, args
|
||
|
|
}
|
||
|
|
|
||
|
|
// runAfterHooks executes all after hooks in order
|
||
|
|
func WrapAfterQuery(hooks []TelemetryStoreHook, ctx context.Context, query string, args []interface{}, rows driver.Rows, err error) {
|
||
|
|
for _, hook := range hooks {
|
||
|
|
hook.AfterQuery(ctx, query, args, rows, err)
|
||
|
|
}
|
||
|
|
}
|