mirror of
https://github.com/SigNoz/signoz.git
synced 2026-09-13 15:00:40 +01:00
131e302c55
Adds JSONExtractMapValue to SQLFormatter so a labels map key is one path segment (dotted label keys work on both dialects), and a visitor over the shared filterquery grammar mapping rule list DSL keys to SQL.
121 lines
3.2 KiB
Go
121 lines
3.2 KiB
Go
package sqlstoretest
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/SigNoz/signoz/pkg/sqlstore"
|
|
"github.com/uptrace/bun/schema"
|
|
)
|
|
|
|
type formatter struct {
|
|
bunf schema.Formatter
|
|
}
|
|
|
|
func newFormatter(dialect schema.Dialect) sqlstore.SQLFormatter {
|
|
return &formatter{bunf: schema.NewFormatter(dialect)}
|
|
}
|
|
|
|
func (f *formatter) JSONExtractString(column, path string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, "json_extract("...)
|
|
sql = f.bunf.AppendIdent(sql, column)
|
|
sql = append(sql, ", "...)
|
|
sql = schema.Append(f.bunf, sql, path)
|
|
sql = append(sql, ")"...)
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) JSONExtractMapValue(column, mapField, key string) []byte {
|
|
// Mirrors the sqlite formatter: the key becomes one quoted path segment
|
|
// so dots inside it are not treated as nesting.
|
|
escapedKey := strings.NewReplacer(`\`, `\\`, `"`, `\"`).Replace(key)
|
|
return f.JSONExtractString(column, `$.`+mapField+`."`+escapedKey+`"`)
|
|
}
|
|
|
|
func (f *formatter) JSONType(column, path string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, "json_type("...)
|
|
sql = f.bunf.AppendIdent(sql, column)
|
|
sql = append(sql, ", "...)
|
|
sql = schema.Append(f.bunf, sql, path)
|
|
sql = append(sql, ")"...)
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) JSONIsArray(column, path string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, f.JSONType(column, path)...)
|
|
sql = append(sql, " = "...)
|
|
sql = schema.Append(f.bunf, sql, "array")
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) JSONArrayElements(column, path, alias string) ([]byte, []byte) {
|
|
var sql []byte
|
|
sql = append(sql, "json_each("...)
|
|
sql = f.bunf.AppendIdent(sql, column)
|
|
if path != "$" && path != "" {
|
|
sql = append(sql, ", "...)
|
|
sql = schema.Append(f.bunf, sql, path)
|
|
}
|
|
sql = append(sql, ") AS "...)
|
|
sql = f.bunf.AppendIdent(sql, alias)
|
|
|
|
return sql, append([]byte(alias), ".value"...)
|
|
}
|
|
|
|
func (f *formatter) JSONArrayOfStrings(column, path, alias string) ([]byte, []byte) {
|
|
return f.JSONArrayElements(column, path, alias)
|
|
}
|
|
|
|
func (f *formatter) JSONKeys(column, path, alias string) ([]byte, []byte) {
|
|
var sql []byte
|
|
sql = append(sql, "json_each("...)
|
|
sql = f.bunf.AppendIdent(sql, column)
|
|
if path != "$" && path != "" {
|
|
sql = append(sql, ", "...)
|
|
sql = schema.Append(f.bunf, sql, path)
|
|
}
|
|
sql = append(sql, ") AS "...)
|
|
sql = f.bunf.AppendIdent(sql, alias)
|
|
|
|
return sql, append([]byte(alias), ".key"...)
|
|
}
|
|
|
|
func (f *formatter) JSONArrayAgg(expression string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, "json_group_array("...)
|
|
sql = append(sql, expression...)
|
|
sql = append(sql, ')')
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) JSONArrayLiteral(values ...string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, "json_array("...)
|
|
for idx, value := range values {
|
|
if idx > 0 {
|
|
sql = append(sql, ", "...)
|
|
}
|
|
sql = schema.Append(f.bunf, sql, value)
|
|
}
|
|
sql = append(sql, ')')
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) TextToJsonColumn(column string) []byte {
|
|
return f.bunf.AppendIdent([]byte{}, column)
|
|
}
|
|
|
|
func (f *formatter) LowerExpression(expression string) []byte {
|
|
var sql []byte
|
|
sql = append(sql, "lower("...)
|
|
sql = append(sql, expression...)
|
|
sql = append(sql, ')')
|
|
return sql
|
|
}
|
|
|
|
func (f *formatter) EscapeLikePattern(value string) string {
|
|
return strings.NewReplacer(`\`, `\\`, `%`, `\%`, `_`, `\_`).Replace(value)
|
|
}
|