Пример #1
0
  public void test_0() throws Exception {
    String sql = "SELECT 1--1 from t where id = 1";

    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);

    SQLSelectStatement selectStmt = (SQLSelectStatement) stmt;

    SQLSelect select = selectStmt.getSelect();
    Assert.assertNotNull(select.getQuery());
    MySqlSelectQueryBlock queryBlock = (MySqlSelectQueryBlock) select.getQuery();
    Assert.assertNull(queryBlock.getOrderBy());

    //        print(statementList);

    Assert.assertEquals(1, statementList.size());

    MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
    stmt.accept(visitor);

    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertEquals(0, visitor.getOrderByColumns().size());

    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("t")));

    String output = SQLUtils.toMySqlString(stmt);
    Assert.assertEquals(
        "SELECT 1 - -1" //
            + "\nFROM t" //
            + "\nWHERE id = 1", //
        output);
  }
Пример #2
0
  public void test_0() throws Exception {
    String sql = "SELECT * FROM t_department  WHERE name IN ('0000','4444') ORDER BY name ASC";

    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);

    SQLSelectStatement selectStmt = (SQLSelectStatement) stmt;

    SQLSelect select = selectStmt.getSelect();
    Assert.assertNotNull(select.getQuery());
    MySqlSelectQueryBlock queryBlock = (MySqlSelectQueryBlock) select.getQuery();
    Assert.assertNotNull(queryBlock.getOrderBy());

    //        print(statementList);

    Assert.assertEquals(1, statementList.size());

    MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
    stmt.accept(visitor);

    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(2, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertEquals(1, visitor.getOrderByColumns().size());

    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("t_department")));
  }
Пример #3
0
  public void test_0() throws Exception {
    String sql = "SELECT `group`.* FROM `group` WHERE (group.group_id=159754)";

    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);

    SQLSelectStatement selectStmt = (SQLSelectStatement) stmt;

    SQLSelect select = selectStmt.getSelect();
    Assert.assertNotNull(select.getQuery());
    MySqlSelectQueryBlock queryBlock = (MySqlSelectQueryBlock) select.getQuery();
    Assert.assertNull(queryBlock.getOrderBy());

    print(statementList);

    Assert.assertEquals(1, statementList.size());

    MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
    stmt.accept(visitor);

    Assert.assertEquals(1, visitor.getTables().size());
    Assert.assertEquals(2, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertEquals(0, visitor.getOrderByColumns().size());

    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("group")));
  }
Пример #4
0
  public void test_0() throws Exception {
    String sql = "SELECT * FROM t1 WHERE column1 = (SELECT column1 FROM t2);";

    MySqlStatementParser parser = new MySqlStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);

    SQLSelectStatement selectStmt = (SQLSelectStatement) stmt;

    SQLSelect select = selectStmt.getSelect();
    Assert.assertNotNull(select.getQuery());
    MySqlSelectQueryBlock queryBlock = (MySqlSelectQueryBlock) select.getQuery();
    Assert.assertNull(queryBlock.getOrderBy());

    //        print(statementList);

    Assert.assertEquals(1, statementList.size());

    MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
    stmt.accept(visitor);

    Assert.assertEquals(2, visitor.getTables().size());
    Assert.assertEquals(3, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getConditions().size());
    Assert.assertEquals(0, visitor.getOrderByColumns().size());

    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("t1")));
    Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("t2")));

    String output = SQLUtils.toMySqlString(stmt);
    Assert.assertEquals(
        "SELECT *" //
            + "\nFROM t1" //
            + "\nWHERE column1 = (" //
            + "\n\tSELECT column1" //
            + "\n\tFROM t2" //
            + "\n\t)", //
        output);

    String output_lcase = SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION);
    Assert.assertEquals(
        "select *" //
            + "\nfrom t1" //
            + "\nwhere column1 = (" //
            + "\n\tselect column1" //
            + "\n\tfrom t2" //
            + "\n\t)", //
        output_lcase);
  }
Пример #5
0
  public boolean visit(MySqlSelectQueryBlock select) {
    print("SELECT ");

    if (SQLSetQuantifier.ALL == select.getDistionOption()) print("ALL ");
    else if (SQLSetQuantifier.DISTINCT == select.getDistionOption()) print("DISTINCT ");
    else if (SQLSetQuantifier.DISTINCTROW == select.getDistionOption()) {
      print("DISTINCTROW ");
    }

    if (select.isHignPriority()) {
      print("HIGH_PRIORITY ");
    }

    if (select.isSmallResult()) {
      print("SQL_SMALL_RESULT ");
    }

    if (select.isBigResult()) {
      print("SQL_BIG_RESULT ");
    }

    if (select.isBufferResult()) {
      print("SQL_BUFFER_RESULT ");
    }

    if (select.getCache() != null) {
      if (select.getCache().booleanValue()) {
        print("SQL_CACHE ");
      } else {
        print("SQL_NO_CACHE ");
      }
    }

    if (select.isCalcFoundRows()) {
      print("SQL_CALC_FOUND_ROWS ");
    }

    printSelectList(select.getSelectList());

    if (select.getOutFile() != null) {
      println();
      print("INTO OUTFILE ");
      select.getOutFile().accept(this);
      if (select.getOutFileCharset() != null) {
        print(" CHARACTER SET ");
        print(select.getOutFileCharset());
      }

      if (select.getOutFileColumnsTerminatedBy() != null
          || select.getOutFileColumnsEnclosedBy() != null
          || select.getOutFileColumnsEscaped() != null) {
        print(" COLUMNS");
        if (select.getOutFileColumnsTerminatedBy() != null) {
          print(" TERMINATED BY ");
          select.getOutFileColumnsTerminatedBy().accept(this);
        }

        if (select.getOutFileColumnsEnclosedBy() != null) {
          if (select.isOutFileColumnsEnclosedOptionally()) {
            print(" OPTIONALLY");
          }
          print(" ENCLOSED BY ");
          select.getOutFileColumnsEnclosedBy().accept(this);
        }

        if (select.getOutFileColumnsEscaped() != null) {
          print(" ESCAPED BY ");
          select.getOutFileColumnsEscaped().accept(this);
        }
      }

      if (select.getOutFileLinesStartingBy() != null
          || select.getOutFileLinesTerminatedBy() != null) {
        print(" LINES");
        if (select.getOutFileLinesStartingBy() != null) {
          print(" STARTING BY ");
          select.getOutFileLinesStartingBy().accept(this);
        }

        if (select.getOutFileLinesTerminatedBy() != null) {
          print(" TERMINATED BY ");
          select.getOutFileLinesTerminatedBy().accept(this);
        }
      }
    }

    if (select.getFrom() != null) {
      println();
      print("FROM ");
      select.getFrom().accept(this);
    }

    if (select.getWhere() != null) {
      println();
      print("WHERE ");
      select.getWhere().accept(this);
    }

    if (select.getGroupBy() != null) {
      println();
      select.getGroupBy().accept(this);
    }

    if (select.getOrderBy() != null) {
      println();
      select.getOrderBy().accept(this);
    }

    if (select.getLimit() != null) {
      println();
      select.getLimit().accept(this);
    }

    if (select.getProcedureName() != null) {
      print(" PROCEDURE ");
      select.getProcedureName().accept(this);
      if (select.getProcedureArgumentList().size() > 0) {
        print("(");
        printAndAccept(select.getProcedureArgumentList(), ", ");
        print(")");
      }
    }

    if (select.isForUpdate()) {
      println();
      print("FOR UPDATE");
    }

    if (select.isLockInShareMode()) {
      println();
      print("LOCK IN SHARE MODE");
    }

    return false;
  }