public long addThesisUnique(ThesisUnique thesisUnique) throws StorageException {
   if (getThesisUniqueByContent(thesisUnique.getContent()) == null) {
     try {
       simpleJdbcTemplate
           .getJdbcOperations()
           .update(
               "INSERT INTO thesis_unique (content, frequency, last_scan, positivity, importance) VALUES(?,?,?,?,?)",
               new Object[] {
                 thesisUnique.getContent(),
                 thesisUnique.getFrequency(),
                 thesisUnique.getLastScan(),
                 thesisUnique.getPositivity(),
                 thesisUnique.getImportance()
               },
               new int[] {Types.VARCHAR, Types.INTEGER, Types.DATE, Types.DOUBLE, Types.DOUBLE});
       long lastId =
           simpleJdbcTemplate.getJdbcOperations().queryForLong("SELECT LAST_INSERT_ID()");
       return lastId;
     } catch (DataAccessException e) {
       log.error(
           "Error while inserting thesis_unique (probably not enough permissions): "
               + thesisUnique,
           e);
       throw new StorageException();
     }
   } else {
     updateThesisUniqueByContent(thesisUnique.getContent(), thesisUnique.getFrequency());
     // TODO must we fix this?
     return -1;
   }
 }
Exemplo n.º 2
0
  /**
   * Checks if the specified table exists
   *
   * @param template
   * @param tableName
   * @return
   */
  private boolean tableExists(
      SimpleJdbcTemplate template, final String schema, final String tableName) {
    try {
      DataSource ds = ((JdbcAccessor) template.getJdbcOperations()).getDataSource();
      return (Boolean)
          JdbcUtils.extractDatabaseMetaData(
              ds,
              new DatabaseMetaDataCallback() {

                public Object processMetaData(DatabaseMetaData dbmd)
                    throws SQLException, MetaDataAccessException {
                  ResultSet rs = null;
                  try {
                    rs = dbmd.getTables(null, schema, tableName.toLowerCase(), null);
                    boolean exists = rs.next();
                    rs.close();
                    if (exists) {
                      return true;
                    }
                    rs = dbmd.getTables(null, schema, tableName, null);
                    return rs.next();
                  } finally {
                    if (rs != null) {
                      rs.close();
                    }
                  }
                }
              });
    } catch (MetaDataAccessException e) {
      return false;
    }
  }
 public List<ThesisUnique> getAllThesesUnique() {
   List<ThesisUnique> products =
       simpleJdbcTemplate
           .getJdbcOperations()
           .query("SELECT * FROM thesis_unique", thesisUniqueMapper);
   return products;
 }
Exemplo n.º 4
0
 private boolean isModuleAlreadyInstalled(SimpleJdbcTemplate template) {
   if (!appliedIds.isEmpty()) {
     // Some changes have already been applied
     return true;
   }
   return checkTableName == null
       || DatabaseUtils.tableExists(template.getJdbcOperations(), new Table(checkTableName));
 }
 @Before
 public void onSetUp() throws Exception {
   getMessages(); // drain queue
   simpleJdbcTemplate.getJdbcOperations().execute("delete from T_BARS");
   jmsTemplate.convertAndSend("queue", "foo");
   provider =
       new ItemReader<String>() {
         public String read() {
           String text = (String) jmsTemplate.receiveAndConvert("queue");
           list.add(text);
           return text;
         }
       };
   retryTemplate = new RetryTemplate();
 }
