3
3
import net .sf .jsqlparser .JSQLParserException ;
4
4
import net .sf .jsqlparser .parser .CCJSqlParserUtil ;
5
5
import net .sf .jsqlparser .statement .Statement ;
6
- import net .sf .jsqlparser .statement .StatementVisitorAdapter ;
7
6
import net .sf .jsqlparser .statement .delete .Delete ;
8
7
import net .sf .jsqlparser .statement .insert .Insert ;
9
8
import net .sf .jsqlparser .statement .select .Select ;
@@ -15,29 +14,17 @@ public class JSQLHibernateStatementParser implements HibernateStatementParser {
15
14
public void parseSqlStatement (String sql , HibernateStatementListener statementCountListener ) {
16
15
try {
17
16
Statement statement = CCJSqlParserUtil .parse (sql );
18
- statement .accept (new StatementVisitorAdapter () {
19
- @ Override
20
- public void visit (Select select ) {
21
- statementCountListener .notifySelectStatement (sql );
22
- }
23
17
24
- @ Override
25
- public void visit (Insert insert ) {
26
- statementCountListener .notifyInsertStatement (sql );
27
- }
28
-
29
- @ Override
30
- public void visit (Update update ) {
31
- statementCountListener .notifyUpdateStatement (sql );
32
- }
33
-
34
- @ Override
35
- public void visit (Delete delete ) {
36
- statementCountListener .notifyDeleteStatement (sql );
37
- }
38
- });
39
- } catch (JSQLParserException e ) {
40
- throw new RuntimeException (e );
18
+ if (statement instanceof Delete ) {
19
+ statementCountListener .notifyDeleteStatement (sql );
20
+ } else if (statement instanceof Insert ) {
21
+ statementCountListener .notifyInsertStatement (sql );
22
+ } else if (statement instanceof Select ) {
23
+ statementCountListener .notifySelectStatement (sql );
24
+ } else if (statement instanceof Update ) {
25
+ statementCountListener .notifyUpdateStatement (sql );
26
+ }
27
+ } catch (JSQLParserException ignored ) {
41
28
}
42
29
}
43
30
}
0 commit comments