fix(parser):adapt order by has parentheses (#32)

Author: Ifffff

parser/mysql_parser/parser.py

@@ -350,9 +350,14 @@ def p_sort_items(p):
 
 
 def p_sort_item(p):
-    r"""sort_item : value_expression order_opt null_ordering_opt"""
-    p[0] = SortItem(p.lineno(1), p.lexpos(1),
-                    sort_key=p[1], ordering=p[2] or 'asc', null_ordering=p[3])
+    r"""sort_item : value_expression order_opt null_ordering_opt
+                  | LPAREN value_expression RPAREN order_opt null_ordering_opt"""
+    if len(p) == 4:
+        p[0] = SortItem(p.lineno(1), p.lexpos(1),
+                        sort_key=p[1], ordering=p[2] or 'asc', null_ordering=p[3])
+    else:
+        p[0] = SortItem(p.lineno(1), p.lexpos(1),
+                        sort_key=p[2], ordering=p[4] or 'asc', null_ordering=p[5])
 
 
 def p_order_opt(p):

parser/oceanbase_parser/parser.py

@@ -343,9 +343,14 @@ def p_sort_items(p):
 
 
 def p_sort_item(p):
-    r"""sort_item : value_expression order_opt null_ordering_opt"""
-    p[0] = SortItem(p.lineno(1), p.lexpos(1),
-                    sort_key=p[1], ordering=p[2] or 'asc', null_ordering=p[3])
+    r"""sort_item : value_expression order_opt null_ordering_opt
+                  | LPAREN value_expression RPAREN order_opt null_ordering_opt"""
+    if len(p) == 4:
+        p[0] = SortItem(p.lineno(1), p.lexpos(1),
+                        sort_key=p[1], ordering=p[2] or 'asc', null_ordering=p[3])
+    else:
+        p[0] = SortItem(p.lineno(1), p.lexpos(1),
+                        sort_key=p[2], ordering=p[4] or 'asc', null_ordering=p[5])
 
 
 def p_order_opt(p):

test/parser/test_parser_dml.py

@@ -456,6 +456,13 @@ def test_union_has_order_limit(self):
         result = oceanbase_parser.parse(sql)
         assert isinstance(result, Statement)
 
+    def test_order_by_has_parentheses(self):
+        sql = """
+SELECT channel_code , contact_number FROM customer_contact_channels WHERE active_to_date - active_from_date = (SELECT active_to_date - active_from_date FROM customer_contact_channels ORDER BY (active_to_date - active_from_date) DESC LIMIT 1)        """
+        sql = Utils.remove_sql_text_affects_parser(sql)
+        result = oceanbase_parser.parse(sql)
+        assert isinstance(result, Statement)
+
 
 if __name__ == '__main__':
     unittest.main()