Exemplo n.º 6
0
  /* (non-Javadoc)
   * @see com.qm.frame.dbtool.dialect.IDialect#getTables(FbrpDbtoolDs ds)
   */
  @Override
  public Map<String, FbrpDbtoolTable> getTables(final FbrpDbtoolDs ds) {
    // JDBC
    SimpleJdbcTemplate simpleJdbcTemplate = this.getDsService().getSimpleJdbcTemplate(ds.getId());
    if (simpleJdbcTemplate == null) {
      return null;
    }

    // 表信息列表
    Map<String, FbrpDbtoolTable> tableInfo =
        simpleJdbcTemplate
            .getJdbcOperations()
            .execute(
                new ConnectionCallback<Map<String, FbrpDbtoolTable>>() {
                  public Map<String, FbrpDbtoolTable> doInConnection(Connection conn)
                      throws SQLException, DataAccessException {
                    Map<String, FbrpDbtoolTable> map = new HashMap<String, FbrpDbtoolTable>();
                    String[] types = {"TABLE"};
                    ResultSet rs = null;
                    try {

                      DatabaseMetaData dmd = conn.getMetaData();

                      String url = (String) ds.getUrl();
                      url = url.substring(url.indexOf("=") + 1, url.indexOf(","));

                      rs = dmd.getTables("", url, "%", types);
                      while (rs.next()) {
                        FbrpDbtoolTable dbtable = new FbrpDbtoolTable();

                        // 表名
                        dbtable.setName(rs.getObject("TABLE_NAME").toString());
                        Object remark = rs.getObject("REMARKS");
                        dbtable.setRemark(remark == null ? "" : rs.getObject("REMARKS").toString());
                        if (!dbtable.getName().startsWith("BIN$")) {
                          map.put(dbtable.getName(), dbtable);
                        }
                      }
                    } catch (Exception exe) {
                      log.error("", exe);
                    } finally {
                      JdbcUtils.closeResultSet(rs);
                    }
                    return map;
                  }
                });
    return tableInfo;
  }
Exemplo n.º 7
0
 /**
  * Checks if the database schema is present, if missing it generates it
  *
  * @param template
  */
 public void initializeTables(String schema, SimpleJdbcTemplate template) {
   String prefix;
   if (schema == null) {
     prefix = "";
   } else {
     prefix = schema + ".";
   }
   for (String table : TABLE_CREATION_MAP.keySet()) {
     if (!tableExists(template, schema, table)) {
       for (String command : TABLE_CREATION_MAP.get(table)) {
         command = command.replace("${schema}", prefix);
         template.getJdbcOperations().execute(command);
       }
     }
   }
 }
 public ThesisUnique getThesisUniqueByContent(String content) {
   List<ThesisUnique> theses =
       simpleJdbcTemplate
           .getJdbcOperations()
           .query(
               "SELECT * FROM thesis_unique WHERE content = ?",
               new Object[] {content},
               new int[] {Types.VARCHAR},
               thesisUniqueMapper);
   // TODO is it good practice?
   if (theses.size() > 0) {
     return theses.get(0);
   } else {
     return null;
   }
 }
Exemplo n.º 9
0
 private Long getLastInsertId() {
   final List<Long> id = new ArrayList<Long>(1);
   jdbcTemplate
       .getJdbcOperations()
       .query(
           "select last_insert_id()",
           new RowCallbackHandler() {
             public void processRow(final ResultSet rs) throws SQLException {
               id.add(rs.getLong(1));
             }
           });
   if (id.isEmpty()) {
     throw new IllegalStateException("No inserted id");
   }
   return id.get(0);
 }
  public void updateThesisUniqueByContent(String content, int frequency) throws StorageException {

    ThesisUnique thesisUnique = getThesisUniqueByContent(content);
    frequency += thesisUnique.getFrequency();
    log.info("thesis = " + thesisUnique.getContent());
    try {
      simpleJdbcTemplate
          .getJdbcOperations()
          .update(
              "UPDATE thesis_unique SET frequency = ? WHERE content = ?",
              new Object[] {frequency, content});
    } catch (DataAccessException e) {
      log.error(
          "Error while updating thesis_unique (probably not enough permissions) with content : "
              + content,
          e);
      throw new StorageException();
    }
  }
