@@ -9677,18 +9677,32 @@ WHERE object_id = table_descriptor_id
96779677 fingerprintFlags := tree .FmtFlags (tree .QueryFormattingForFingerprintsMask .Get (
96789678 & evalCtx .Settings .SV ,
96799679 ))
9680- targetStmt , err := parserutils .ParseOne (stmtFingerprint )
9680+ stmts , err := parserutils .Parse (stmtFingerprint )
96819681 if err != nil {
96829682 return nil , pgerror .Wrap (
96839683 err , pgcode .InvalidParameterValue , "could not parse statement fingerprint" ,
96849684 )
96859685 }
9686- donorStmt , err := parserutils .ParseOne (donorSQL )
9686+ if len (stmts ) != 1 {
9687+ return nil , pgerror .New (
9688+ pgcode .InvalidParameterValue ,
9689+ "could not parse statement fingerprint as a single SQL statement" ,
9690+ )
9691+ }
9692+ targetStmt := stmts [0 ]
9693+ stmts , err = parserutils .Parse (donorSQL )
96879694 if err != nil {
96889695 return nil , pgerror .Wrap (
96899696 err , pgcode .InvalidParameterValue , "could not parse hint donor statement" ,
96909697 )
96919698 }
9699+ if len (stmts ) != 1 {
9700+ return nil , pgerror .New (
9701+ pgcode .InvalidParameterValue ,
9702+ "could not parse hint donor statement as a single SQL statement" ,
9703+ )
9704+ }
9705+ donorStmt := stmts [0 ]
96929706 donor , err := tree .NewHintInjectionDonor (donorStmt .AST , fingerprintFlags )
96939707 if err != nil {
96949708 return nil , errors .NewAssertionErrorWithWrappedErrf (err , "error while creating donor" )
0 commit comments