source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2017 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.server.platform.db.migration.step;
  21
  22 import java.io.ByteArrayInputStream;
  23 import java.sql.PreparedStatement;
  24 import java.sql.SQLException;
  25 import java.sql.Timestamp;
  26 import java.sql.Types;
  27 import java.util.Date;
  28 import javax.annotation.Nullable;
  29 import org.apache.commons.dbutils.DbUtils;
  30
  31 class BaseSqlStatement<CHILD extends SqlStatement> implements SqlStatement<CHILD> {
  32   protected PreparedStatement pstmt;
  33
  34   protected BaseSqlStatement(PreparedStatement pstmt) {
  35     this.pstmt = pstmt;
  36   }
  37
  38   @Override
  39   public void close() {
  40     DbUtils.closeQuietly(pstmt);
  41     pstmt = null;
  42   }
  43
  44   @Override
  45   public CHILD setString(int columnIndex, @Nullable String value) throws SQLException {
  46     pstmt.setString(columnIndex, value);
  47     return (CHILD) this;
  48   }
  49
  50   @Override
  51   public CHILD setInt(int columnIndex, @Nullable Integer value) throws SQLException {
  52     if (value == null) {
  53       pstmt.setNull(columnIndex, Types.INTEGER);
  54     } else {
  55       pstmt.setInt(columnIndex, value);
  56     }
  57     return (CHILD) this;
  58   }
  59
  60   @Override
  61   public CHILD setLong(int columnIndex, @Nullable Long value) throws SQLException {
  62     if (value == null) {
  63       pstmt.setNull(columnIndex, Types.BIGINT);
  64     } else {
  65       pstmt.setLong(columnIndex, value);
  66     }
  67     return (CHILD) this;
  68   }
  69
  70   @Override
  71   public CHILD setBoolean(int columnIndex, @Nullable Boolean value) throws SQLException {
  72     if (value == null) {
  73       pstmt.setNull(columnIndex, Types.BOOLEAN);
  74     } else {
  75       pstmt.setBoolean(columnIndex, value);
  76     }
  77     return (CHILD) this;
  78   }
  79
  80   @Override
  81   public CHILD setBytes(int columnIndex, @Nullable byte[] value) throws SQLException {
  82     if (value == null) {
  83       pstmt.setNull(columnIndex, Types.BINARY);
  84     } else {
  85       pstmt.setBinaryStream(columnIndex, new ByteArrayInputStream(value));
  86     }
  87     return (CHILD) this;
  88   }
  89
  90   @Override
  91   public CHILD setDouble(int columnIndex, @Nullable Double value) throws SQLException {
  92     if (value == null) {
  93       pstmt.setNull(columnIndex, Types.DECIMAL);
  94     } else {
  95       pstmt.setDouble(columnIndex, value);
  96     }
  97     return (CHILD) this;
  98   }
  99
 100   @Override
 101   public CHILD setDate(int columnIndex, @Nullable Date value) throws SQLException {
 102     if (value == null) {
 103       pstmt.setNull(columnIndex, Types.TIMESTAMP);
 104     } else {
 105       pstmt.setTimestamp(columnIndex, new Timestamp(value.getTime()));
 106     }
 107     return (CHILD) this;
 108   }
 109 }