Exemplo n.º 11
0
 /*
  * (non-Javadoc)
  *
  * @see
  * com.qm.frame.dbtool.dialect.impl.AbstractDialect#getTableColInfo
  * (com.qm.frame.dbtool.vo.FbrpDbtoolDsVO, java.lang.String)
  */
 @Override
 public Map<String, FbrpDbtoolTableCol> getTableColInfo(
     final FbrpDbtoolDs ds, final String tableName) {
   if (tableName == null) {
     return null;
   }
   // jdbc
   SimpleJdbcTemplate simpleJdbcTemplate = this.getDsService().getSimpleJdbcTemplate(ds.getId());
   if (simpleJdbcTemplate == null) {
     return null;
   }
   // 查询表的字段信息
   Map<String, FbrpDbtoolTableCol> tableColInfo =
       simpleJdbcTemplate
           .getJdbcOperations()
           .execute(
               new ConnectionCallback<Map<String, FbrpDbtoolTableCol>>() {
                 public Map<String, FbrpDbtoolTableCol> doInConnection(Connection conn)
                     throws SQLException, DataAccessException {
                   Map<String, FbrpDbtoolTableCol> dbtableColInfoVOs =
                       new HashMap<String, FbrpDbtoolTableCol>();
                   ResultSet rs = null;
                   try {
                     DatabaseMetaData dmd = conn.getMetaData();
                     String url = (String) ds.getUrl();
                     url = url.substring(url.indexOf("=") + 1, url.indexOf(","));
                     // 主键字段
                     rs = dmd.getPrimaryKeys("", url, tableName);
                     List<String> keyColumns = new ArrayList<String>();
                     while (rs.next()) {
                       keyColumns.add(rs.getObject("COLUMN_NAME").toString());
                     }
                     // 所有字段
                     rs = dmd.getColumns(conn.getCatalog(), url, tableName, "%");
                     String columnName = null;
                     // 排序号
                     int sortNo = 0;
                     while (rs.next()) {
                       FbrpDbtoolTableCol dntableColinfo = new FbrpDbtoolTableCol();
                       // 字段名
                       columnName = rs.getObject("COLUMN_NAME").toString();
                       dntableColinfo.setName(columnName);
                       // 是否是主键
                       if (keyColumns.contains(columnName)) {
                         dntableColinfo.setPrimaryKeyFlag(
                             Long.valueOf(BaseConstants.YESORNO_TYPE_YES));
                       } else {
                         dntableColinfo.setPrimaryKeyFlag(
                             Long.valueOf(BaseConstants.YESORNO_TYPE_NO));
                       }
                       // 允许为空
                       dntableColinfo.setNullable(
                           Long.valueOf(rs.getObject("NULLABLE").toString()));
                       // 长度
                       String length = String.valueOf(rs.getObject("COLUMN_SIZE").toString());
                       if (rs.getObject("DECIMAL_DIGITS") != null
                           && ((Integer) rs.getObject("DECIMAL_DIGITS")) != 0) {
                         length += "," + rs.getObject("DECIMAL_DIGITS").toString();
                       }
                       dntableColinfo.setLength(length);
                       // 备注
                       dntableColinfo.setRemark((String) rs.getObject("REMARKS"));
                       // 类型
                       dntableColinfo.setColType(rs.getObject("TYPE_NAME").toString());
                       // 默认值
                       dntableColinfo.setDefaultValue((String) rs.getObject("COLUMN_DEF"));
                       // 排序号
                       dntableColinfo.setSortNo(Long.valueOf(sortNo++));
                       dbtableColInfoVOs.put(dntableColinfo.getName(), dntableColinfo);
                     }
                   } catch (Exception exe) {
                     // TODO
                     // com.teradata.jdbc.jdbc_4.util.JDBCException:
                     // throw new ServiceException("没有对DBC.UDTInfo的SELECT权限");
                     // [Teradata Database][TeraJDBC 13.10.00.10][Error 3523][SQLState 42000]
                     // The user does not have SELECT access to
                     // DBC.UDTInfo.
                     log.error("", exe);
                   } finally {
                     JdbcUtils.closeResultSet(rs);
                   }
                   return dbtableColInfoVOs;
                 }
               });
   return tableColInfo;
 }