feat: collect entity (#265)
* feat: add text and word utils * feat: add entity collector class * refactor: rename SyntaxContextType to EntityContextType * refactor: improve EntityCollector * feat: improve mysql parser grammar * feat: add mysql entity collector * test: mysql entity collector tests * feat: remove useless method * feat: improve spark grammar file * feat: add spark entity collector * test: spark entity collector unit tests * feat: remove useless code * feat: add queryStatement label * feat: add crateDatabaseStmt * feat: add trino entity collector * feat: rename trinosql to trino * test: trino collect entity unit tests * test: fix spark test * feat(impala): support impale entity collector (#256) * Feat/collect entity hive (#263) * feat(hive): support hive collect entity * feat(hive): update tableAllColumns * feat: replace antlr4ts with antlr4ng * feat(pgsql): pgsql collect entity (#268) * feat(pgsql): pgsql collect entity * feat(pgsql): optimize some name --------- Co-authored-by: zhaoge <> * feat: get word text by token.text * feat: supprt collect db/function and add splitListener (#270) * feat: supprt collect db/function and add splitListner * feat: remove SplitListener interface in baseParser to use SplitListener in root * fix(mysql): fix show create xxx not celloct as createXXXEntity type * test: fix pgsql unit tests * Feat/error recover predicate (#274) * feat: optimize pgsql grammar * feat: add sql parser base * feat: apply SQLParserBase * feat: add geAllEntities method * test: test collect table when missing column * feat: compose collect and suggestion (#276) * feat: mark stmt which contain caret * test: correct name of getAllEntities * test: remove misscolumn unit tests * test: add suggestionWithEntity tests * feat: flink collect entity (#277) * feat: improve flink sql parser * feat: support flink entity collector * test: flink entity collect unit test * feat: move combine entities to parent class --------- Co-authored-by: 霜序 <976060700@qq.com> Co-authored-by: XCynthia <942884029@qq.com>
This commit is contained in:
		
							
								
								
									
										22
									
								
								test/parser/trino/suggestion/fixtures/multipleStatement.sql
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								test/parser/trino/suggestion/fixtures/multipleStatement.sql
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,22 @@
 | 
			
		||||
CREATE TABLE  VALUES -- unfinished
 | 
			
		||||
 | 
			
		||||
ALTER SCHEMA foo RENAME TO bar;
 | 
			
		||||
 | 
			
		||||
DELETE FROM t;
 | 
			
		||||
 | 
			
		||||
DENY SELECT ON SCHEMA s TO USER u;
 | 
			
		||||
 | 
			
		||||
SELECT ids FROM db. ; -- unfinished
 | 
			
		||||
 | 
			
		||||
INSERT INTO weather (date, city, temp_hi, temp_lo) VALUES ('1994-11-29', 'Hayward', 54, 37);
 | 
			
		||||
 | 
			
		||||
EXPLAIN ANALYZE VERBOSE SELECT * FROM t;
 | 
			
		||||
 | 
			
		||||
INSERT INTO  weather (date, city, temp_hi, temp_lo) VALUES ('1994-11-29', 'Hayward', 54, 37); -- unfinished
 | 
			
		||||
 | 
			
		||||
DENY SELECT ON SCHEMA s TO USER u;
 | 
			
		||||
 | 
			
		||||
CALL catalog.schema.test();
 | 
			
		||||
 | 
			
		||||
INSERT INTO products (product_no, name, price) SELECT * FROM db. ; -- unfinished
 | 
			
		||||
 | 
			
		||||
@ -0,0 +1,11 @@
 | 
			
		||||
SELECT  FROM my_db.tb;
 | 
			
		||||
 | 
			
		||||
SELECT name, calculate_age(birthdate) AS age,  FROM students;
 | 
			
		||||
 | 
			
		||||
INSERT INTO insert_tb SELECT  FROM from_tb;
 | 
			
		||||
 | 
			
		||||
INSERT INTO insert_tb SELECT id, age,  FROM from_tb;
 | 
			
		||||
 | 
			
		||||
CREATE TABLE sorted_census_data AS SELECT  FROM unsorted_census_data;
 | 
			
		||||
 | 
			
		||||
CREATE TABLE sorted_census_data AS SELECT id, age,  FROM unsorted_census_data;
 | 
			
		||||
							
								
								
									
										37
									
								
								test/parser/trino/suggestion/fixtures/syntaxSuggestion.sql
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								test/parser/trino/suggestion/fixtures/syntaxSuggestion.sql
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,37 @@
 | 
			
		||||
INSERT INTO db.tb ;
 | 
			
		||||
 | 
			
		||||
SELECT ids FROM db.;
 | 
			
		||||
 | 
			
		||||
CREATE TABLE db. VALUES;
 | 
			
		||||
 | 
			
		||||
DROP TABLE IF EXISTS db.a;
 | 
			
		||||
 | 
			
		||||
CREATE OR REPLACE VIEW db.v;
 | 
			
		||||
 | 
			
		||||
DROP VIEW db.v ;
 | 
			
		||||
 | 
			
		||||
SELECT name, calculate_age(birthday) AS age FROM students;
 | 
			
		||||
 | 
			
		||||
CREATE SCHEMA db ;
 | 
			
		||||
 | 
			
		||||
DROP SCHEMA IF EXISTS sch;
 | 
			
		||||
 | 
			
		||||
SHOW COLUMNS FROM tb ;
 | 
			
		||||
 | 
			
		||||
COMMENT ON COLUMN tb.;
 | 
			
		||||
 | 
			
		||||
ALTER TABLE tb RENAME COLUMN ;
 | 
			
		||||
 | 
			
		||||
ALTER TABLE tb RENAME COLUMN ids TO ;
 | 
			
		||||
 | 
			
		||||
ALTER TABLE users DROP COLUMN ;
 | 
			
		||||
 | 
			
		||||
ALTER TABLE users ADD COLUMN zi ;
 | 
			
		||||
 | 
			
		||||
SHOW COMMENT ON COLUMN tb.c ;
 | 
			
		||||
 | 
			
		||||
INSERT INTO tb (id, );
 | 
			
		||||
 | 
			
		||||
SELECT * FROM tb ORDER BY ;
 | 
			
		||||
 | 
			
		||||
SELECT * FROM tb GROUP BY ;
 | 
			
		||||
							
								
								
									
										13
									
								
								test/parser/trino/suggestion/fixtures/tokenSuggestion.sql
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								test/parser/trino/suggestion/fixtures/tokenSuggestion.sql
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,13 @@
 | 
			
		||||
ALTER ;
 | 
			
		||||
 | 
			
		||||
CREATE ;
 | 
			
		||||
 | 
			
		||||
DEALLOCATE ;
 | 
			
		||||
 | 
			
		||||
DELETE ;
 | 
			
		||||
 | 
			
		||||
DESCRIBE ;
 | 
			
		||||
 | 
			
		||||
DROP ;
 | 
			
		||||
 | 
			
		||||
INSERT ;
 | 
			
		||||
		Reference in New Issue
	
	Block a user