コード例 #1
0
ファイル: TableQueries.java プロジェクト: koberj/OpenIDM
    void setConfiguredQueries(Map<String, String> replacements, JsonValue queriesConfig) {
      configured.clear();
      for (String queryName : queriesConfig.keys()) {
        String rawQuery = queriesConfig.get(queryName).required().asString();

        TokenHandler tokenHandler = new TokenHandler();
        // Replace the table name tokens.
        String tempQueryString = tokenHandler.replaceSomeTokens(rawQuery, replacements);

        // Convert to ? for prepared statement, populate token replacement info
        List<String> tokenNames = tokenHandler.extractTokens(tempQueryString);
        String queryString = tokenHandler.replaceTokens(tempQueryString, "?", PREFIX_LIST);

        QueryInfo queryInfo = new QueryInfo(queryString, tokenNames);
        configured.put(queryName, queryInfo);
        logger.info(
            "Configured query converted to JDBC query {} and tokens {}", queryString, tokenNames);
      }
    }
コード例 #2
0
ファイル: TableQueries.java プロジェクト: koberj/OpenIDM
  /**
   * Resolves a query filter.
   *
   * @param con The db connection
   * @param filter the query filter to parse
   * @return A resolved statement
   */
  PreparedStatement parseQueryFilter(Connection con, QueryFilter filter, Map<String, Object> params)
      throws SQLException, ResourceException {
    Map<String, Object> replacementTokens = new LinkedHashMap<String, Object>();

    String rawQuery = tableHandler.renderQueryFilter(filter, replacementTokens, params);

    Map<String, String> replacements = new LinkedHashMap<String, String>();
    replacements.put("_mainTable", mainTableName);
    replacements.put("_propTable", propTableName);
    replacements.put("_dbSchema", dbSchemaName);

    TokenHandler tokenHandler = new TokenHandler();
    // Replace the table name tokens.
    String tempQueryString = tokenHandler.replaceSomeTokens(rawQuery, replacements);

    logger.debug("Tokenized statement: {} with replacementTokens: {}", rawQuery, replacementTokens);

    // Convert to ? for prepared statement, populate token replacement info
    List<String> tokenNames = tokenHandler.extractTokens(tempQueryString);
    String queryString = tokenHandler.replaceTokens(tempQueryString, "?", PREFIX_LIST);

    QueryInfo queryInfo = new QueryInfo(queryString, tokenNames);
    return resolveQuery(queryInfo, con, replacementTokens);
  }
コード例 #3
0
ファイル: TableQueries.java プロジェクト: koberj/OpenIDM
  /**
   * Resolves a query, given a QueryInfo
   *
   * @param info The info encapsulating the query information
   * @param con the db connection
   * @param params the parameters passed to query
   * @return the resolved query
   * @throws SQLException if resolving the query failed
   */
  PreparedStatement resolveQuery(QueryInfo info, Connection con, Map<String, Object> params)
      throws SQLException, ResourceException {
    String queryStr = info.getQueryString();
    List<String> tokenNames = info.getTokenNames();

    // replace ${list:variable} tokens with the correct number of bind variables
    Map<String, Integer> listReplacements = new HashMap<String, Integer>();
    for (String tokenName : tokenNames) {
      String[] tokenParts = tokenName.split(":", 2);
      if (PREFIX_LIST.equals(tokenParts[0]) && params.containsKey(tokenParts[1])) {
        listReplacements.put(tokenName, ((String) params.get(tokenParts[1])).split(",").length);
      }
    }
    if (listReplacements.size() > 0) {
      TokenHandler tokenHandler = new TokenHandler();
      queryStr = tokenHandler.replaceListTokens(queryStr, listReplacements, "?");
    }

    // now prepare the statement using the correct number of bind variables
    PreparedStatement statement = getPreparedStatement(con, queryStr);
    int count = 1; // DB column count starts at 1
    for (String tokenName : tokenNames) {
      String[] tokenParts = tokenName.split(":", 2);
      if (tokenParts.length == 1) {
        // handle single value - assume String
        Object objValue = params.get(tokenName);
        String value = null;
        if (objValue != null) {
          value = trimValue(objValue);
        } else {
          // fail with an exception if token not found
          throw new BadRequestException(
              "Missing entry in params passed to query for token " + tokenName);
        }
        statement.setString(count, value);
        count++;
      } else {
        Object objValue = params.get(tokenParts[1]);
        if (objValue == null) {
          // fail with an exception if token not found
          throw new BadRequestException(
              "Missing entry in params passed to query for token " + tokenName);
        }
        if (PREFIX_INT.equals(tokenParts[0])) {
          // handle single integer value
          Integer int_value = null;
          if (objValue != null) {
            int_value = Integer.parseInt(objValue.toString());
          }
          statement.setInt(count, int_value);
          count++;
        } else if (PREFIX_LIST.equals(tokenParts[0])) {
          // handle list of values - presently assumes Strings, TODO support integer lists
          if (objValue != null) {
            for (String list_value : objValue.toString().split(",")) {
              // if list value is surrounded by single quotes remove them
              if (list_value != null && list_value.startsWith("'") && list_value.endsWith("'")) {
                list_value = list_value.substring(1, list_value.length() - 1);
              }
              statement.setString(count, trimValue(list_value));
              count++;
            }
          } else {
            statement.setString(count, null);
            count++;
          }
        }
      }
    }
    logger.debug("Prepared statement: {}", statement);

    return statement;
  }