diff --git a/bindings/java/src/org/sleuthkit/datamodel/SleuthkitCase.java b/bindings/java/src/org/sleuthkit/datamodel/SleuthkitCase.java index 44e3d60f8e701cfd74def6e4e452eb4e0b42a08e..7024417ce66c424273822a8b6cacd50b58fb6b6f 100644 --- a/bindings/java/src/org/sleuthkit/datamodel/SleuthkitCase.java +++ b/bindings/java/src/org/sleuthkit/datamodel/SleuthkitCase.java @@ -9836,19 +9836,14 @@ public TagName addOrUpdateTagName(String displayName, String description, TagNam statement.setString(3, color.getName()); statement.setByte(4, knownStatus.getFileKnownValue()); connection.executeUpdate(statement); - resultSet = statement.getGeneratedKeys(); - resultSet.next(); - - long tagId = resultSet.getLong(1); - resultSet.close(); - statement = connection.getPreparedStatement(PREPARED_STATEMENT.SELECT_TAG_NAME_BY_ID); + statement = connection.getPreparedStatement(PREPARED_STATEMENT.SELECT_TAG_NAME_BY_NAME); statement.clearParameters(); - statement.setLong(1, tagId); + statement.setString(1, displayName); resultSet = connection.executeQuery(statement); resultSet.next(); - return new TagName(tagId, displayName, description, color, knownStatus, resultSet.getLong("tag_set_id"), resultSet.getInt("rank")); + return new TagName(resultSet.getLong("tag_name_id"), displayName, description, color, knownStatus, resultSet.getLong("tag_set_id"), resultSet.getInt("rank")); } catch (SQLException ex) { throw new TskCoreException("Error adding row for " + displayName + " tag name to tag_names table", ex); @@ -11291,7 +11286,8 @@ private enum PREPARED_STATEMENT { INSERT_POOL_INFO("INSERT INTO tsk_pool_info (obj_id, pool_type) VALUES (?, ?)"), INSERT_FS_INFO("INSERT INTO tsk_fs_info (obj_id, data_source_obj_id, img_offset, fs_type, block_size, block_count, root_inum, first_inum, last_inum, display_name)" + "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"), - SELECT_TAG_NAME_BY_ID("SELECT * FROM tag_names where tag_name_id = ?"); + SELECT_TAG_NAME_BY_ID("SELECT * FROM tag_names where tag_name_id = ?"), + SELECT_TAG_NAME_BY_NAME("SELECT * FROM tag_names where display_name = ?"); private final String sql;