ANTLR4 \u7684\u5e94\u7528\u573a\u666f<\/h3>\n
Apache Spark<\/strong>: \u6d41\u884c\u7684\u5927\u6570\u636e\u5904\u7406\u6846\u67b6\uff0c\u4f7f\u7528ANTLR\u4f5c\u4e3a\u5176SQL\u89e3\u6790\u5668\u7684\u4e00\u90e8\u5206\uff0c\u652f\u6301SQL\u67e5\u8be2\u3002 \u4e3a\u4e86\u5b9e\u73b0\u4e00\u95e8\u8ba1\u7b97\u673a\u7f16\u7a0b\u8bed\u8a00\uff0c\u6211\u4eec\u9700\u8981\u6784\u5efa\u4e00\u4e2a\u7a0b\u5e8f\u6765\u8bfb\u53d6\u8f93\u5165\u8bed\u53e5\uff0c\u5bf9\u5176\u4e2d\u7684\u8bcd\u7ec4\u548c\u7b26\u53f7\u8fdb\u884c\u8bc6\u522b\u5904\u7406\uff0c\u5373\u6211\u4eec\u9700\u8981\u8bed\u6cd5\u89e3\u91ca\u5668\u6216\u8005\u7ffb\u8bd1\u5668\u6765\u8bc6\u522b\u51fa\u4e00\u95e8\u7279\u5b9a\u8bed\u8a00\u7684\u6240\u6709\u8bcd\u7ec4\uff0c\u5b50\u8bcd\u7ec4\uff0c\u8bed\u53e5\u3002\u6211\u4eec\u5c06\u8bed\u6cd5\u5206\u6790\u8fc7\u7a0b\u62c6\u5206\u4e3a\u4e24\u4e2a\u72ec\u7acb\u7684\u9636\u6bb5\u5219\u4e3a\u8bcd\u6cd5\u5206\u6790\u548c\u8bed\u6cd5\u5206\u6790\u3002<\/p>\n ANTLR\u8bed\u6cd5\u9075\u5faa\u4e86\u4e00\u79cd\u4e13\u95e8\u7528\u6765\u63cf\u8ff0\u5176\u4ed6\u8bed\u8a00\u7684\u8bed\u6cd5\uff0c\u6211\u4eec\u79f0\u4e4b\u4e3aANTLR\u5143\u8bed\u8a00\uff08ANTLR\u2019s meta-language\uff09\u3002ANTLR\u5143\u8bed\u53e5\u662f\u4e00\u4e2a\u5f3a\u5927\u7684\u5de5\u5177\uff0c\u53ef\u4ee5\u7528\u6765\u5b9a\u4e49\u7f16\u7a0b\u8bed\u8a00\u7684\u8bed\u6cd5\u3002\u901a\u8fc7\u5b9a\u4e49\u8bcd\u6cd5\u548c\u8bed\u6cd5\u89c4\u5219\uff0c\u53ef\u4ee5\u57fa\u4e8eantlr\u751f\u6210\u89e3\u6790\u5668\u548c\u8bcd\u6cd5\u5206\u6790\u5668\u3002<\/p>\n 1\u3001\u81ea\u9876\u5411\u4e0b<\/strong> 2\u3001\u8bed\u8a00\u6a21\u5f0f<\/strong><\/p>\n \u8ba1\u7b97\u673a\u8bed\u8a00\u5e38\u89c14\u79cd\u8bed\u8a00\u6a21\u5f0f\uff1a\u5e8f\u5217\uff08sequence\uff09\u3001\u9009\u62e9\uff08choice\uff09\u3001\u8bcd\u6cd5\u7b26\u53f7\u4f9d\u8d56 \uff08token dependency\uff09\uff0c\u4ee5\u53ca\u5d4c\u5957\u7ed3\u6784\uff08nested phrase\uff09<\/strong>\u3002\u4ee5\u4e0b\u662fANTLR\u5bf94\u79cd\u6a21\u5f0f\u7684\u8bed\u6cd5\u89c4\u5219\u63cf\u8ff0\u3002<\/p>\n 3\u3001\u8bed\u6cd5\u6b67\u4e49<\/strong><\/p>\n \u5728\u81ea\u9876\u5411\u4e0b\u7684\u8bed\u6cd5\u548c\u624b\u5de5\u7f16\u5199\u7684\u9012\u5f52\u4e0b\u964d\u8bed\u6cd5\u5206\u6790\u5668\u4e2d\uff0c\u5904\u7406\u8868\u8fbe\u5f0f\u90fd\u662f\u4e00\u4ef6\u76f8\u5f53\u68d8\u624b\u7684\u4e8b\u60c5\uff0c\u8fd9\u9996\u5148\u662f\u56e0\u4e3a\u5927\u591a\u6570\u8bed\u6cd5\u90fd\u5b58\u5728\u6b67\u4e49\uff0c\u5176\u6b21\u662f\u56e0\u4e3a\u5927\u591a\u6570\u8bed\u8a00\u7684\u89c4\u8303\u4f7f\u7528\u4e86\u4e00\u79cd\u7279\u6b8a\u7684\u9012\u5f52\u65b9\u5f0f\uff0c\u79f0\u4e3a\u5de6\u9012\u5f52\u3002<\/p>\n \u6211\u4eec\u4e3e\u4e2a\u8fd0\u7b97\u7b26\u4f18\u5148\u7ea7\u5e26\u6765\u7684\u8bed\u6cd5\u6b67\u4e49<\/strong>\u95ee\u9898\uff0c\u540c\u6837\u7684\u89c4\u5219\u53ef\u4ee5\u5339\u914d\u591a\u4e2a\u8f93\u5165\u5b57\u7b26\u6d41<\/strong>\u3002<\/p>\n \u5728\u5176\u4ed6\u8bed\u6cd5\u5de5\u5177\u4e2d\uff0c\u901a\u5e38\u901a\u8fc7\u6307\u5b9a\u989d\u5916\u7684\u6807\u8bb0\u6765\u6307\u5b9a\u8fd0\u7b97\u7b26\u4f18\u5148\u7ea7\u3002\u800c\u5728ANTLR4\u4e2d\u901a\u8fc7\u5907\u9009\u5206\u652f\u7684\u6392\u5e8f\u6765\u6307\u5b9a\u4f18\u5148\u7ea7<\/strong>\uff0c\u8d8a\u9760\u524d\u4f18\u5148\u7ea7\u8d8a\u9ad8\u3002<\/p>\n ANTLR\u53ef\u4ee5\u6839\u636elexer.g4\u548cparser.g4\u81ea\u52a8\u751f\u6210\u8bcd\u6cd5\u5206\u6790\u5668\uff0c\u8bed\u6cd5\u5206\u6790\u5668\uff0c\u76d1\u542c\u5668\uff0c\u8bbf\u95ee\u5668<\/strong>\u7b49\u3002<\/p>\n \u5728ANTLR4\u4e2d\u8bed\u6cd5\u89e3\u6790\u548c\u4e1a\u52a1\u903b\u8f91\u7684\u9ad8\u5ea6\u89e3\u8026\u662f\u4e00\u4e2a\u91cd\u8981\u7684\u8bbe\u8ba1\u7406\u5ff5\uff0c\u4f18\u70b9\u5c31\u662f\u540c\u4e00\u4e2a AST \u7ed3\u6784\u80fd\u591f\u5728\u4e0d\u540c\u7684\u4e1a\u52a1\u903b\u8f91\u5b9e\u73b0\u4e4b\u95f4\u5b9e\u73b0\u590d\u7528\u3002\u4e0d\u540c\u7684\u4e1a\u52a1\u903b\u8f91\uff08\u5982\u6267\u884c\u3001\u8f6c\u6362\u3001\u4f18\u5316\u7b49\uff09\u53ef\u4ee5\u5bf9\u540c\u4e00\u4e2a AST \u8fdb\u884c\u4e0d\u540c\u7684\u5904\u7406\uff0c\u800c\u4e0d\u9700\u8981\u5173\u5fc3\u89e3\u6790\u8fc7\u7a0b\u3002\u6838\u5fc3\u51e0\u4e2a\u8bbe\u8ba1\u65b9\u6848\u5982\u4e0b\uff1a<\/p>\n Spark SQL \u662f Apache Spark \u7684\u4e00\u4e2a\u6a21\u5757\uff0c\u4e13\u95e8\u7528\u4e8e\u5904\u7406\u7ed3\u6784\u5316\u6570\u636e\uff0cSpark SQL \u7684\u7279\u70b9\u5305\u62ec\uff1a<\/p>\n \u5728Aparch Spark\u6e90\u7801\u4e2d\u5c31\u662f\u4f7f\u7528ANTLR4\u6765\u89e3\u6790\u548c\u5904\u7406SQL\u8bed\u53e5\uff0c\u4ee5\u4e0b\u4e3aApach Spark\u4e2d\u57fa\u4e8eANTLR\u5143\u8bed\u8a00\u5b9a\u4e49\u7684\u8bcd\u6cd5\u5206\u6790\u5668\u548c\u8bed\u6cd5\u5206\u6790\u5668\uff0c\u5728\u8bed\u6cd5\u5b9a\u4e49\u4e0a\u6211\u4eec\u53ea\u9700\u8981\u57fa\u4e8e\u8fd9\u5957\u6807\u51c6\u7684SparkSQL\u8bed\u6cd5<\/strong>\u53bb\u9002\u914d\u5f97\u7269\u81ea\u7814\u5f15\u64ce<\/strong>\u7684\u80fd\u529b\uff0c\u505a\u80fd\u529b\u5bf9\u9f50\u3002<\/p>\n \u4ee5\u4e0b\u6211\u4eec\u4ee5\u5b57\u6bb5\u8865\u5168\u573a\u666f\u4e3a\u4f8b\u89e3\u6790\u4ece\u8bed\u6cd5\u5b9a\u4e49\uff0c\u8bed\u6cd5\u89e3\u6790\uff0c\u8bed\u6cd5\u8865\u5168\uff0c\u4e0a\u4e0b\u6587\u4fe1\u606f\u91c7\u96c6\u5404\u4e2a\u6d41\u7a0b\u8282\u70b9\u5256\u6790\u6700\u540e\u5b8c\u6210\u7684\u8868\u5b57\u6bb5\u4fe1\u606f\u7cbe\u51c6\u63a8\u8350\u3002\u5728\u4e0b\u5217\u8bed\u6cd5\u573a\u666f\u4e2d\uff0c\u5b58\u5728\u591a\u5c42Select\u8bed\u6cd5\u5d4c\u5957\uff0c\u540c\u65f6\u8868du_emr_test.empsalary tableB<\/strong>\u548c\u8868du_emr_test.hujh_type_tk AS tableB<\/strong>\u8bbe\u7f6e\u4e86\u540c\u4e00\u522b\u540d, \u5982\u56fe\u5728\u7236\u5b50\u67e5\u8be2\u4e2d\u90fd\u4f7f\u7528\u4e86\u540c\u4e00\u4e2a\u8868\u522b\u540d\uff08tableB\uff09\uff0c\u5f53\u7528\u6237\u5728\u7236\u5b50\u67e5\u8be2\u4e2d\u5206\u522b\u8f93\u5165tableB.<\/strong>\u65f6\uff0c\u8fd9\u65f6\u5019\u9700\u8981\u7ed3\u5408\u5f53\u524d\u4e0a\u4e0b\u6587\u8bed\u5883\uff0c\u5bf9tableB\u522b\u540d\u63a8\u8350\u4e0d\u540c\u8868\u7684\u5b57\u6bb5<\/strong>\u3002<\/p>\n \u5728\u5b50\u67e5\u8be2\u4e2d\u6211\u4eec\u671f\u671b\u63a8\u8350tableB\u6765\u81eadu_emr_test.empsalary tableB\u7684\u5b57\u6bb5\u4fe1\u606f\uff0c\u800c\u5728\u6700\u5916\u5c42\u4e2d\u6211\u4eec\u671f\u671b\u7684\u662fdu_emr_test.hujh_type_tk\u7684\u5b57\u6bb5\uff0c\u5982\u4e0a\u56fe\u3002<\/p>\n \u57fa\u4e8e\u4ee5\u4e0a\u573a\u666f\u6211\u4eec\u6838\u5fc3\u8981\u89e3\u51b32\u4e2a\u95ee\u9898\uff1a<\/p>\n \u95ee\u98981\uff1a\u5f53\u524d\u5149\u6807\u5e94\u8be5\u63d0\u793a\u54ea\u4e9b\u63a8\u8350\u8bed\u6cd5\u7c7b\u578b<\/strong><\/p>\n \u76ee\u524d\uff0c\u5f00\u6e90\u65b9\u6848ANTLR-C3\u5f15\u64ce\u5c31\u80fd\u5b8c\u7f8e\u89e3\u51b3\u6211\u4eec\u95ee\u9898\uff0c\u7528\u6237\u5728\u7f16\u8f91\u5668\u5b9e\u65f6\u8f93\u5165\u65f6\uff0c\u83b7\u53d6\u5f53\u524d\u5149\u6807\u4f4d\u7f6e\uff0c\u5b9e\u65f6\u505a\u8bed\u6cd5\u89e3\u6790\uff0c\u7136\u540e\u57fa\u4e8e\u5f00\u6e90\u7684ANTLR-C3\u5f15\u64ce\u80fd\u529b\u7ed3\u5408ANTLR \u751f\u6210\u7684AST\u5373\u53ef\u83b7\u53d6\u5f53\u524d\u5149\u6807\u4f4d\u7f6e\u6240\u9700\u8981\u7684\u8bed\u6cd5\u89c4\u5219<\/strong>\u3002<\/p>\n \u95ee\u98982: \u83b7\u53d6\u5f53\u524d\u4e0a\u4e0b\u6587\u4fe1\u606f\u4ee5\u5b9e\u73b0\u7cbe\u51c6\u63a8\u8350<\/strong><\/p>\n \u6839\u636e\u4e0d\u540c\u4e1a\u52a1\u573a\u666f\u9700\u8981\u91c7\u96c6\u7684\u4e0a\u4e0b\u6587\u4fe1\u606f\u4e0d\u540c\uff0c\u57fa\u4e8e\u5b57\u6bb5\u63a8\u8350\u7684\u573a\u666f\uff0c\u6211\u4eec\u9700\u8981\u83b7\u53d6\u5f53\u524d\u5149\u6807\u4f4d\u7f6e\u5904\u53ef\u4ee5\u63a8\u8350\u7684\u8868\u4fe1\u606f\uff0c\u8868\u522b\u540d\u4fe1\u606f\uff0c\u7ed3\u5408\u7f16\u8f91\u5668\u80fd\u529b\u5b9e\u65f6\u83b7\u53d6\u8868\u5bf9\u5e94\u7684\u5b57\u6bb5\u4fe1\u606f\u8fdb\u884c\u5b57\u6bb5\u63a8\u8350\u8865\u5168\uff0c\u800c\u4e0a\u4e0b\u6587\u4fe1\u606f\u7684\u91c7\u96c6<\/strong>\uff0c\u6211\u4eec\u53ef\u4ee5\u901a\u8fc7ANTLR\u751f\u6210\u7684\u76d1\u542c\u5668<\/strong>\u6765\u5b9e\u73b0\u3002<\/p>\n \u4ee5\u4e0b\u6211\u4eec\u7528ANTLR\u5143\u8bed\u8a00\u5b9e\u73b0\u4e00\u6bb5\u7b80\u5316\u7248\u7684SQL\u67e5\u8be2\u573a\u666f\u7684\u8bed\u6cd5\u89c4\u5219(QueryStatment)\uff0c\u65b9\u4fbf\u6211\u4eec\u7406\u89e3\u3002<\/p>\n \u4ee5\u4e0b\u662f\u90e8\u5206\u751f\u6210\u4ee3\u7801\uff1a<\/p>\n 1\u3001\u8bcd\u6cd5\u5206\u6790\u5668<\/strong><\/p>\n 2\u3001\u8bed\u6cd5\u5206\u6790\u5668<\/strong><\/p>\n ANTLR\u81ea\u52a8\u4e3a\u6bcf\u4e2a\u89c4\u5219\u751f\u6210\u4e86\u4e00\u4e2a\u89e3\u6790\u65b9\u6cd5\uff0c\u4ee5\u4e0b\u662ftableName\u7684 ANTLR \u4e2d\u7684\u89e3\u6790\u5668\u65b9\u6cd5\uff0c\u5177\u5907\u4e86\u5904\u7406\u6807\u8bc6\u7b26\u3001\u53ef\u9009\u7684\u522b\u540d\u548c\u9519\u8bef\u5904\u7406\u7684\u80fd\u529b\u3002<\/p>\n ANTLR4\u4ee3\u7801\u8865\u5168\u6838\u5fc3\uff08antlr4-c3\uff09 \u662f\u4e00\u4e2a\u5f00\u521b\u6027\u7684\u5de5\u5177\uff0c\u5b83\u4e3aANTLR4\u751f\u6210\u7684\u89e3\u6790\u5668\u63d0\u4f9b\u4e86\u4e00\u4e2a\u901a\u7528\u7684\u4ee3\u7801\u8865\u5168\u89e3\u51b3\u65b9\u6848\u3002\u65e0\u8bba\u4f60\u7684\u9879\u76ee\u662f\u5904\u7406\u54ea\u79cd\u7f16\u7a0b\u8bed\u8a00\u6216\u9886\u57df\u7279\u5b9a\u8bed\u8a00\uff08DSL\uff09\uff0c\u53ea\u8981\u662f\u57fa\u4e8eANTLR\u5c31\u80fd\u591f\u5229\u7528\u8fd9\u4e2a\u5e93\u5b9e\u73b0\u7cbe\u51c6\u7684\u4ee3\u7801\u5efa\u8bae\u548c\u81ea\u52a8\u8865\u5168\uff0c\u6781\u5927\u5730\u589e\u5f3a\u5f00\u53d1\u4f53\u9a8c\u3002\u901a\u8fc7antlr4-c3 \u80fd\u529b\u6211\u4eec\u901a\u8fc7\u624b\u52a8\u914d\u7f6e\u9700\u8981\u6536\u96c6\u7684\u8bed\u6cd5\u89c4\u5219\uff0c\u83b7\u53d6\u5728\u5f53\u524d\u5149\u6807\u5904\u9700\u8981\u63a8\u8350\u7684\u8bed\u6cd5\u89c4\u5219\u7c7b\u578b\u3002<\/p>\n 1\u3001\u8bed\u6cd5\u89c4\u5219<\/strong><\/p>\n \u901a\u8fc7ANTLR4\u5de5\u5177\u6211\u4eec\u53ef\u4ee5\u81ea\u52a8\u751f\u6210Sqllexer.ts\u8bcd\u6cd5\u89e3\u6790\u5668\uff0cSqlParser.ts\u8bed\u6cd5\u89e3\u6790\u5668\uff0cSqlParserLister.ts\u8bbf\u95ee\u5668\uff0cSqlParseVisitor.ts\u76d1\u542c\u5668\uff0c\u5728SqlParser \u8bed\u6cd5\u89e3\u6790\u5668\u81ea\u52a8\u751f\u6210\u4e86\u6211\u4eec\u5728\u8bed\u6cd5\u5b9a\u4e49\u4e2d\u7684\u8bed\u6cd5\u89c4\u5219\u3002<\/p>\n 2\u3001\u4ee3\u7801\u8865\u5168<\/strong><\/p>\n \u4ee5\u4e0b\u6211\u4eec\u5b9e\u73b0\u4e00\u5957\u7b80\u5316\u7248\u7684\u4ee3\u7801\u8865\u5168\u80fd\u529b\u3002<\/p>\n \u5f53\u7528\u6237\u5728\u7f16\u8f91\u5668\u5b9e\u65f6\u8f93\u5165\u65f6\uff0c\u8c03\u7528getSuggestionAtCaretPosition\u83b7\u53d6\u5f53\u524d\u8bed\u5883\u4e2d\u9700\u8981\u63a8\u8350\u7684\u4fe1\u606f\uff0c\u5305\u542b\u8bed\u6cd5\u89c4\u5219\uff0c\u5173\u952e\u8bcd\uff0c\u4e0a\u4e0b\u6587\u4fe1\u606f\uff0c\u5728\u7ed3\u5408\u4e1a\u52a1\u5c42\u6570\u636e\u505a\u81ea\u52a8\u8865\u5168\uff0c\u5176\u4e2d\u5305\u542b5\u4e2a\u6838\u5fc3\u6b65\u9aa4\uff1a<\/p>\n \u5728\u8fd9\u91cc\u6211\u4eec\u7b80\u5316\u4e86\u6d41\u7a0b\uff0c\u5ffd\u7565\u4e86\u5f88\u591a\u5f02\u5e38case\u7684\u5904\u7406\uff0c\u81ea\u52a8\u8865\u5168\u7684\u524d\u63d0\u662f\u5728\u5f53\u524d\u8bed\u6cd5\u89c4\u5219\u6b63\u786e\uff0c\u800c\u5728\u591a\u7ea7\u5b50\u67e5\u8be2\u5d4c\u5957\u573a\u666f\u6211\u4eec\u9700\u8981\u8003\u8651\u5230\u8fc7\u6ee4\u5f02\u5e38QueryStatment, \u5728\u5f53\u524d\u5149\u6807\u51fa\u6700\u5c0f\u8303\u56f4\u6709\u6548\u7684QueryStatment\u505a\u8865\u5168\u3002\u8fd9\u65f6\u5019\u9700\u8981\u914d\u5408\u76d1\u542c\u5668\u53bb\u505a\u4e0a\u4e0b\u6587\u91c7\u96c6\u505a\u5bb9\u9519\u6027\u66f4\u9ad8\u7684\u81ea\u52a8\u8865\u5168\u3002<\/p>\n \u5982\u56fe\uff1a\u6bcf\u4e2atable\u90fd\u5f52\u5c5e\u4e8e\u4e00\u4e2aQueryStatment\u8868\u8fbe\u5f0f, \u67e5\u8be2\u4e2d\u53c8\u5b58\u5728\u5b50\u5c42\u7ea7\u67e5\u8be2\u7684\u5d4c\u5957\u3002\u6211\u4eec\u9700\u8981\u901a\u8fc7\u4e0a\u4e0b\u6587\u6536\u96c6\u4ee5\u4e0b\u4fe1\u606f\uff1a<\/p>\n \u5219\u6211\u4eec\u9700\u8981\u76d1\u542c3\u4e2a\u8bed\u6cd5\u89c4\u5219\u5305\u542bQueryStatment, TableName,TableAlias<\/strong>, \u91c7\u96c6QueryStatment\u4fe1\u606f\uff0cTable\u4fe1\u606f\u540c\u65f6\u5c06table\u4e0e\u5f53\u524d\u5f52\u5c5e\u7684QueryStatment\u505a\u5173\u8054<\/strong>\uff0c \u8fd8\u6709\u4e0e\u522b\u540d\u4fe1\u606f\u4f5c\u914d\u5bf9\u5173\u8054<\/strong>\u3002\u8fd9\u5c31\u8981\u6c42\u5728\u4e0d\u540c\u76d1\u542c\u5668\u4e4b\u95f4\u7684\u4fe1\u606f\u9700\u8981\u505a\u5171\u4eab\uff0c\u4e0a\u4e0b\u6587\u4fe1\u606f\u9700\u8981\u505a\u4f20\u9012\u548c\u4fdd\u7559\u3002ANTLR\u5e38\u7528\u76843\u79cd\u4fe1\u606f\u5171\u4eab\u65b9\u6848<\/strong>\u5305\u542b\uff1a<\/p>\n \u5728\u8fd9\u91cc\u6211\u4eec\u4f7f\u7528\u7b2c\u4e8c\u79cd\uff08\u5728\u8fd9\u91cc\u6211\u4eec\u7b80\u5316\u4e86SQL\u7684\u8bed\u6cd5\u5b9a\u4e49\uff0c\u5728\u5b9e\u9645\u573a\u666f\u4e2d\u8bed\u6cd5\u5c42\u7ea7\u6df1\u5ea6\u548c\u590d\u6742\u5ea6\u8fdc\u6bd4\u5f53\u524d\u9ad8\uff0c\u8fd9\u4e5f\u4f7f\u5f97\u65b9\u68481\u548c3\u5b9e\u9645\u64cd\u4f5c\u8d77\u6765\u66f4\u9ebb\u70e6\uff0c\u89c4\u5219\u5d4c\u5957\u5c42\u7ea7\u6df1\u4f7f\u5f97\u65b9\u6848\u4e00\u548c\u65b9\u6848\u4e09\u5f00\u53d1\u6210\u672c\u548c\u7ef4\u62a4\u6210\u672c\u66f4\u9ad8\uff09<\/p>\n 1\u3001\u76d1\u542c\u5668\uff08SqlParserLister\uff09<\/strong> 2\u3001\u81ea\u5b9a\u4e49\u76d1\u542c\u5668\u6269\u5c55<\/strong><\/p>\n \u901a\u8fc7SqlParserListener\u6211\u4eec\u53ef\u4ee5\u81ea\u5b9a\u4e49\u91c7\u96c6\u4e0a\u4e0b\u6587\u4fe1\u606f\u3002\u5728<\/p>\n \u5728\u8fd9\u91cc\u6211\u4eec\u7b80\u5316\u4e86\u8bed\u6cd5\u5b9a\u4e49\u7684\u89c4\u5219\u4fbf\u4e8e\u8bb2\u89e3\uff0c\u4f46\u5728\u5b9e\u9645\u4e2d\u8bed\u6cd5\u89c4\u5219\u7684\u6574\u4f53\u5d4c\u5957\u5c42\u7ea7\u662f\u5f88\u6df1\u7684\uff0c\u4ece\u4ee5\u4e0b\u7684SparkSql\u8bed\u6cd5\u5b9a\u4e49\u4e2d\u6211\u4eec\u53ef\u4ee5\u770b\u5230\u53f3\u4fa7\u805a\u5408\u7684\u8868\u8fbe\u5f0f\u9ad8\u8fbe200+<\/strong>\u4e2a\uff0c\u5355\u4e2a\u8868\u8fbe\u5f0f\u7684\u5907\u9009\u5206\u652f\u6700\u591a\u9ad8\u8fbe140+<\/strong>\uff0c\u8fd9\u4e5f\u52a0\u5927\u4e86\u4e0a\u4e0b\u6587\u5206\u6790\u91c7\u96c6\u7684\u590d\u6742\u5ea6\uff0c\u5373\u6211\u4eec\u65e0\u6cd5\u7b80\u5355\u7684\u4eceQueryStmt\u5f53\u524dQueryStatmentContext<\/strong>\u4e2d\u83b7\u53d6\u5168\u91cf\u4fe1\u606f\u3002<\/p>\n 3\u3001\u89e6\u53d1\u76d1\u542c\u5668\u91c7\u96c6\u4e0a\u4e0b\u6587\u4fe1\u606f<\/strong><\/p>\n ANRLR\u5728\u751f\u6210\u8bed\u6cd5\u5206\u6790\u5668\u4e2d\u5185\u7f6e\u4e86\u81ea\u52a8\u9519\u8bef\u62a5\u544a\u548c\u6062\u590d\u7b56\u7565\uff0c\u80fd\u591f\u5728\u9047\u5230\u53e5\u6cd5\u9519\u8bef\u65f6\u81ea\u52a8\u4ea7\u751f\u9519\u8bef\u6d88\u606f\uff0c\u4e3a\u6bcf\u4e2a\u53e5\u6cd5\u9519\u8bef\u4ea7\u751f\u4e00\u6761\u9519\u8bef\u6d88\u606f\u3002<\/p>\n \u5e38\u89c1\u7684\u8bcd\u6cd5\u9519\u8bef\u5305\u542b\u5b57\u7b26\u9057\u6f0f\uff0c\u8bcd\u6cd5\u9519\u8bef\u3002\u4e3e\u4e2a\u4f8b\u5b50\uff0c\u5728spark\u6807\u51c6\u8bed\u6cd5\u5b9a\u4e49\u4e2d tableName\u89c4\u5219\u4e0d\u652f\u6301\u8868\u53d8\u91cf\u573a\u666f\uff08${variable}\uff09\uff0c\u5982\u679c\u8981\u517c\u5bb9\u8fd9\u91cc\u8bcd\u6cd5\uff0c\u5c31\u9700\u8981\u5728\u8bed\u6cd5\u5b9a\u4e49\u4e2d\u53d8\u66f4tableName\u7684\u8bed\u6cd5\u89c4\u5219\u5b9a\u4e49\u3002<\/p>\n \u4ee5\u4e0b\u662f\u8bed\u6cd5\u5b9a\u4e49\u53d8\u66f4:<\/p>\n \u8bed\u6cd5\u5206\u6790\u5668\u4e0d\u5e94\u8be5\u5728\u9047\u5230\u975e\u6cd5\u7684\u6210\u5458\u5b9a\u4e49\u65f6\u7ed3\u675f\uff0c\u800c\u662f\u5e94\u5c3d\u6700\u5927\u53ef\u80fd\u5339\u914d\u5230\u4e00\u4e2a\u5408\u6cd5\u7684\u7c7b\u5b9a\u4e49\uff0cANRTL4\u81ea\u52a8\u9519\u8bef\u6062\u590d\u673a\u5236\u80fd\u5728\u8bed\u6cd5\u5206\u6790\u5668\u5728\u53d1\u73b0\u8bed\u6cd5\u9519\u8bef\u540e\u8fd8\u80fd\u7ee7\u7eed\u8fdb\u884c\u5c1d\u8bd5\u8bed\u6cd5\u89e3\u6790\u548c\u81ea\u52a8\u6062\u590d\u3002<\/p>\n 1\u3001\u5f02\u5e38\u6355\u83b7<\/strong><\/p>\n ANRLT\u81ea\u52a8\u751f\u6210\u7684\u8bed\u6cd5\u89e3\u6790\u5668\u4e2d\u81ea\u52a8\u4e3a\u6bcf\u4e2a\u89c4\u5219\u5305\u88f9\u5f02\u5e38\u6355\u83b7\u80fd\u529b\uff0c\u5e76\u5728catch\u4e2d\u5c1d\u8bd5\u9519\u8bef\u6062\u590d\u3002<\/p>\n 2\u3001\u6062\u590d\u7b56\u7565<\/strong><\/p>\n \u4e00\u822c\u60c5\u51b5\u4e0b\uff0c\u8bed\u6cd5\u5206\u6790\u5668\u5728\u9047\u5230\u65e0\u6cd5\u5339\u914d\u7684\u9519\u8bef\u65f6\u4f1a\u5c1d\u8bd5\u6700\u7b80\u5355\u7684\u7b26\u53f7\u8865\u5168\u548c\u79fb\u9664\u6765\u5c1d\u8bd5\u89e3\u6790\uff0c\u90fd\u4e0d\u7ba1\u7528\u65f6\uff0c\u8fd9\u65f6\u5019\u5c31\u4f1a\u7528\u66f4\u9ad8\u9636\u7684\u7b56\u7565\u6765\u8fdb\u884c\u6062\u590d\u3002\u5305\u62ec\u626b\u63cf\u540e\u7eed\u8bcd\u6cd5\u7b26\u53f7\u6765\u6062\u590d\uff0c\u4ece\u4e0d\u5339\u914d\u7684\u8bcd\u6cd5\u7b26\u53f7\u4e2d\u6062\u590d\uff0c\u4ece\u5b50\u89c4\u5219\u7684\u9519\u8bef\u4e2d\u6062\u590d\uff0c\u6355\u83b7\u5931\u8d25\u7684\u8bed\u4e49\u5224\u5b9a\u3002<\/strong><\/p>\n \u867d\u7136ANTLR\u63d0\u4f9b\u4e86\u5f88\u591a\u7b56\u7565\u6765\u8fdb\u884c\u9519\u8bef\u6062\u590d\uff0c\u4f46\u5728\u5b9e\u9645\u4e1a\u52a1\u573a\u666f\u4e2d\uff0c\u9700\u8981\u7ed3\u5408\u8003\u8651\u8bed\u6cd5\u3001\u8bed\u5883\u7684\u590d\u6742\u5ea6\u53bb\u6743\u8861\u6027\u80fd\u4e0e\u66f4\u53cb\u597d\u7684\u9519\u8bef\u63d0\u793a\u4e4b\u95f4\u7684\u6289\u62e9\u3002\u5728\u590d\u6742\u573a\u666f\u4e2dANTLR\u8868\u73b0\u5e76\u4e0d\u7406\u60f3\uff0c\u5728\u4e00\u4e9b\u590d\u6742\u8bed\u6cd5\u548c\u8bed\u5883\u7684\u60c5\u51b5\u4e0b\u89e3\u6790\u5668\u5728\u68c0\u6d4b\u9519\u8bef\u65f6\u96be\u4ee5\u505a\u51fa\u5408\u7406\u7684\u51b3\u7b56\uff0c\u4f8b\u5982\uff1a\u9012\u5f52\u548c\u5d4c\u5957\u7ed3\u6784<\/strong>\u4e2d\u4f1a\u4f7f\u5f97\u9519\u8bef\u6062\u590d\u53d8\u5f97\u5f88\u590d\u6742\uff0c\u5bfc\u81f4\u89e3\u6790\u5668\u65e0\u6cd5\u505a\u51fa\u5408\u7406\u51b3\u7b56\u3002\u8fd8\u6709\u5728\u4e0a\u4e0b\u6587\u654f\u611f\u7684\u8bed\u5883<\/strong>\u4e2d\uff0c\u9519\u8bef\u6062\u590d\u673a\u5236\u57fa\u672c\u65e0\u6cd5\u63d0\u4f9b\u6709\u6548\u6062\u590d\u3002<\/p>\n \u5728 ANTLR 4 \u4e2d\uff0c\u8bed\u6cd5\u590d\u6742\u5ea6\u3001\u8bed\u6cd5\u6b67\u4e49\u3001\u8bed\u6cd5\u89c4\u5219\u5d4c\u5957\u6df1\u5ea6\u4e0e\u9884\u6d4b\u7b97\u6cd5\u7684\u9009\u62e9\u90fd\u4f1a\u663e\u8457\u5f71\u54cd\u89e3\u6790\u5668\u7684\u6027\u80fd\u548c\u51c6\u786e\u6027\u3002Spark SQL\u8bed\u6cd5\u89c4\u5219\u8fbe200+\uff0c\u5907\u9009\u5206\u652f\u6700\u9ad8\u8fbe140, \u5d4c\u5957\u6df1\u5ea6\u8fbe20+\uff0c\u540c\u65f6\u53c8\u5b58\u5728\u8d1f\u8d23\u5faa\u73af\u5d4c\u5957\u573a\u666f\uff0c \u8fd9\u4e5f\u610f\u5473\u7740\u5728\u6574\u4e2a\u8bed\u6cd5\u89e3\u6790\uff0c\u8bed\u6cd5\u9519\u8bef\u7684\u5904\u7406\u8fc7\u7a0b\u662f\u5f88\u590d\u6742\u7684\uff0c\u5f53\u9047\u5230\u590d\u6742\u5927SQL\u91cf\u548c\u4e00\u7247\u72fc\u7c4d\u7684\u8bed\u6cd5\u9519\u8befSQL\uff0c\u4f1a\u5bfc\u81f4\u8bed\u6cd5\u89e3\u6790\u8fc7\u7a0b\u53d8\u5f97\u7f13\u6162\u5f15\u53d1\u6027\u80fd\u95ee\u9898\u3002\u76ee\u524d\u5728\u6027\u80fd\u4f18\u5316\u4e0a\uff0c\u6709\u4ee5\u4e0b\u51e0\u4e2a\u65b9\u5411\u3002<\/p>\n \u5728antlr4\u4e2d\u8bcd\u6cd5\u89e3\u6790\u548c\u8bed\u6cd5\u89e3\u6790\u80fd\u529b\u548c\u4e1a\u52a1\u662f\u5b8c\u5168\u89e3\u8026<\/strong>\u7684\uff0c\u8fd9\u4e5f\u610f\u5473\u7740\u5e95\u5c42\u57fa\u4e8e\u540c\u4e2aSQL\u5185\u5bb9\u89e3\u6790\u51fa\u6765\u7684tokens\u548cparserTree\u90fd\u662f\u53ef\u4ee5\u5728\u4e0d\u540c\u4e1a\u52a1\u903b\u8f91<\/strong>\u5e94\u7528\u91cc\u590d\u7528<\/strong>\u3002\u6211\u4eec\u53ef\u4ee5\u901a\u8fc7\u7f13\u5b58tokens,parseTree\u51cf\u5c11\u8bcd\u6cd5\u89e3\u6790\u548c\u8bed\u6cd5\u89e3\u6790\u7684\u635f\u8017\u3002<\/p>\n \u901a\u8fc7\u51cf\u5c11\u8bed\u6cd5\u6811\u7684\u5c42\u7ea7\u548c\u4f18\u5316\u8868\u8fbe\u5f0f\u51cf\u5c11\u89e3\u6790\u8fc7\u7a0b\u4e2d\u201c\u4e8c\u4e49\u6027<\/strong>\u201d\u7684\u6b21\u6570\uff0c\u53ef\u4ee5\u52a0\u901f\u8bed\u6cd5\u89e3\u6790\u7684\u901f\u5ea6\uff0c\u4f18\u5316AST\u751f\u6210\u6027\u80fd\u3002\u5408\u7406\u4f7f\u7528\u8bed\u6cd5\u5b9a\u4e49\u4e2d\u7528\u6cd5\uff0c\u4f8b\u5982\u6811\u6807\u8bb0<\/strong>\uff08\u7528\u4e8e\u4e0a\u4e0b\u6587\u901a\u4fe1\u6570\u636e\u5171\u4eab\uff09\uff0c\u5728\u8bed\u6cd5\u89e3\u6790\u8fc7\u7a0b\u4e2d\u4f1a\u4e3a\u6bcf\u4e2a\u6807\u8bb0\u751f\u6210\u4e0a\u4e0b\u6587\uff0c\u8fd9\u4e5f\u610f\u5473\u7740\u6bcf\u4e2a\u5c40\u90e8\u7ed3\u679c\u90fd\u4f1a\u4fdd\u7559\uff0c\u4f1a\u6709\u66f4\u5927\u7684\u5185\u5b58\u6d88\u8017<\/strong>\u3002<\/p>\n \u5728\u8bed\u6cd5\u89e3\u6790\u4e2d\u4e0d\u540c\u9884\u6d4b\u6a21\u578b\u7684\u9009\u62e9\u5bf9\u89e3\u6790\u6027\u80fd\u6709\u663e\u8457\u5f71\u54cd\uff0c\u9488\u5bf9\u4e0d\u540c\u7684\u573a\u666f\u9700\u8981\u8bc4\u4f30\u65f6\u6548\u6027\u4e0e\u6b63\u786e\u6027\u4e4b\u95f4\u7684\u8861\u91cf\u3002<\/p>\n ANTLR4\u9884\u6d4b\u6a21\u578b\uff1a<\/p>\n https:\/\/www.antlr.org\/api\/Java\/org\/antlr\/v4\/runtime\/atn\/Predi…<\/a><\/p>\n \u6211\u4eec\u53ef\u4ee5\u9009\u62e9\u6027\u4ef7\u6bd4\u66f4\u9ad8\u7684SLL\u9884\u6d4b\u6a21\u578b\u4f5c\u4e3a\u8bed\u6cd5\u5206\u6790\u7b56\u7565\uff0c\u7ed3\u5408\u5b9a\u5236\u5316\u7684\u9519\u8bef\u76d1\u542c\u5668\u505a\u9519\u8bef\u7ea0\u6b63\u3002<\/p>\n
Twitter<\/strong>: Twitter \u4f7f\u7528ANTLR\u6765\u89e3\u6790\u548c\u5206\u6790\u7528\u6237\u7684\u67e5\u8be2\u8bed\u8a00\uff0c\u8fd9\u6709\u52a9\u4e8e\u4ed6\u4eec\u7684\u641c\u7d22\u548c\u5206\u6790\u529f\u80fd\u3002
IBM<\/strong>: IBM\u4f7f\u7528ANTLR\u6765\u652f\u6301\u4e00\u4e9b\u5176\u4ea7\u54c1\u548c\u5de5\u5177\u4e2d\u7684DSL\uff08\u9886\u57df\u7279\u5b9a\u8bed\u8a00\uff09\u89e3\u6790\u9700\u6c42\uff0c\u4f8b\u5982\uff0c\u5728\u5176\u4f01\u4e1a\u96c6\u6210\u89e3\u51b3\u65b9\u6848\u4e2d\u3002<\/p>\nANTLR4\u5165\u95e8<\/h3>\n
ANTLR\u5143\u8bed\u8a00<\/h4>\n
<\/p>\n
\u5728\u8bed\u8a00\u7ed3\u6784\u4e2d\uff0c\u6574\u4f53\u7684\u8fa8\u8bc6\u90fd\u662f\u4ece\u6700\u7c97\u7684\u7c92\u5ea6\u5f00\u59cb\uff0c\u4e00\u76f4\u8fdb\u884c\u5230\u6700\u8be6\u7ec6\u7684\u5c42\u6b21\uff0c\u5e76\u628a\u5b83\u4eec\u7f16\u5199\u6210\u4e3a\u8bed\u6cd5\u89c4\u5219\uff0cANTLR4\u5c31\u662f\u91c7\u7528\u81ea\u9876\u5411\u4e0b\u7684\uff0c\u8bcd\u6cd5\u8bed\u6cd5\u5206\u79bb\uff0c\u4e0a\u4e0b\u6587\u65e0\u5173\u7684\u8bed\u6cd5\u6846\u67b6\u6765\u63cf\u8ff0\u8bed\u8a00\u3002<\/p>\n\/\/ MyGLexer.g4\nlexer grammar MyGLexer;\n\nSEMICOLON: ';';\nLEFT_PAREN: '(';\nRIGHT_PAREN: ')';\nCOMMA: ',';\nDOT: '.';\nLEFT_BRACKET: '[';\nRIGHT_BRACKET: ']';\nLEFT_BRACES: '{';\nRIGHT_RACES: '}';\nEQ: '=';\n\nFUNCTOM: 'FUNCTION';\nLET: 'LET';\nCONST: 'CONST';\nVAR: 'VAR';\nIF: 'IF';\nELSE: 'ELSE';\nWHILE: 'WHILE';\nFOR: 'FOR';\nRETURN: 'RETURN';<\/code><\/pre>\n\/\/ MyGParser.g4\nparser grammar MyGParser;\n\noptions {\n tokenVocab = MyGLexer;\n}\n\n\/\/ \u5165\u53e3\u89c4\u5219\nprogram: statement* EOF;\n\nstatement:\n variableDeclaration\n | functionDeclaration\n | expressionStatement\n | blockStatement\n | ifStatement\n | whileStatement\n | forStatement\n | returnStatement;\n ......<\/code><\/pre>\n<\/p>\n
expr : expr '*' expr\n | expr '+' expr\n | INT\n ;\n<\/code><\/pre>\n<\/p>\n
\u4ee3\u7801\u81ea\u52a8\u751f\u6210<\/h3>\n
antlr4ng -Dlanguage=TypeScript -visitor -listener -Xexact-output-dir -o .\/src\/lib .\/src\/grammar\/*.g<\/code><\/pre>\n<\/p>\n
\u8bed\u6cd5\u89e3\u6790\u4e0e\u4e1a\u52a1\u903b\u8f91\u89e3\u8026<\/h4>\n
\n
\u4e09\u3001SparkSQL\u4ecb\u7ecd<\/h2>\n
\n
\u56db\u3001\u6280\u672f\u5b9e\u73b0<\/h2>\n
\u8bed\u6cd5\u8bbe\u8ba1<\/h3>\n
Lexer.g4\n\nhttps:\/\/github.com\/apache\/spark\/blob\/master\/sql\/api\/src\/main\/antlr4\/org\/apache\/spark\/sql\/catalyst\/parser\/SqlBaseLexer.g4\n\nParser.g4\n\nhttps:\/\/github.com\/apache\/spark\/blob\/master\/sql\/api\/src\/main\/antlr4\/org\/apache\/spark\/sql\/catalyst\/parser\/SqlBaseParser.g4<\/code><\/pre>\n\u8bed\u6cd5\u8865\u5168<\/h3>\n
SELECT \n tableB.c1\n FROM\n (\n SELECT\n tableB.empno,\n tableC.department\n FROM\n du_emr_test.empsalary as tableB\n LEFT JOIN du_emr_test.employees AS tableC\n WHERE tableC.department = tableB.depname\n\n ) AS tableA\nLEFT JOIN du_emr_test.hujh_type_tk AS tableB\nWHERE tableB.c1 = tableA.dename<\/code><\/pre>\n<\/p>\n
\u8bed\u6cd5\u5b9a\u4e49<\/h3>\n
lexer grammar SqlLexer;\n\n\/\/ \u57fa\u7840\u8bcd\u6cd5\nCOMMA: ',';\nLEFT_PAREN: '(';\nRIGHT_PAREN: ')';\nIDENTIFY: (LETTER | DIGIT | '_' | '.')+;\nfragment DIGIT: [0-9];\nfragment LETTER: [A-Z];\nSEMICOLON: ';';\n\nparser grammar SqlParser;\n\nprogram: statment* EOF;\n\nstatment: queryStatment SEMICOLON?;\n\/\/ \u67e5\u8be2\u8bed\u53e5\nqueryStatment:\n SELECT columnNames FROM (\n tableName\n | (LEFT_PAREN queryStatment LEFT_PAREN)\n ) whereExpression? relationsExpresssion? SEMICOLON\uff1f;\n\n\/\/ \u5b57\u6bb5\ncolumnNames: columnName (COMMA columnName)*;\n\ntableName: IDENTIFY AS? tableAlis;\n\ntableAlis: IDENTIFY;\n\ncolumnName: IDENTIFY AS? columnAlis;\n\ncolumnAlis: IDENTIFY;\n\nwhereExpression: WHERE booleanExpression;\n\nbooleanExpression: (NOT | BANG) booleanExpression # logicalBinary\n | left = booleanExpression operator = AND right = booleanExpression # logicalBinary\n | left = booleanExpression operator = OR right = booleanExpression # logicalBinary;\n\nrelationsExpresssion:\n LEFT JOIN tableName whereExpression?\n | RIGHT JOIN tableName whereExpression?;\n<\/code><\/pre>\n\u4ee3\u7801\u751f\u6210<\/h3>\n
<\/p>\n
\/\/ SqlLexer.ts\n \n public static readonly COMMA = 1;\n public static readonly LEFT_PAREN = 2;\n public static readonly RIGHT_PAREN = 3;\n public static readonly IDENTIFY = 4;\n public static readonly SEMICOLON = 5;\n\n \/\/ \u8bcd\u6cd5\u5206\u6790\u5668\u53ef\u4ee5\u4f7f\u7528\u7684\u901a\u9053\n public static readonly channelNames = [\n "DEFAULT_TOKEN_CHANNEL", "HIDDEN"\n ];\n \/\/ \u5305\u542b\u4e86\u6240\u6709\u5b57\u9762\u91cf\u8bb0\u53f7\u7684\u540d\u79f0\n public static readonly literalNames = [\n null, "','", "'('", "')'", null, "';'"\n ];\n \/\/ \u5305\u542b\u4e3a\u6bcf\u4e2a\u8bb0\u53f7\u5206\u914d\u7684\u7b26\u53f7\u540d\uff0c\u8fd9\u4e9b\u7b26\u53f7\u5728\u751f\u6210\u89e3\u6790\u5668\u65f6\u7528\u4e8e\u6807\u8bc6\u8bb0\u53f7\n public static readonly symbolicNames = [\n null, "COMMA", "LEFT_PAREN", "RIGHT_PAREN", "IDENTIFY", "SEMICOLON"\n ];\n \n \/\/ ANTLR \u751f\u6210\u7684\u7c7b\u4e2d\u7684\u4e00\u4e2a\u5b57\u6bb5\uff0c\u5217\u51fa\u4e86\u6240\u6709\u5b9a\u4e49\u7684\u89c4\u5219\n public static readonly ruleNames = [\n "COMMA", "LEFT_PAREN", "RIGHT_PAREN", "IDENTIFY", "DIGIT", "LETTER", \n "SEMICOLON",\n ];\n<\/code><\/pre>\n\/\/ SQLParse.ts\n\/\/ ANTLR\u81ea\u52a8\u751f\u6210\u4e86\u4e00\u4e2a\u89e3\u6790 SQL \u8868\u540d\u7684 ANTLR \u4e2d\u7684\u89e3\u6790\u5668\u65b9\u6cd5\uff0c\u5177\u5907\u4e86\u5904\u7406\u6807\u8bc6\u7b26\u3001\u53ef\u9009\u7684\u522b\u540d\u548c\u9519\u8bef\u5904\u7406\u7684\u80fd\u529b\npublic tableName(): TableNameContext {\n let localContext = new TableNameContext(this.context, this.state);\n this.enterRule(localContext, 8, SqlParser.RULE_tableName);\n let _la: number;\n try {\n this.enterOuterAlt(localContext, 1);\n {\n this.state = 60;\n this.match(SqlParser.IDENTIFY);\n this.state = 62;\n this.errorHandler.sync(this);\n _la = this.tokenStream.LA(1);\n if (_la === 8) {\n {\n this.state = 61;\n this.match(SqlParser.AS);\n }\n }\n\n this.state = 64;\n this.tableAlis();\n }\n }\n catch (re) {\n if (re instanceof antlr.RecognitionException) {\n this.errorHandler.reportError(this, re);\n this.errorHandler.recover(this, re);\n } else {\n throw re;\n }\n }\n finally {\n this.exitRule();\n }\n return localContext;\n }\n<\/code><\/pre>\n\u81ea\u52a8\u8865\u5168<\/h4>\n
preferredRules = new Set([\n SqlParser.RULE_tableName,\n SqlParser.RULE_columnName,\n]);<\/code><\/pre>\n\n
public getSuggestionAtCaretPosition(\n sqlContent: string,\n caretPosition: CaretPosition\n preferredRules: Set\n ): Suggestions | null {\n \n \/\/ 1\u3001 \u4f7f\u7528SqlParse\u89e3\u6790\u5668\u83b7\u53d6\n const sqlParserIns = new SqlParse(sqlContent)\n \n \/\/ 2\u3001\u83b7\u53d6\u5f53\u524d\u5149\u6807\u5904token\n const charStreams = CharStreams.fromString(sqlContent);\n const lexer = new SqlLexer(charStreams);\n const tokenStream = new CommonTokenStream(lexer)\uff1b\n tokenStream.fill()\n const allTokens = tokenStream.getTokens(); \n let caretTokenIndex = findCaretToken(caretPosition, allTokens); \n\n \/\/ 3\u3001\u83b7\u53d6AST\u62bd\u8c61\u8bed\u6cd5\u6811\n const parseTree = sqlParserIns.program()\n \n \/\/ 4\u3001\u901a\u8fc7\u76d1\u542c\u5668\u91c7\u96c6\u4e0a\u4e0b\u6587\u8868\u4fe1\u606f\uff08\u4e0b\u9762\u4e0a\u4e0b\u6587\u5206\u6790\u90e8\u5206\u9610\u8ff0\u7ec6\u8282\uff09\n const tableEntity = getTableEntitys()\n \n \/\/ \u5f02\u5e38\u573a\u666f\u517c\u5bb9\u5b58\u5728\u591a\u6761sql\uff0c \u83b7\u53d6\u6709\u6548\u6700\u5c0fSQL\u8303\u56f4\u7ed9\u5230antlr4-c3\u505a\u63a8\u8350\u3002\n const statementCount = splitListener.statementsContext?.length;\n const statementsContext = splitListener.statementsContext; \n\n \/\/ 5\u3001antlr4-c3\u63a5\u5165\u83b7\u53d6\u63a8\u8350\u8bed\u6cd5\u89c4\u5219\n let tokenIndexOffset: number = 0;\n const core = new CodeCompletionCore(sqlParserIns);\n \/\/ \u63a8\u8350\u89c4\u5219 \u6765\u81eaSQLparse\u89e3\u6790\u5668\u7684\u89c4\u5219\uff08\u5143\u8bed\u8a00\u5b9a\u4e49\uff09\n core.preferredRules = preferredRules; \n \/\/ \u901a\u8fc7AST\u548c\u5f53\u524d\u5149\u6807Token\u83b7\u53d6\u63a8\u8350\u7c7b\u578b\n const candidates = core.collectCandidates(caretTokenIndex, parseTree); \n \n \/\/ ruleType -> preferredRules \n \/\/ const [rules, tokens] = candidate;\n const rules = [];\n const keywords = [\n \n for (let candidate of candidates.rules) {\n const [ruleType] = candidate;\n let synContextType;\n switch (ruleType) {\n case SqlParser.RULE_tableName: {\n syntaxContextType = 'table';\n break;\n }\n case SqlParser.RULE_columnName: {\n syntaxContextType = 'column';\n break;\n }\n default:\n break;\n }\n if (synContextType) {\n rules.push(syntaxContextType)\n }\n }\n\n \/\/ \u83b7\u53d6\u5bf9\u5e94keywords\n for (let candidate of candidates.tokens) {\n const displayName = sqlParserIns.vocabulary.getDisplayName(candidate[0]);\n const keyword = displayName.startsWith("'") && displayName.endsWith("'")\n ? displayName.slice(1, -1)\n : displayName\n keywords.push(keyword);\n }\n\n return {\n rules,\n keywords,\n tableEntity\n };\n }<\/code><\/pre>\n\u4e0a\u4e0b\u6587\u5206\u6790<\/h4>\n
<\/p>\n
\n
\n
\u901a\u8fc7ANTLR4\u5de5\u5177\u6211\u4eec\u53ef\u4ee5\u81ea\u52a8\u751f\u6210SqlParserLister.ts\u76d1\u542c\u5668\u8fdb\u884c\u81ea\u5b9a\u4e49\u6269\u5c55\u3002<\/p>\n\/\/ SqlParserListener.ts\nexport class QueryStatmentContext extends antlr.ParserRuleContext {\n public override enterRule(listener: SqlParserListener): void {\n if(listener.enterQueryStatment) {\n listener.enterQueryStatment(this);\n }\n }\n public override exitRule(listener: SqlParserListener): void {\n if(listener.exitQueryStatment) {\n listener.exitQueryStatment(this);\n }\n }\n }\n \n export class TableNameContext extends antlr.ParserRuleContext {\n public override enterRule(listener: SparkSqlParserListener): void {\n if(listener.enterTableName) {\n listener.enterTableName(this);\n }\n }\n public override exitRule(listener: SparkSqlParserListener): void {\n if(listener.exitTableName) {\n listener.exitTableName(this);\n }\n }\n }\n\/\/ ....\n\nexport class TableAliasContext extends antlr.ParserRuleContext {\n public KW_AS(): antlr.TerminalNode | null {\n return this.getToken(SparkSqlParser.KW_AS, 0);\n }\n public override enterRule(listener: SparkSqlParserListener): void {\n if(listener.enterTableAlias) {\n listener.enterTableAlias(this);\n }\n }\n public override exitRule(listener: SparkSqlParserListener): void {\n if(listener.exitTableAlias) {\n listener.exitTableAlias(this);\n }\n }\n}\n<\/code><\/pre>\n\n
\/\/ tableEntityCollect\n export class SqlEntityCollector implements SqlParserListener {\n super() {\n this._tableEntitiesSet = new Set();\n this._queryStmtsStack = [];\n this._tableAliasStack = [];\n this._currentTable = '';\n }\n \n enterQueryStatment(ctx: QueryStatmentContext) {\n this.pushQueryStmt(ctx);\n }\n\n exitQueryStatment(ctx: QueryStatmentContext) {\n this.popQueryStmt(); \n }\n\n exitTableName(ctx: TableNameContext) {\n this.pushTableEntity(ctx);\n this.setCurrentTable(ctx);\n }\n \n exitTableAlias(ctx: TableAliasContext) {\n this.pushTableEntity(ctx);\n }\n \n pushQueryStmt() {} \/\/ \u91c7\u96c6QueryStmt\u4fe1\u606f\n \n popQueryStmt() {} \/\/ \u63a8\u51fa\u5f53\u524dQueryStmt\uff0c\u8fdb\u5165\u4e0b\u4e2a\u540c\u7ea7Stmt\n \n pushTableEntity() {} \/\/ \u91c7\u96c6\u5f53\u524d\u8868\u4fe1\u606f\uff0c\u5173\u8054\u5f53\u524dStmt\n \n pushTableEntity() {} \/\/ \u91c7\u96c6\u5173\u8054\u8868\n \n enterProgram() {} \/\/ \u6e05\u7a7a\u91cd\u7f6e\n \n getTableEntity() {\n return this.TableEntity(ctx)\n }\n \n }\n<\/code><\/pre>\n<\/p>\n
getTableEntitys() {\n const collectListener = new SqlEntityCollector(sqlContent, caretTokenIndex);\n const parse = new SqlParse(sqlContent);\n const parseTree= sqlParserIns.program();\n ParseTreeWalker.DEFAULT.walk(collectListener, parseTree); \n return collectListener.getTableEntity()\n}\n<\/code><\/pre>\n\u8bed\u6cd5\u6821\u9a8c<\/h3>\n
\u8bcd\u6cd5\u9519\u8bef<\/h4>\n
<\/p>\n
\n
SqlLexer.g4\n\/\/ \u65b0\u589e\u8bcd\u6cd5\nLEFT_BRACE : '{';\nRIGHT_BRACE : '}';\n\nVARIABLE : '$';\n\nSqlParse.g4\n\/\/ before tableName: IDENTIFY AS? tableAlis; \ntableName: identifyVar AS? tableAlis; \n\nidentifyVar\n : IDENTIFY \/\/ odps_table_a\n | IDENTIFY? VARIABLE LEFT_BRACE IDENTIFY RIGHT_BRACE IDENTIFY? \/\/ odps_table_a_${variable} odps_table_a_${prefix_variable}_abs\n<\/code><\/pre>\n\u81ea\u52a8\u6062\u590d\u673a\u5236<\/h4>\n
<\/p>\n
\u6027\u80fd<\/h3>\n
\u7f13\u5b58\u4f18\u5316<\/h4>\n
\u8bed\u6cd5\u4f18\u5316<\/h4>\n
\u9884\u6d4b\u6a21\u578b\u9009\u62e9<\/h4>\n
<\/p>\n
\u7f16\u8f91\u5668\u5e94\u7528<\/h3>\n
\u7f16\u8f91\u5668\u96c6\u6210<\/h4>\n