content: /*
   * Copyright 1999-2018 Alibaba Group Holding Ltd.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package com.alibaba.druid.bvt.filter;
  
  import com.alibaba.druid.mock.MockBlob;
  import com.alibaba.druid.mock.MockClob;
  import com.alibaba.druid.mock.MockDriver;
  import com.alibaba.druid.mock.MockNClob;
  import com.alibaba.druid.mock.MockRowId;
  import com.alibaba.druid.mock.MockStatementBase;
  import com.alibaba.druid.pool.DruidDataSource;
  import com.alibaba.druid.stat.JdbcSqlStat;
  import com.alibaba.druid.support.json.JSONUtils;
  import com.alibaba.druid.util.JdbcUtils;
  import junit.framework.TestCase;
  import org.nutz.lang.util.ByteInputStream;
  
  import java.math.BigDecimal;
  import java.sql.Connection;
  import java.sql.PreparedStatement;
  import java.sql.ResultSet;
  import java.sql.SQLException;
  import java.sql.Time;
  import java.sql.Types;
  import java.text.SimpleDateFormat;
  import java.util.Date;
  import java.util.List;
  
  public class StatFilterBuildSlowParameterTest extends TestCase {
      private DruidDataSource dataSource;
  
      protected void setUp() throws Exception {
          dataSource = new DruidDataSource();
  
          dataSource.setUrl("jdbc:mock:xxx");
          dataSource.setFilters("stat");
          dataSource.setTestOnBorrow(false);
          dataSource.setConnectionProperties("druid.stat.slowSqlMillis=1");
  
          MockDriver driver = new MockDriver() {
              public ResultSet executeQuery(MockStatementBase stmt, String sql) throws SQLException {
                  try {
                      Thread.sleep(2);
                  } catch (InterruptedException e) {
                      e.printStackTrace();
                  }
                  return super.executeQuery(stmt, sql);
              }
          };
  
          dataSource.setDriver(driver);
  
          dataSource.init();
      }
  
      protected void tearDown() throws Exception {
          JdbcUtils.close(dataSource);
      }
  
      public void test_buildSlowSql() throws Exception {
          long currentMillis = System.currentTimeMillis();
          String sql = "select ?, ?, ?, ?, ?";
  
          SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
          Date date = new Date(System.currentTimeMillis());
          String dateText = dateFormat.format(date);
  
          {
              Connection conn = dataSource.getConnection();
  
              PreparedStatement stmt = conn.prepareStatement(sql);
  
              stmt.setBoolean(1, true);
              stmt.setInt(2, 123);
              stmt.setLong(3, 10001);
              stmt.setTimestamp(4, new java.sql.Timestamp(currentMillis));
              stmt.setDate(5, new java.sql.Date(currentMillis));
  
              ResultSet rs = stmt.executeQuery();
              rs.close();
  
              stmt.close();
  
              conn.close();
  
              // //////
  
              JdbcSqlStat sqlStat = dataSource.getDataSourceStat().getSqlStat(sql);
              assertNotNull(sqlStat);
  
              String slowParameters = sqlStat.getLastSlowParameters();
              assertNotNull(slowParameters);
  
              List<Object> parameters = (List<Object>) JSONUtils.parse(slowParameters);
              assertEquals(5, parameters.size());
  
              assertEquals(true, parameters.get(0));
              assertEquals(123, parameters.get(1));
              assertEquals(10001, parameters.get(2));
              assertEquals(dateText, parameters.get(3));
              assertEquals(dateText, parameters.get(4));
          }
  
          currentMillis = System.currentTimeMillis();
          date = new Date(System.currentTimeMillis());
          dateText = dateFormat.format(date);
          {
              Connection conn = dataSource.getConnection();
  
              PreparedStatement stmt = conn.prepareStatement(sql);
  
              stmt.setBoolean(1, false);
              stmt.setInt(2, 234);
              stmt.setLong(3, 10002);
              stmt.setTimestamp(4, new java.sql.Timestamp(currentMillis));
              stmt.setDate(5, new java.sql.Date(currentMillis));
  
              ResultSet rs = stmt.executeQuery();
              rs.close();
  
              stmt.close();
  
              conn.close();
  
              // //////
  
              JdbcSqlStat sqlStat = dataSource.getDataSourceStat().getSqlStat(sql);
              assertNotNull(sqlStat);
  
              String slowParameters = sqlStat.getLastSlowParameters();
              assertNotNull(slowParameters);
  
              List<Object> parameters = (List<Object>) JSONUtils.parse(slowParameters);
              assertEquals(5, parameters.size());
  
              assertEquals(false, parameters.get(0));
              assertEquals(234, parameters.get(1));
              assertEquals(10002, parameters.get(2));
              assertEquals(dateText, parameters.get(3));
              assertEquals(dateText, parameters.get(4));
          }
  
          {
              StringBuilder buf = new StringBuilder();
              for (int i = 0; i < 10; ++i) {
                  buf.append("abcdefghijklmnABCDEFGHIJKLMN1234567890!@#$%^&*(");
              }
  
              Connection conn = dataSource.getConnection();
  
              PreparedStatement stmt = conn.prepareStatement(sql);
  
              stmt.setNull(1, Types.VARCHAR);
              stmt.setString(2, buf.toString());
              stmt.setClob(3, new MockClob());
              stmt.setNClob(4, new MockNClob());
              stmt.setBlob(5, new MockBlob());
  
              ResultSet rs = stmt.executeQuery();
              rs.close();
  
              stmt.close();
  
              conn.close();
  
              // //////
  
              JdbcSqlStat sqlStat = dataSource.getDataSourceStat().getSqlStat(sql);
              assertNotNull(sqlStat);
  
              String slowParameters = sqlStat.getLastSlowParameters();
              assertNotNull(slowParameters);
  
              List<Object> parameters = (List<Object>) JSONUtils.parse(slowParameters);
              assertEquals(5, parameters.size());
  
              assertEquals(null, parameters.get(0));
              assertEquals(buf.substring(0, 97) + "...", parameters.get(1));
              assertEquals("<Clob>", parameters.get(2));
              assertEquals("<NClob>", parameters.get(3));
              assertEquals("<Blob>", parameters.get(4));
          }
          {
              StringBuilder buf = new StringBuilder();
              for (int i = 0; i < 10; ++i) {
                  buf.append("中国abcdefghijklmnABCDEFGHIJKLMN1234567890!@#$%^&*(");
              }
  
              Connection conn = dataSource.getConnection();
  
              PreparedStatement stmt = conn.prepareStatement(sql);
  
              stmt.setBinaryStream(1, new ByteInputStream(new byte[0]));
              stmt.setString(2, buf.toString());
              stmt.setTime(3, new Time(currentMillis));
              stmt.setBigDecimal(4, new BigDecimal("56789.123"));
              stmt.setRowId(5, new MockRowId());
  
              ResultSet rs = stmt.executeQuery();
              rs.close();
  
              stmt.close();
  
              conn.close();
  
              // //////
  
              JdbcSqlStat sqlStat = dataSource.getDataSourceStat().getSqlStat(sql);
              assertNotNull(sqlStat);
  
              String slowParameters = sqlStat.getLastSlowParameters();
              assertNotNull(slowParameters);
  
              List<Object> parameters = (List<Object>) JSONUtils.parse(slowParameters);
              assertEquals(5, parameters.size());
  
              assertEquals("<InputStream>", parameters.get(0));
              assertEquals(buf.substring(0, 97) + "...", parameters.get(1));
              assertEquals(dateText, parameters.get(2));
              assertEquals(56789.123, parameters.get(3));
              assertEquals("<com.alibaba.druid.mock.MockRowId>", parameters.get(4));
          }
      }
  }
  