get diagnostics

corpus/plpgsql/get_diagnostics_statement.txt

@@ -0,0 +1,25 @@
+================================================================================
+get diagnostics
+================================================================================
+DO $$
+BEGIN
+    get current diagnostics foo = ROW_COUNT;
+    get diagnostics bar := ROW_COUNT;
+END
+$$;
+--------------------------------------------------------------------------------
+
+(source_file
+  (do_block
+    (block
+      (dollar_quote)
+      (body
+        (get_diagnostics_statement
+          (assign_statement
+            (identifier)
+            (identifier)))
+        (get_diagnostics_statement
+          (assign_statement
+            (identifier)
+            (identifier))))
+      (dollar_quote))))

grammar.js

@@ -410,6 +410,7 @@ module.exports = grammar({
       choice(
         $._statement,
         $.assign_statement,
+        $.get_diagnostics_statement,
         $.return_statement,
         $.raise_statement,
         $.if_statement,
@@ -420,6 +421,8 @@ module.exports = grammar({
       ";",
     ),
 
+    get_diagnostics_statement: $ => seq(kw("get"), optional(kw("current")), kw("diagnostics"), $.assign_statement),
+
     for_statement: $ => seq(
       kw("for"), commaSep1($.identifier), kw("in"), choice(
         seq(