Return inserted row as a result of Insert action

[shulkaolka]

1. Define Primary Key from dbMetadata

2. Define is Primary Key generated

3. After executing Insert query:
   3.1. If the table contains PrimaryKey:

   • If Primary Key is generated - get Primary Key Value from the statement
   • If Primary Key isn't generated - get Primary Key Value from body

   execute query Select By Primary Key and return selected row
   3.2. If the table doesn't contain PrimaryKey - return input body

[shulkaolka]

As example:

  public JsonObject executeInsertGeneral(Connection connection, String tableName, JsonObject body,
      String primaryKeyName)
      throws SQLException {
    validateQuery();
    JsonObject keysValues = getKeyValue(body);
    String keys = keysValues.getString("keys");
    String values = keysValues.getString("values");
    String sql = "INSERT INTO " + tableName +
        " (" + keys + ")" +
        " VALUES (" + values + ")";
    try (PreparedStatement stmt = connection
        .prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {
      int i = 1;
      for (String key : body.keySet()) {
        Utils.setStatementParam(stmt, i, key, body);
        i++;
      }
      int affectedRows = stmt.executeUpdate();
      if (affectedRows != 1) {
        throw new SQLException("Inserting failed, no rows affected.");
      }
      if (primaryKeyName != null) {
        JsonObjectBuilder primaryKeyBody = Json.createObjectBuilder();
        int generatedKeysCount = 0;
        try (ResultSet generatedKeys = stmt.getGeneratedKeys()) {
          while (generatedKeys.next()) {
            primaryKeyBody.add(primaryKeyName, generatedKeys.getLong(1));
            generatedKeysCount++;
            if (generatedKeysCount > 1) {
              throw new SQLException("More than one generated key!");
            }
          }
        }
        if (generatedKeysCount == 0) {
          return getRowByPrimaryKey(connection, primaryKeyName, body);
        }
        return getRowByPrimaryKey(connection, primaryKeyName, primaryKeyBody.build());
      } else {
        return body;
      }
    }
  }