Skip to content

Commit def34f4

Browse files
committed
Fix FileUtility#getFileNames(...) extension removal
1 parent a71551b commit def34f4

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/main/java/xyz/srnyx/javautilities/FileUtility.java

+6-4
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import java.nio.file.Files;
88
import java.nio.file.Path;
99
import java.util.Arrays;
10-
import java.util.HashSet;
10+
import java.util.Collections;
1111
import java.util.Set;
1212
import java.util.stream.Collectors;
1313

@@ -41,11 +41,13 @@ public static void deleteFile(@NotNull Path path, boolean silentFail) {
4141
@NotNull
4242
public static Set<String> getFileNames(@NotNull File file, @NotNull String extension) {
4343
final File[] files = file.listFiles();
44-
if (files == null) return new HashSet<>();
44+
if (files == null) return Collections.emptySet();
45+
final String fullExtension = "." + extension;
46+
final int extensionLength = fullExtension.length();
4547
return Arrays.stream(files)
4648
.map(File::getName)
47-
.filter(name -> name.endsWith("." + extension))
48-
.map(name -> name.substring(0, name.length() - 4))
49+
.filter(name -> name.endsWith(fullExtension))
50+
.map(name -> name.substring(0, name.length() - extensionLength))
4951
.collect(Collectors.toSet());
5052
}
5153

0 commit comments

Comments
 (0)