mirror of
https://github.com/SigNoz/signoz.git
synced 2026-02-20 15:52:41 +00:00
83eeb46f99
* chore: added sql formatter for json * chore: updated json extract columns * chore: added apend ident * chore: resolved pr comments * chore: resolved pr comments * chore: resolved pr comments * chore: resolved pr comments * chore: minor changes * chore: minor changes * chore: minor changes * chore: minor changes * chore: resolve comments * chore: added append value * chore: added append value * chore: added append value * chore: added append value * chore: added append value * chore: added append value * chore: added append value * chore: added append value --------- Co-authored-by: Vikrant Gupta <vikrant@signoz.io>
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package sqlstoretest
|
|
|
|
import (
|
|
"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) 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
|
|
}
|