fix annotations for unions

Thriftpy · Feb 8, 2024 · 5784039 · 5784039
1 parent fc26372
commit 5784039
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/parser-cases/annotations.thrift b/tests/parser-cases/annotations.thrift
@@ -55,3 +55,7 @@ service foo_service {
   void foo() ( foo = "bar" )
 } (a.b="c")
 
+union foo_union {
+   1: optional bool abc
+   2: optional i32 xyz
+}  (a.b="c")
diff --git a/thriftpy2/parser/parser.py b/thriftpy2/parser/parser.py
@@ -229,7 +229,7 @@ def p_seen_struct(p):
 
 
 def p_union(p):
-    '''union : seen_union '{' field_seq '}' '''
+    '''union : seen_union '{' field_seq '}' type_annotations'''
     val = _fill_in_struct(p[1], p[3])
     _add_thrift_meta('unions', val)