Skip to content

Commit 32bb5f7

Browse files
committed
Added query identified for the ordinal queries
Allows easier identification of the queries for example when viewing pg_stat_activity
1 parent fcc75b8 commit 32bb5f7

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

Diff for: go.mod

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
module github.com/radim/queries
22

3-
go 1.15
3+
go 1.21

Diff for: queries.go

+6-4
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ type (
2727
}
2828

2929
Query struct {
30+
Name string
3031
Raw string
3132
OrdinalQuery string
3233
Mapping map[string]int
@@ -130,21 +131,22 @@ func (s *QueryStore) loadQueriesFromFile(fileName string, r io.Reader) error {
130131
return fmt.Errorf("Query '%s' already exists", name)
131132
}
132133

133-
q := NewQuery(query)
134+
q := NewQuery(name, query)
134135

135136
s.queries[name] = q
136137
}
137138

138139
return nil
139140
}
140141

141-
func NewQuery(query string) *Query {
142+
func NewQuery(name, query string) *Query {
142143
var (
143144
position int = 1
144145
)
145146

146147
q := Query{
147-
Raw: query,
148+
Name: name,
149+
Raw: query,
148150
}
149151

150152
mapping := make(map[string]int)
@@ -171,7 +173,7 @@ func NewQuery(query string) *Query {
171173
query = r.ReplaceAllLiteralString(query, fmt.Sprintf("$%d", ord))
172174
}
173175

174-
q.OrdinalQuery = query
176+
q.OrdinalQuery = fmt.Sprintf("-- %s\n%s", name, query)
175177
q.Mapping = mapping
176178

177179
return &q

0 commit comments

Comments
 (0)