Examples with DB2StatementParser com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser used on opensource projects

Example 16 with DB2StatementParser

use of com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser in project druid by alibaba.

the class DB2SelectTest_2 method test_0.

public void test_0() throws Exception {
    String sql = "SELECT CTRYNUM, FMS_INSTANCE_CD FROM DBEFMSDR.FMSA_O_WW_CTRY_AG WHERE ACCT_YR=? WITH UR";
    DB2StatementParser parser = new DB2StatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    print(statementList);
    Assert.assertEquals(1, statementList.size());
    DB2SchemaStatVisitor visitor = new DB2SchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    //        System.out.println("orderBy : " + visitor.getOrderByColumns());
    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(3, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("DBEFMSDR.FMSA_O_WW_CTRY_AG")));
    Assert.assertTrue(visitor.getColumns().contains(new Column("DBEFMSDR.FMSA_O_WW_CTRY_AG", "CTRYNUM")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "first_name")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "full_name")));
    String output = SQLUtils.toSQLString(stmt, JdbcConstants.DB2);
    Assert.assertEquals(//
    "SELECT CTRYNUM, FMS_INSTANCE_CD" + //
    "\nFROM DBEFMSDR.FMSA_O_WW_CTRY_AG" + //
    "\nWHERE ACCT_YR = ?" + //
    "\nWITH UR", output);
}

Example 17 with DB2StatementParser

use of com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser in project druid by alibaba.

the class DB2SelectTest_20 method test_0.

public void test_0() throws Exception {
    String sql = "SELECT EMPNO.PREVVAL FROM SYSIBM.SYSDUMMY1";
    DB2StatementParser parser = new DB2StatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    print(statementList);
    Assert.assertEquals(1, statementList.size());
    DB2SchemaStatVisitor visitor = new DB2SchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    //        System.out.println("orderBy : " + visitor.getOrderByColumns());
    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(0, visitor.getColumns().size());
    Assert.assertEquals(0, visitor.getConditions().size());
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("SYSIBM.SYSDUMMY1")));
    //         Assert.assertTrue(visitor.getColumns().contains(new Column("DSN8B10.EMP", "WORKDEPT")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "first_name")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "full_name")));
    Assert.assertEquals("SELECT EMPNO.PREVVAL" + //
    "\nFROM SYSIBM.SYSDUMMY1", SQLUtils.toSQLString(stmt, JdbcConstants.DB2));
    Assert.assertEquals("select EMPNO.prevval" + //
    "\nfrom SYSIBM.SYSDUMMY1", SQLUtils.toSQLString(stmt, JdbcConstants.DB2, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
}

Example 18 with DB2StatementParser

use of com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser in project druid by alibaba.

the class DB2SelectTest_22 method test_0.

public void test_0() throws Exception {
    String sql = "SELECT ID,IDENTITY,BUSICODE" + " FROM BUSINESS_ENTERPRISE_BUSIINFO" + " WHERE SYSCODE= '603'" + " ORDER BY INTIME desc" + " FETCH first 1 rows only";
    DB2StatementParser parser = new DB2StatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    print(statementList);
    Assert.assertEquals(1, statementList.size());
    DB2SchemaStatVisitor visitor = new DB2SchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    //        System.out.println("orderBy : " + visitor.getOrderByColumns());
    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(4, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("BUSINESS_ENTERPRISE_BUSIINFO")));
    //         Assert.assertTrue(visitor.getColumns().contains(new Column("DSN8B10.EMP", "WORKDEPT")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "first_name")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "full_name")));
    Assert.assertEquals("SELECT ID, IDENTITY, BUSICODE" + "\nFROM BUSINESS_ENTERPRISE_BUSIINFO" + "\nWHERE SYSCODE = '603'" + "\nORDER BY INTIME DESC" + //
    "\nFETCH FIRST 1 ROWS ONLY", SQLUtils.toSQLString(stmt, JdbcConstants.DB2));
    Assert.assertEquals("select ID, IDENTITY, BUSICODE" + "\nfrom BUSINESS_ENTERPRISE_BUSIINFO" + "\nwhere SYSCODE = '603'" + "\norder by INTIME desc" + //
    "\nfetch first 1 rows only", SQLUtils.toSQLString(stmt, JdbcConstants.DB2, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
}

Example 19 with DB2StatementParser

use of com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser in project druid by alibaba.

the class DB2SelectTest_23 method test_0.

public void test_0() throws Exception {
    String sql = "select current date,current timestamp from dual";
    DB2StatementParser parser = new DB2StatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    System.out.println(SQLUtils.toDB2String(stmt));
    Assert.assertEquals(1, statementList.size());
    DB2SchemaStatVisitor visitor = new DB2SchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    //        System.out.println("orderBy : " + visitor.getOrderByColumns());
    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(0, visitor.getConditions().size());
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("dual")));
    //         Assert.assertTrue(visitor.getColumns().contains(new Column("DSN8B10.EMP", "WORKDEPT")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "first_name")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "full_name")));
    Assert.assertEquals("SELECT CURRENT DATE, CURRENT DATE\n" + //
    "FROM dual", SQLUtils.toSQLString(stmt, JdbcConstants.DB2));
    Assert.assertEquals("select CURRENT DATE, CURRENT DATE\n" + //
    "from dual", SQLUtils.toSQLString(stmt, JdbcConstants.DB2, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
}

Example 20 with DB2StatementParser

use of com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser in project druid by alibaba.

the class DB2SelectTest_3 method test_0.

public void test_0() throws Exception {
    String sql = "select * from t for read only";
    DB2StatementParser parser = new DB2StatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    print(statementList);
    Assert.assertEquals(1, statementList.size());
    DB2SchemaStatVisitor visitor = new DB2SchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    //        System.out.println("orderBy : " + visitor.getOrderByColumns());
    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(0, visitor.getConditions().size());
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("t")));
    Assert.assertTrue(visitor.getColumns().contains(new Column("t", "*")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "first_name")));
    // Assert.assertTrue(visitor.getColumns().contains(new Column("mytable", "full_name")));
    String output = SQLUtils.toSQLString(stmt, JdbcConstants.DB2);
    Assert.assertEquals(//
    "SELECT *" + //
    "\nFROM t" + //
    "\nFOR READ ONLY", output);
}