Skip to content

Commit cc9a593

Browse files
authored
Add alternate ValidateSqlQueries which, unlike ValidateSqlQuery, does not consider multiple statements in one query to be an error. (#42)
1 parent 5f05ce8 commit cc9a593

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

pkg/vet/vet.go

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -958,6 +958,25 @@ func ValidateSqlQuery(ctx VetContext, queryStr string) ([]QueryParam, error) {
958958
return params, err
959959
}
960960

961+
func ValidateSqlQueries(ctx VetContext, queryStr string) ([][]QueryParam, error) {
962+
var ret [][]QueryParam
963+
964+
tree, err := pg_query.Parse(queryStr)
965+
if err != nil {
966+
return nil, err
967+
}
968+
969+
for _, s := range tree.Stmts {
970+
params, _, err := validateSqlQuery(ctx, s.Stmt)
971+
if err != nil {
972+
return nil, err
973+
}
974+
ret = append(ret, params)
975+
}
976+
977+
return ret, nil
978+
}
979+
961980
func validateSqlQuery(ctx VetContext, node *pg_query.Node) ([]QueryParam, []ColumnUsed, error) {
962981

963982
switch {

0 commit comments

Comments
 (0)