mirror of
https://github.com/SigNoz/signoz.git
synced 2026-06-30 11:50:43 +01:00
5ab6636863
* feat: add api to fetch v2 dashboards for a metric name * chore: switch to query param * chore: generate API specs * chore: use proper struct in return type of GetByMetricNamesV2 * chore: add method for escaping like patterns in sqlstore formatter * fix: use only one db call in GetByMetricNamesV2 * chore: dont use type alias for list of references
114 lines
2.8 KiB
Go
114 lines
2.8 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) 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)
|
|
}
|